DTStudios commited on
Commit
5fb6ced
·
verified ·
1 Parent(s): 582dd84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool,tool_registry,HfApiModel,load_tool,tool
2
  import datetime
3
  from duckduckgo_search import DDGS
4
  import requests
@@ -65,7 +65,7 @@ with open("prompts.yaml", 'r') as stream:
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,7 +74,7 @@ agent_tools = [entry["tool"] for entry in tool_registry.values() if entry["enabl
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",
 
1
+ from smolagents import CodeAgent,DuckDuckGoSearchTool,ToolRegistry,HfApiModel,load_tool,tool
2
  import datetime
3
  from duckduckgo_search import DDGS
4
  import requests
 
65
  prompt_templates = yaml.safe_load(stream)
66
 
67
  # --- Tool registry ---
68
+ tool_registry = ToolRegistry([
69
  DuckDuckGoSearchTool,
70
  get_current_time_in_timezone,
71
  ])
 
74
  ]
75
  # --- Agent setup ---
76
  agent = CodeAgent(
77
+ tools=ToolRegistry,
78
  model="gpt-4o-mini", # or whichever model you’re using
79
  add_base_tools=True,
80
  name="MyAgent",