Update main.py

main.py

@@ -58,7 +58,7 @@ def create_agent(filename: str):
     df = pd.read_csv(filename)
 
     # Create a Pandas DataFrame agent.
-    return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION)
+    return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
 
 def query_agent(agent, query):
     """
@@ -178,7 +178,7 @@ async def on_message(message: cl.Message):
     cb = cl.AsyncLangchainCallbackHandler()
     try:
         #res = await agent.acall("Réponds en langue française à la question suivante : " + message.content, callbacks=[cb])
-        res = await agent.ainvoke("Réponds en langue française à la question suivante : " + message.content)
+        res = await agent.ainvoke("Réponds de la manière la plus complète et la plus intelligible, en langue française, à la question suivante : " + message.content + ". Réponds au format markdown ou au format tableau si le résultat nécessite l'affichage d'un tableau.")
         await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(res['output'])).send()
     except ValueError as e:
         res = str(e)