Sborole commited on
Commit
0e64b2b
·
verified ·
1 Parent(s): 9598d95

Update src/agent.py

Browse files
Files changed (1) hide show
  1. src/agent.py +3 -4
src/agent.py CHANGED
@@ -7,7 +7,7 @@ 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 YouTubeTool
11
  from tools.PythonRunnerTool import PythonRunnerTool
12
  from tools.PythonCalcTool import PythonCalcTool
13
  from tools.SemanticScholar import AcademicPaperSearchTool
@@ -68,17 +68,16 @@ class BasicAgent():
68
  GoogleSearchTool(),
69
  FinalAnswerTool(),
70
  PythonCalcTool(),
71
- AcademicPaperSearchTool(),
72
  ReadAudioTool(),
73
  ReadImageTool(),
74
  ReadTextTool(),
75
- ReadVideoTool(),
76
- YouTubeTool(),
77
  PythonRunnerTool(),
78
  MathSolver(),
79
  LocalWikipediaTool(),
80
  WikiTitleFinder(),
81
  WikiContentFetcher(),
 
82
  ],
83
  max_steps=10,
84
  planning_interval=5,
 
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 AcademicPaperSearchTool
 
68
  GoogleSearchTool(),
69
  FinalAnswerTool(),
70
  PythonCalcTool(),
 
71
  ReadAudioTool(),
72
  ReadImageTool(),
73
  ReadTextTool(),
74
+ YouTubeSearchTool(),
 
75
  PythonRunnerTool(),
76
  MathSolver(),
77
  LocalWikipediaTool(),
78
  WikiTitleFinder(),
79
  WikiContentFetcher(),
80
+ AcademicPaperSearchTool(),
81
  ],
82
  max_steps=10,
83
  planning_interval=5,