feat(agent): add file URL support to agent query handling

Add optional `file_url` parameter to `run()` and `BasicAgent.__call__()` methods. When a task_id is available, construct and pass the file URL to the agent alongside the question text, allowing the supervisor to access attached files during processing.

.gitignore

@@ -1,3 +1,5 @@
 .env
 *.pyc
 __pycache__/
+workspace/*
+!workspace/README.md

agent/agent.py

@@ -4,6 +4,7 @@ from colorama import Fore, Style  # type: ignore[import]
 from langchain.agents import create_agent
 from langchain_core.messages import HumanMessage
 from agent.tools.math_solver import math_solver
+from agent.tools.file_downloader import file_downloader
 
 from agent.agents.websearchagents import web_search_agents
 
@@ -18,7 +19,7 @@ def supervisor_agent():
     return create_agent(
         model="google_genai:gemini-3-flash-preview",
         # tools=[math_solver, websearch_agent, web_search_agents],
-        tools=[math_solver, web_search_agents],
+        tools=[math_solver, web_search_agents, file_downloader],
         system_prompt=(
             f"You are a supervisor agent. "
             f"Current time is: {datetime.now(timezone.utc).isoformat()}. "
@@ -34,18 +35,23 @@ def supervisor_agent():
     )
 
 
-def run(query: str, max_retries: int = 3) -> str:
+def run(query: str, file_url: str | None = None, max_retries: int = 3) -> str:
     """Entry point: let the supervisor agent finish the work."""
     last_error: str | None = None
 
+    # Append file URL info to the query if available
+    full_query = query
+    if file_url:
+        full_query += f"\n\nAttached file URL: {file_url}"
+
     for attempt in range(1, max_retries + 1):
         print(
             f"{Fore.CYAN}[Supervisor] Processing query (attempt {attempt}/{max_retries})...\n"
-            f"[Supervisor] Query: {query}{Style.RESET_ALL}"
+            f"[Supervisor] Query: {full_query}{Style.RESET_ALL}"
         )
         agent = supervisor_agent()
 
-        messages = [HumanMessage(content=query)]
+        messages = [HumanMessage(content=full_query)]
         if last_error:
             messages.append(
                 HumanMessage(

agent/tools/file_downloader.py

@@ -0,0 +1,34 @@
+import os
+from pathlib import Path
+
+import httpx
+from colorama import Fore, Style  # type: ignore[import]
+from langchain_core.tools import tool
+
+WORKSPACE_DIR = Path(__file__).resolve().parents[2] / "workspace"
+
+
+@tool
+def file_downloader(url: str) -> str:
+    """Download a file from a URL and save it to the workspace directory.
+
+    Args:
+        url: The URL of the file to download.
+
+    Returns:
+        The local file path of the downloaded file, or an error message.
+    """
+    try:
+        filename = url.rstrip("/").split("/")[-1].split("?")[0] or "downloaded_file"
+        dest = WORKSPACE_DIR / filename
+
+        with httpx.stream("GET", url, follow_redirects=True, timeout=60) as r:
+            r.raise_for_status()
+            with open(dest, "wb") as f:
+                for chunk in r.iter_bytes():
+                    f.write(chunk)
+
+        print(f"{Fore.GREEN}[FileDownloader] Saved: {dest}{Style.RESET_ALL}")
+        return str(dest)
+    except Exception as e:
+        return f"Error downloading file: {e}"

app.py

@@ -16,9 +16,11 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, file_url: str | None = None) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = agent_run(question)
+        if file_url:
+            print(f"Agent received file URL: {file_url}")
+        answer = agent_run(question, file_url=file_url)
         print(f"Agent returning answer (first 50 chars): {answer[:50]}...")
         return answer
 
@@ -83,8 +85,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+        file_url = f"{api_url}/files/{task_id}"
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, file_url=file_url)
             answers_payload.append(
                 {"task_id": task_id, "submitted_answer": submitted_answer}
             )

workspace/README.md

@@ -0,0 +1 @@
+Workspace directory for downloaded files.