Redmind commited on
Commit
a062e5a
·
verified ·
1 Parent(s): 0abb8fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -503,7 +503,10 @@ def chat_with_llm(df,question):
503
  def bind_llm(llm, tools,prompt_template):
504
  llm = llm.bind()
505
  agent = create_tool_calling_agent(llm, tools, ChatPromptTemplate.from_template(prompt_template))
506
- agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
 
 
 
507
  return agent_executor
508
 
509
  # Define input and output models using Pydantic
 
503
  def bind_llm(llm, tools,prompt_template):
504
  llm = llm.bind()
505
  agent = create_tool_calling_agent(llm, tools, ChatPromptTemplate.from_template(prompt_template))
506
+ agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True,
507
+ agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, # or CONVERSATIONAL_REACT_DESCRIPTION
508
+
509
+ agent_kwargs={"system_message": agent_prompt})
510
  return agent_executor
511
 
512
  # Define input and output models using Pydantic