SanteriVtj commited on
Commit
693cd51
·
verified ·
1 Parent(s): d024a11

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -2
agent.py CHANGED
@@ -1,5 +1,5 @@
1
  from smolagents import tool
2
- from smolagents import CodeAgent, GoogleSearchTool, InferenceClientModel, VisitWebpageTool
3
 
4
  @tool
5
  def multiply(a: int, b: int) -> int:
@@ -59,7 +59,9 @@ def return_agent(question):
59
  add,
60
  subtract,
61
  divide,
62
- modulus
 
 
63
  ],
64
  model=InferenceClientModel()
65
  )
 
1
  from smolagents import tool
2
+ from smolagents import CodeAgent, InferenceClientModel, VisitWebpageTool, DuckDuckGoSearchTool
3
 
4
  @tool
5
  def multiply(a: int, b: int) -> int:
 
59
  add,
60
  subtract,
61
  divide,
62
+ modulus,
63
+ DuckDuckGoSearchTool,
64
+ VisitWebpageTool
65
  ],
66
  model=InferenceClientModel()
67
  )