Sborole commited on
Commit
8b130cd
·
verified ·
1 Parent(s): eb75560

Update src/agent.py

Browse files
Files changed (1) hide show
  1. src/agent.py +4 -4
src/agent.py CHANGED
@@ -5,12 +5,12 @@ from tools.ReadAudioTool import ReadAudioTool
5
  from tools.ReadImageTool import ReadImageTool
6
  from tools.ReadTextTool import ReadTextTool
7
  from tools.ReadVideoTool import ReadVideoTool
8
- from tools.WebSearchTool import GoogleSearchTool
9
  from tools.WikipediaTool import LocalWikipediaTool
10
  from tools.YouTubeTool import YouTubeSearchTool
11
  from tools.PythonRunnerTool import PythonRunnerTool
12
  from tools.PythonCalcTool import PythonCalcTool
13
- from tools.SemanticScholar import GoogleAcademicSearchTool
14
  from src.utils import InputTokenRateLimiter
15
  import wikipedia as wiki
16
  from markdownify import markdownify as to_markdown
@@ -65,8 +65,8 @@ class BasicAgent():
65
  self.agent = CodeAgent(
66
  model=self.model,
67
  tools=[
68
- GoogleSearchTool(),
69
- GoogleAcademicSearchTool(),
70
  FinalAnswerTool(),
71
  PythonCalcTool(),
72
  ReadAudioTool(),
 
5
  from tools.ReadImageTool import ReadImageTool
6
  from tools.ReadTextTool import ReadTextTool
7
  from tools.ReadVideoTool import ReadVideoTool
8
+ from tools.WebSearchTool import TavilySearchTool
9
  from tools.WikipediaTool import LocalWikipediaTool
10
  from tools.YouTubeTool import YouTubeSearchTool
11
  from tools.PythonRunnerTool import PythonRunnerTool
12
  from tools.PythonCalcTool import PythonCalcTool
13
+ from tools.SemanticScholar import TavilyResearchTool
14
  from src.utils import InputTokenRateLimiter
15
  import wikipedia as wiki
16
  from markdownify import markdownify as to_markdown
 
65
  self.agent = CodeAgent(
66
  model=self.model,
67
  tools=[
68
+ TavilySearchTool(),
69
+ TavilyResearchTool(),
70
  FinalAnswerTool(),
71
  PythonCalcTool(),
72
  ReadAudioTool(),