Upload agent.py

agent.py

@@ -21,9 +21,14 @@ from smolagents import (
     Tool,
 )
 
-# Custom Tools
-from .tools import (PythonRunTool, ExcelLoaderTool, YouTubeTranscriptTool,
-                    AudioTranscriptionTool, SimpleOCRTool)
+# Custom Tools from tools.py
+from .tools import (
+    PythonRunTool,
+    ExcelLoaderTool,
+    YouTubeTranscriptTool,
+    AudioTranscriptionTool,
+    SimpleOCRTool,
+)
 
 # ---------------------------------------------------------------------------
 # Model selection helper
@@ -31,7 +36,6 @@ from .tools import (PythonRunTool, ExcelLoaderTool, YouTubeTranscriptTool,
 
 load_dotenv()  # Make sure we read credentials from .env when running locally
 
-
 def _select_model():
     """Return a smolagents *model* as configured by the ``MODEL_PROVIDER`` env."""
 
@@ -88,14 +92,10 @@ class GAIAAgent(CodeAgent):
 # ---------------------------------------------------------------------------
 
 def gaia_agent(*, extra_tools: Sequence[Tool] | None = None) -> GAIAAgent:
-    base_tools = [
-        DuckDuckGoSearchTool(),
-        CustomTool1(),
-        CustomTool2(),
-    ]
+    # Compose the toolset: always include all default tools, plus any extras
+    toolset = list(DEFAULT_TOOLS)
     if extra_tools:
-        base_tools.extend(extra_tools)
-    return GAIAAgent(tools=base_tools)
-
+        toolset.extend(extra_tools)
+    return GAIAAgent(tools=toolset)
 
 __all__ = ["GAIAAgent", "gaia_agent"]