Spaces:
Sleeping
Sleeping
Update Agent's tool list
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
|
|
| 36 |
import wikipedia
|
| 37 |
|
| 38 |
@tool
|
| 39 |
-
def
|
| 40 |
"""
|
| 41 |
A tool that fetches a Wikipedia summary for a given topic.
|
| 42 |
|
|
@@ -79,7 +79,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 79 |
|
| 80 |
agent = CodeAgent(
|
| 81 |
model=model,
|
| 82 |
-
tools=[final_answer], ## add your tools here (don't remove final answer)
|
| 83 |
max_steps=6,
|
| 84 |
verbosity_level=1,
|
| 85 |
grammar=None,
|
|
|
|
| 36 |
import wikipedia
|
| 37 |
|
| 38 |
@tool
|
| 39 |
+
def get_wikipedia_summary(query: str) -> str:
|
| 40 |
"""
|
| 41 |
A tool that fetches a Wikipedia summary for a given topic.
|
| 42 |
|
|
|
|
| 79 |
|
| 80 |
agent = CodeAgent(
|
| 81 |
model=model,
|
| 82 |
+
tools=[final_answer, get_wikipedia_summary], ## add your tools here (don't remove final answer)
|
| 83 |
max_steps=6,
|
| 84 |
verbosity_level=1,
|
| 85 |
grammar=None,
|