DTStudios commited on
Commit
9d5e193
·
verified ·
1 Parent(s): 7bd93b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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 = ToolRegistry([
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=tool_registry,
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",