Update main.py

main.py

@@ -1,4 +1,6 @@
 from langchain_community.llms import HuggingFaceEndpoint
+from langchain_community.utilities import SQLDatabase
+from sqlalchemy import create_engine
 from langchain_core.prompts import PromptTemplate
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain.memory import ChatMessageHistory, ConversationBufferMemory
@@ -6,13 +8,15 @@ from langchain.schema import StrOutputParser
 from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableLambda
 from langchain.schema.runnable.config import RunnableConfig
 from langchain.schema import StrOutputParser
+from langchain_community.agent_toolkits import create_sql_agent
+
 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
+#from langchain_experimental.agents.agent_toolkits import create_csv_agent
 
 from deep_translator import GoogleTranslator
 
@@ -43,12 +47,24 @@ 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/enquete-DClic-juin2022-dataviz-ord1.csv",
-        verbose=True,
-        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
-    )
+    if cl.user_session.get("createdb") == None:
+        df = pd.read_csv("https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv")
+        engine = create_engine("sqlite:///epnum.db")
+        df.to_sql("epnum", engine, index=False)
+        db = SQLDatabase(engine=engine)
+        cl.user_session.set("createdb", "OK")
+        cl.user_session.set("db", db)
+    else:
+        db = cl.user_session.get("db")
+    
+    agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True)
+    
+    #agent = create_csv_agent(
+    #    llm,
+    #    "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv",
+    #    verbose=True,
+    #    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
+    #)
     cl.user_session.set("runnable", agent)
 
 @cl.on_message