Sborole commited on
Commit
6a3ef84
·
verified ·
1 Parent(s): 842a0b0

Update src/agent.py

Browse files
Files changed (1) hide show
  1. src/agent.py +2 -1
src/agent.py CHANGED
@@ -10,7 +10,7 @@ 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 AcademicPaperSearchTool
14
  from src.utils import InputTokenRateLimiter
15
  import wikipedia as wiki
16
  from markdownify import markdownify as to_markdown
@@ -66,6 +66,7 @@ class BasicAgent():
66
  model=self.model,
67
  tools=[
68
  GoogleSearchTool(),
 
69
  FinalAnswerTool(),
70
  PythonCalcTool(),
71
  ReadAudioTool(),
 
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
 
66
  model=self.model,
67
  tools=[
68
  GoogleSearchTool(),
69
+ GoogleAcademicSearchTool
70
  FinalAnswerTool(),
71
  PythonCalcTool(),
72
  ReadAudioTool(),