added gemini too

controller.py

@@ -301,6 +301,8 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         csv_url = request.get("csv_url")
         decoded_url = unquote(csv_url)
         detailed_answer = request.get("detailed_answer")
+        conversation_history = request.get("conversation_history", [])
+        return {"answer": jsonable_encoder(conversation_history)}
 
         if if_initial_chat_question(query):
             answer = await asyncio.to_thread(
@@ -312,7 +314,7 @@ async def csv_chat(request: Dict, authorization: str = Header(None)):
         # Orchestrate the execution
         if detailed_answer is True:
            orchestrator_answer = await asyncio.to_thread(
-            csv_orchestrator_chat, decoded_url, query
+            csv_orchestrator_chat, decoded_url, query, conversation_history
            )
            if orchestrator_answer is not None:
               return {"answer": jsonable_encoder(orchestrator_answer)}
@@ -798,6 +800,8 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         query = request.get("query", "")
         csv_url = unquote(request.get("csv_url", ""))
         detailed_answer = request.get("detailed_answer", False)
+        conversation_history = request.get("conversation_history", [])
+        return {"orchestrator_response": jsonable_encoder(conversation_history)}
         
         loop = asyncio.get_running_loop()
         # First, try the langchain-based method if the question qualifies
@@ -817,7 +821,7 @@ async def csv_chart(request: dict, authorization: str = Header(None)):
         # Use orchestrator to handle the user's chart query first
         if detailed_answer is True:
            orchestrator_answer = await asyncio.to_thread(
-             csv_orchestrator_chat, csv_url, query
+             csv_orchestrator_chat, csv_url, query, conversation_history
            )
         
            if orchestrator_answer is not None:

orchestrator_agent.py

@@ -90,7 +90,7 @@ async def generate_chart(csv_url: str, user_questions: List[str]) -> Any:
     return charts
 
 # Function to create an agent with a specific CSV URL
-def create_agent(csv_url: str, api_key: str) -> Agent:
+def create_agent(csv_url: str, api_key: str, conversation_history: List) -> Agent:
     csv_metadata = get_csv_basic_info(csv_url)
     
     system_prompt = f"""
@@ -130,6 +130,7 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
 ## Current Context:
 - Working with CSV_URL: {csv_url}
 - Dataset overview: {csv_metadata}
+- Your conversation history: {conversation_history}
 - Output format: Markdown compatible
 
 ## Response Template:
@@ -192,7 +193,7 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
         system_prompt=system_prompt
     )
 
-def csv_orchestrator_chat(csv_url: str, user_question: str) -> str:
+def csv_orchestrator_chat(csv_url: str, user_question: str, conversation_history: List) -> str:
     print("CSV URL:", csv_url)
     print("User questions:", user_question)
 
@@ -200,7 +201,7 @@ def csv_orchestrator_chat(csv_url: str, user_question: str) -> str:
     for api_key in GEMINI_API_KEYS:
         try:
             print(f"Attempting with API key: {api_key}")
-            agent = create_agent(csv_url, api_key)
+            agent = create_agent(csv_url, api_key, conversation_history)
             result = agent.run_sync(user_question)
             print("Orchestrator Result:", result.data)
             return result.data