added csv-agent-default in next.js

controller.py

@@ -621,7 +621,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         chat_id = request.get("chat_id", "")
         
         if generate_report is True:
-            report_files = await generate_csv_report(csv_url, query, chat_id)
+            report_files = await generate_csv_report(csv_url, query, chat_id, conversation_history)
             if report_files is not None:
                 return {"orchestrator_response": jsonable_encoder(report_files)}
 

gemini_report_generator.py

@@ -146,6 +146,7 @@ class RethinkAgent(BaseModel):
     current_retry: int = Field(default=0, ge=0)
     repl: Optional[PythonREPL] = None
     key_manager: Optional[GeminiKeyManager] = None
+    conversation: List[Dict[str, Any]] = Field(default_factory=list)
     
     class Config:
         arbitrary_types_allowed = True
@@ -155,7 +156,10 @@ class RethinkAgent(BaseModel):
         return code_match.group(1).strip() if code_match else response.strip()
     
     def _generate_initial_prompt(self, query: str) -> str:
-        return f"""Generate DIRECT EXECUTION CODE (no functions, no explanations) following STRICT RULES:
+        initial_prompt = f"""Generate DIRECT EXECUTION CODE (no functions, no explanations) following STRICT RULES:
+    
+        CONVERSATION HISTORY:
+        {json.dumps(self.conversation)}
         
         MANDATORY REQUIREMENTS:
         1. Operate directly on existing 'df' variable
@@ -189,6 +193,8 @@ class RethinkAgent(BaseModel):
         # Save results
         sales_by_region.to_csv(f'{{output_dir}}/sales_by_region.csv')
         """
+        logger.info(initial_prompt)
+        return initial_prompt
 
     def _generate_retry_prompt(self, query: str, error: str, code: str) -> str:
         return f"""FIX THIS CODE (failed with: {error}) by STRICTLY FOLLOWING:
@@ -261,10 +267,10 @@ class RethinkAgent(BaseModel):
             "image_files": []
         }
 
-def gemini_llm_chat(csv_url: str, query: str) -> Dict[str, Any]:
+def gemini_llm_chat(csv_url: str, query: str, conversation_history: List[Dict[str, Any]]) -> Dict[str, Any]:
     try:
         df = pd.read_csv(csv_url)
-        agent = RethinkAgent(df=df)
+        agent = RethinkAgent(df=df, conversation_history=conversation_history)
         
         if not agent.initialize_model(API_KEYS):
             return {"error": "API configuration failed"}
@@ -288,9 +294,9 @@ def gemini_llm_chat(csv_url: str, query: str) -> Dict[str, Any]:
         }
         
 
-async def generate_csv_report(csv_url: str, query: str, chat_id: str) -> FileBoxProps:
+async def generate_csv_report(csv_url: str, query: str, chat_id: str, conversation_history: List[Dict[str, Any]]) -> FileBoxProps:
     try:
-        result = gemini_llm_chat(csv_url, query)
+        result = gemini_llm_chat(csv_url, query, conversation_history)
         logger.info(f"Raw result from gemini_llm_chat: {result}")
         
         csv_files = []