Spaces:
Sleeping
Sleeping
Update src/agent.py
Browse files- 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 |
-
|
| 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(),
|