Update src/agent.py

src/agent.py

@@ -6,6 +6,9 @@ from tools.ReadImageTool import ReadImageTool
 from tools.ReadTextTool import ReadTextTool
 from tools.ReadVideoTool import ReadVideoTool
 from tools.WebSearchTool import WebSearchTool
+from tools.YouTubeTool import YouTubeTool
+from tools.PythonRunnerTool import PythonRunnerTool
+from tools.WikipediaTool import WikipediaTool
 from src.utils import InputTokenRateLimiter
 import time
 import random
@@ -25,11 +28,14 @@ class BasicAgent():
                 ReadImageTool(),
                 ReadTextTool(),
                 ReadVideoTool(),
-                WebSearchTool()
+                WebSearchTool(),
+                WikipediaTool(),
+                YouTubeTool(),
+                PythonRunnerTool()
                 # TODO: MP3 interpretation
             ],
-            max_steps=1,
-            planning_interval=1,
+            max_steps=3,
+            planning_interval=3,
             model=self.model
         )
         self.token_rate_limiter = InputTokenRateLimiter()