Update src/agent.py

src/agent.py

@@ -1,6 +1,11 @@
 from src.settings import Settings
 from smolagents import LiteLLMModel, ToolCallingAgent
-from src.FinalAnswerTool import FinalAnswerTool
+from tools.FinalAnswerTool import FinalAnswerTool
+from tools.ReadAudioTool import ReadAudioTool
+from tools.ReadImageTool import ReadImageTool
+from tools.ReadTextTool import ReadTextTool
+from tools.ReadVideoTool import ReadVideoTool
+from tools.WebSearchTool import WebSearchTool
 from src.utils import InputTokenRateLimiter
 import time
 import random
@@ -16,6 +21,11 @@ class BasicAgent():
         self.agent = ToolCallingAgent(
             tools=[
                 FinalAnswerTool(),
+                ReadAudioTool(),
+                ReadImageTool(),
+                ReadTextTool(),
+                ReadVideoTool(),
+                WebSearchTool())
                 # TODO: MP3 interpretation
             ],
             max_steps=10,