add wiki
Browse files
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 |
|