Update main.py
Browse files
main.py
CHANGED
@@ -83,7 +83,7 @@ def create_agent(filename: str):
|
|
83 |
db = cl.user_session.get("db")
|
84 |
# Create a SAL agent.
|
85 |
#e.g agent_executor.invoke({"input": "Quel est le nombre de chargé d'affaires en agencement par entreprise?"})
|
86 |
-
return create_sql_agent(llm, db=db, agent_type="
|
87 |
|
88 |
def query_agent(agent, query):
|
89 |
"""
|
|
|
83 |
db = cl.user_session.get("db")
|
84 |
# Create a SAL agent.
|
85 |
#e.g agent_executor.invoke({"input": "Quel est le nombre de chargé d'affaires en agencement par entreprise?"})
|
86 |
+
return create_sql_agent(llm, db=db, agent_type="zero-shot-react-description", verbose=False)
|
87 |
|
88 |
def query_agent(agent, query):
|
89 |
"""
|