datacipen commited on
Commit
593a90d
·
verified ·
1 Parent(s): 6cfb001

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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="tool-calling", verbose=False)
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
  """