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

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="openai-tools", 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="tool-calling", verbose=False)
87
 
88
  def query_agent(agent, query):
89
  """