Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -59,8 +59,8 @@ def wiki_search(query: str) -> list:
|
|
| 59 |
return llama_docs
|
| 60 |
|
| 61 |
tools = [
|
| 62 |
-
FunctionTool(
|
| 63 |
-
FunctionTool(
|
| 64 |
]
|
| 65 |
|
| 66 |
agent = AgentWorkflow.from_tools_or_functions(tools, llm=llm)
|
|
|
|
| 59 |
return llama_docs
|
| 60 |
|
| 61 |
tools = [
|
| 62 |
+
FunctionTool(web_search, name="web_search", description="Tavily 3-hit search"),
|
| 63 |
+
FunctionTool(wiki_search, name="wiki_search", description="Wikipedia 2-page fetch"),
|
| 64 |
]
|
| 65 |
|
| 66 |
agent = AgentWorkflow.from_tools_or_functions(tools, llm=llm)
|