k0smes commited on
Commit
4e0245a
·
verified ·
1 Parent(s): e195492

add ddg_tool

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import datetime
3
  import requests
4
  import pytz
@@ -8,7 +8,7 @@ from tools.final_answer import FinalAnswerTool
8
  from Gradio_UI import GradioUI
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
- # ddg_tool = DuckDuckGoSearchTool()
12
 
13
  # @tool
14
  # def web_search(query: str) -> str:
@@ -59,7 +59,11 @@ with open("prompts.yaml", 'r') as stream:
59
 
60
  agent = CodeAgent(
61
  model=model,
62
- tools=[final_answer, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
 
 
 
 
63
  max_steps=6,
64
  verbosity_level=1,
65
  grammar=None,
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
2
  import datetime
3
  import requests
4
  import pytz
 
8
  from Gradio_UI import GradioUI
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
+ ddg_tool = DuckDuckGoSearchTool()
12
 
13
  # @tool
14
  # def web_search(query: str) -> str:
 
59
 
60
  agent = CodeAgent(
61
  model=model,
62
+ tools=[
63
+ final_answer,
64
+ get_current_time_in_timezone,
65
+ ddg_tool
66
+ ], ## add your tools here (don't remove final answer)
67
  max_steps=6,
68
  verbosity_level=1,
69
  grammar=None,