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