JPM34 commited on
Commit
d68b105
·
1 Parent(s): eac1fe8

Update tools: Replaced DuckDuckGo by Google

Browse files
Files changed (1) hide show
  1. agent.py +4 -4
agent.py CHANGED
@@ -3,8 +3,8 @@ import os
3
  from smolagents import (
4
  CodeAgent,
5
  PythonInterpreterTool,
6
- DuckDuckGoSearchTool,
7
- WikipediaSearchTool,
8
  )
9
  from models import select_model
10
 
@@ -28,9 +28,9 @@ class BasicAgent:
28
  ):
29
  self.verbose = verbose
30
  tools = [
31
- DuckDuckGoSearchTool(),
32
  PythonInterpreterTool(),
33
- WikipediaSearchTool(),
34
  save_and_read_file,
35
  analyze_excel_file,
36
  analyze_csv_file,
 
3
  from smolagents import (
4
  CodeAgent,
5
  PythonInterpreterTool,
6
+ GoogleSearchTool,
7
+ VisitWebpageTool,
8
  )
9
  from models import select_model
10
 
 
28
  ):
29
  self.verbose = verbose
30
  tools = [
31
+ GoogleSearchTool(provider="serper"),
32
  PythonInterpreterTool(),
33
+ VisitWebpageTool(),
34
  save_and_read_file,
35
  analyze_excel_file,
36
  analyze_csv_file,