Update main.py

main.py

@@ -11,6 +11,7 @@ import os
 import pandas as pd
 import numpy as np
 
+from langchain.agents import AgentExecutor
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_csv_agent
 
@@ -71,7 +72,13 @@ async def on_chat_start():
             verbose=False,
             agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
         )
-        cl.user_session.set("runnable", agent)
+        agent_executor = AgentExecutor(
+            agent=agent,
+            tools=tools,
+            verbose=False,
+            max_iterations=2,
+        )
+        cl.user_session.set("runnable", agent_executor)
 
 @cl.on_message
 async def on_message(message: cl.Message):
@@ -79,7 +86,7 @@ async def on_message(message: cl.Message):
     runnable = cl.user_session.get("runnable")  # type: Runnable
     cb = cl.AsyncLangchainCallbackHandler()
     try:
-        res = await runnable.acall("Tu es un spécialiste du marché de l'emploi. Tu es en capacité d'analyser les emplois et d'en décrire les caractéristiques par rapport aux types de contrat, par rapport aux compétences professionnelles, par rapport aux niveaux de qualification. Réponds en langue française à la question suivante : " + message.content + "\nSi l'information est manquante ou de type nan, fais un projection ou une estimation.\nDétaille la réponse en faisant une analyse complète sachant que la colonne Poste correspond à la colonne des emplois, la colonne Contrat correspond à la colonne des types de contrat, la colonne Savoir correspond à la colonne des compétences professionnelles demandées par les employeurs, la colonne Localisation correspond à la colonne des départements, la colonne Niveau correspond à la colonne des niveaux de qualification.", callbacks=[cb])
+        res = await runnable.ainvoke({"input": "Tu es un spécialiste du marché de l'emploi. Tu es en capacité d'analyser les emplois et d'en décrire les caractéristiques par rapport aux types de contrat, par rapport aux compétences professionnelles, par rapport aux niveaux de qualification. Réponds en langue française à la question suivante : " + message.content + "\nSi l'information est manquante ou de type nan, fais un projection ou une estimation.\nDétaille la réponse en faisant une analyse complète sachant que la colonne Poste correspond à la colonne des emplois, la colonne Contrat correspond à la colonne des types de contrat, la colonne Savoir correspond à la colonne des compétences professionnelles demandées par les employeurs, la colonne Localisation correspond à la colonne des départements, la colonne Niveau correspond à la colonne des niveaux de qualification."}, callbacks=[cb])
         await cl.Message(author="COPILOT",content=res['output']).send()
         listPrompts_name = f"Liste des requêtes"
         prompt_elements = []