Sborole commited on
Commit
8df8dbc
·
verified ·
1 Parent(s): 6bb6cd7

Update src/agent.py

Browse files
Files changed (1) hide show
  1. src/agent.py +4 -2
src/agent.py CHANGED
@@ -6,6 +6,7 @@ from tools.ReadImageTool import ReadImageTool
6
  from tools.ReadTextTool import ReadTextTool
7
  from tools.ReadVideoTool import ReadVideoTool
8
  from tools.WebSearchTool import SerpApiSearchTool
 
9
  from tools.YouTubeTool import YouTubeTool
10
  from tools.PythonRunnerTool import PythonRunnerTool
11
  from tools.PythonCalcTool import PythonCalcTool
@@ -69,9 +70,10 @@ class BasicAgent():
69
  ReadImageTool(),
70
  ReadTextTool(),
71
  ReadVideoTool(),
72
- SerpAPISearchTool(),
 
73
  WikiTitleFinder(),
74
- WikiContentFetcher(),
75
  YouTubeTool(),
76
  PythonRunnerTool(),
77
  PythonCalcTool(),
 
6
  from tools.ReadTextTool import ReadTextTool
7
  from tools.ReadVideoTool import ReadVideoTool
8
  from tools.WebSearchTool import SerpApiSearchTool
9
+ from tools.WikipediaTool import LocalWikipediaTool
10
  from tools.YouTubeTool import YouTubeTool
11
  from tools.PythonRunnerTool import PythonRunnerTool
12
  from tools.PythonCalcTool import PythonCalcTool
 
70
  ReadImageTool(),
71
  ReadTextTool(),
72
  ReadVideoTool(),
73
+ SerpApiSearchTool(),
74
+ LocalWikipediaTool(),
75
  WikiTitleFinder(),
76
+ WikiContentFetcher(),
77
  YouTubeTool(),
78
  PythonRunnerTool(),
79
  PythonCalcTool(),