Paperbag commited on
Commit
f958cf7
·
1 Parent(s): 334e8ea
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -144,7 +144,7 @@ def restart_required(state: AgentState) -> AgentState:
144
  # return {"messages": messages + [response]}
145
 
146
  # Augment the LLM with tools
147
- tools = [web_search]
148
  tools_by_name = {tool.name: tool for tool in tools}
149
  model_with_tools = model.bind_tools(tools)
150
 
 
144
  # return {"messages": messages + [response]}
145
 
146
  # Augment the LLM with tools
147
+ tools = [web_search,wiki_search]
148
  tools_by_name = {tool.name: tool for tool in tools}
149
  model_with_tools = model.bind_tools(tools)
150