🚧 Move GaiaAgent to a module

app.py

@@ -3,11 +3,8 @@ import os
 import gradio as gr
 import pandas as pd
 import requests
+from gaiaagent import GaiaAgent
 from smolagents import (
-    CodeAgent,
-    Tool,
-    Model,
-    PromptTemplates,
     FinalAnswerTool,
     HfApiModel,
     DuckDuckGoSearchTool,
@@ -33,57 +30,6 @@ model = HfApiModel(
 )
 
 
-class GaiaAgent(CodeAgent):
-    def __init__(
-        self,
-        api_url: str,
-        tools: list[Tool],
-        model: Model,
-        prompt_templates: PromptTemplates | None = None,
-        additional_authorized_imports: list[str] | None = None,
-        planning_interval: int | None = None,
-        executor_type: str | None = "local",
-        executor_kwargs: dict[str, Any] | None = None,
-        max_print_outputs_length: int | None = None,
-        stream_outputs: bool = False,
-        use_structured_outputs_internally: bool = False,
-        grammar: dict[str, str] | None = None,
-        **kwargs,
-    ):
-        super().__init__(
-            tools=tools,
-            model=model,
-            prompt_templates=prompt_templates,
-            additional_authorized_imports=additional_authorized_imports,
-            planning_interval=planning_interval,
-            executor_type=executor_type,
-            executor_kwargs=executor_kwargs,
-            max_print_outputs_length=max_print_outputs_length,
-            use_structured_outputs_internally=use_structured_outputs_internally,
-            grammar=grammar,
-            **kwargs,
-        )
-        self._files_base_url = f"{api_url}/files"
-
-    def __call__(self, item: dict) -> str:
-        question = item["question"]
-        task_id = item["task_id"]
-        file_name = item["file_name"]
-        if file_name:
-            url = f"{self._files_base_url}/{task_id}"
-            resp = requests.get(url, timeout=30)
-            resp.raise_for_status()
-            content = resp.content
-            return super().run(
-                question,
-                additional_args={
-                    "file_content": content,
-                    "file_name": file_name,
-                },
-            )
-        return super().run(question)
-
-
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,

gaiaagent.py

@@ -0,0 +1,43 @@
+from typing import Any
+from smolagents import CodeAgent, PromptTemplates, Model, Tool
+
+
+class GaiaAgent(CodeAgent):
+    def __init__(
+        self,
+        tools: list[Tool],
+        model: Model,
+        prompt_templates: PromptTemplates | None = None,
+        additional_authorized_imports: list[str] | None = None,
+        planning_interval: int | None = None,
+        executor_type: str | None = "local",
+        executor_kwargs: dict[str, Any] | None = None,
+        max_print_outputs_length: int | None = None,
+        stream_outputs: bool = False,
+        use_structured_outputs_internally: bool = False,
+        grammar: dict[str, str] | None = None,
+        **kwargs,
+    ):
+        super().__init__(
+            tools=tools,
+            model=model,
+            prompt_templates=prompt_templates,
+            additional_authorized_imports=additional_authorized_imports,
+            planning_interval=planning_interval,
+            executor_type=executor_type,
+            executor_kwargs=executor_kwargs,
+            max_print_outputs_length=max_print_outputs_length,
+            use_structured_outputs_internally=use_structured_outputs_internally,
+            grammar=grammar,
+            **kwargs,
+        )
+
+    def __call__(self, question: str, filename: str) -> str:
+        if filename:
+            return super().run(
+                question,
+                additional_args={
+                    "filename": filename,
+                },
+            )
+        return super().run(question)