DTStudios commited on
Commit
dad7940
·
verified ·
1 Parent(s): 492f9f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -4,7 +4,7 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
- from tools.web_search import
8
 
9
 
10
 
@@ -47,14 +47,14 @@ with open("prompts.yaml", 'r') as stream:
47
  prompt_templates = yaml.safe_load(stream)
48
 
49
  # registry.py
50
- from tool import calculator, get_current_time_in_timezone
51
  from tools.final_answer import FinalAnswerTool
52
  from smolagents import load_tool
53
 
54
  tool_registry = {
55
- "VisitWebPageTool": {
56
- "tool": VisitWebpageTool,
57
- "schema": VisitWebpageTool(),
58
  "enabled": True
59
  },
60
  "get_current_time_in_timezone": {
@@ -75,7 +75,7 @@ agent_tools = [entry["tool"] for entry in tool_registry.values() if entry["enabl
75
  ]
76
  agent = CodeAgent(
77
  model=model,
78
- tools=[final_answer, VisitWebpageTool, get_current_time_in_timezone],
79
  max_steps=6,
80
  verbosity_level=1,
81
  grammar=None,
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ from tools.web_search import DuckDuckGoSearchTool
8
 
9
 
10
 
 
47
  prompt_templates = yaml.safe_load(stream)
48
 
49
  # registry.py
50
+ from tool import DuckDuckGoSearchTool, get_current_time_in_timezone
51
  from tools.final_answer import FinalAnswerTool
52
  from smolagents import load_tool
53
 
54
  tool_registry = {
55
+ "DuckDuckGoSearchTool": {
56
+ "tool": DuckDuckGoSearchTool,
57
+ "schema": DuckDuckGoSearchTool(),
58
  "enabled": True
59
  },
60
  "get_current_time_in_timezone": {
 
75
  ]
76
  agent = CodeAgent(
77
  model=model,
78
+ tools=[final_answer, DuckDuckGoSearchTool, get_current_time_in_timezone],
79
  max_steps=6,
80
  verbosity_level=1,
81
  grammar=None,