Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,6 @@ import pytz
|
|
| 5 |
import yaml
|
| 6 |
from tools.final_answer import FinalAnswerTool
|
| 7 |
from tools.web_search import DuckDuckGoSearchTool
|
| 8 |
-
from smolagents import tool
|
| 9 |
|
| 10 |
|
| 11 |
@tool
|
|
@@ -66,7 +65,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 66 |
prompt_templates = yaml.safe_load(stream)
|
| 67 |
|
| 68 |
# --- Tool registry ---
|
| 69 |
-
tool_registry =
|
| 70 |
DuckDuckGoSearchTool,
|
| 71 |
get_current_time_in_timezone,
|
| 72 |
])
|
|
@@ -75,7 +74,7 @@ agent_tools = [entry["tool"] for entry in tool_registry.values() if entry["enabl
|
|
| 75 |
]
|
| 76 |
# --- Agent setup ---
|
| 77 |
agent = CodeAgent(
|
| 78 |
-
tools=
|
| 79 |
model="gpt-4o-mini", # or whichever model you’re using
|
| 80 |
add_base_tools=True,
|
| 81 |
name="MyAgent",
|
|
|
|
| 5 |
import yaml
|
| 6 |
from tools.final_answer import FinalAnswerTool
|
| 7 |
from tools.web_search import DuckDuckGoSearchTool
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
@tool
|
|
|
|
| 65 |
prompt_templates = yaml.safe_load(stream)
|
| 66 |
|
| 67 |
# --- Tool registry ---
|
| 68 |
+
tool_registry = load_tool([
|
| 69 |
DuckDuckGoSearchTool,
|
| 70 |
get_current_time_in_timezone,
|
| 71 |
])
|
|
|
|
| 74 |
]
|
| 75 |
# --- Agent setup ---
|
| 76 |
agent = CodeAgent(
|
| 77 |
+
tools=tool,
|
| 78 |
model="gpt-4o-mini", # or whichever model you’re using
|
| 79 |
add_base_tools=True,
|
| 80 |
name="MyAgent",
|