Update main.py

main.py

@@ -13,7 +13,11 @@ 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
+#from langchain_experimental.agents.agent_toolkits import create_csv_agent
+
+from langchain_community.agent_toolkits import create_sql_agent
+from langchain_community.utilities import SQLDatabase
+from sqlalchemy import create_engine
 
 from deep_translator import GoogleTranslator
 
@@ -69,13 +73,22 @@ async def on_chat_start():
         llm = HuggingFaceEndpoint(
             repo_id=repo_id, temperature=1.0
         )
-        agent = create_csv_agent(
-            llm,
-            "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv",
-            verbose=False,
-            agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-            max_iterations=5,
-        )
+        
+        
+        df = pd.read_csv("https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv")
+        engine = create_engine("sqlite:///sphinx" + str(cl.user_session.get("id")) + ".db")
+        df.to_sql("sphinx" + str(cl.user_session.get("id")), engine, index=False)
+        db = SQLDatabase(engine=engine)
+        agent = create_sql_agent(llm, db=db, verbose=True)
+        
+        
+        #agent = create_csv_agent(
+        #    llm,
+        #    "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv",
+        #    verbose=False,
+        #    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+        #    max_iterations=5,
+        #)
         cl.user_session.set("runnable", agent)
 
 @cl.on_message