Fixed download_file tool.

agent_config.py

@@ -6,6 +6,7 @@ import requests
 import re
 import tempfile
 from pathlib import Path
+from typing import Optional
 
 @tool
 def check_answer(answer: str) -> str:
@@ -34,37 +35,41 @@ def reverse_sentence_tool(reverse_sentence: str) -> str:
     return " ".join(correct_words)
 
 @tool
-def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
+def download_file_if_any(base_api_url: str, task_id: str) -> Optional[str]:
     """
-    Try GET /files/{task_id}.
-    • On HTTP 200 → save to a temp dir and return local path.
-    • On 404 → return None.
-    • On other errors → raise so caller can log / handle.
+    Downloads a file from a remote API using a task ID.
+
+    Args:
+        base_api_url (str): Base URL of the API to fetch files from (e.g., "https://example.com/api").
+        task_id (str): Task identifier used to locate the file.
+
+    Returns:
+        Optional[str]: Path to the downloaded file, or None if the file is not found (404).
     """
     url = f"{base_api_url}/files/{task_id}"
     try:
         resp = requests.get(url, timeout=30)
         if resp.status_code == 404:
-            return None          # no file
-        resp.raise_for_status()   # raise on 4xx/5xx ≠ 404
+            return None  # file not found
+        resp.raise_for_status()  # raise other errors
     except requests.exceptions.HTTPError as e:
-        # propagate non-404 errors (403, 500, …)
-        raise e
+        raise e  # propagate for logging/debugging
 
-    # ▸ Save bytes to a named file inside the system temp dir
-    #    Try to keep original extension from Content-Disposition if present.
+    # Extract filename from Content-Disposition if present
     cdisp = resp.headers.get("content-disposition", "")
-    filename = task_id                                 # default base name
+    filename = task_id  # fallback
     if "filename=" in cdisp:
-        m = re.search(r'filename="([^"]+)"', cdisp)
-        if m:
-            filename = m.group(1)                      # keep provided name
+        match = re.search(r'filename="([^"]+)"', cdisp)
+        if match:
+            filename = match.group(1)
 
     tmp_dir = Path(tempfile.gettempdir()) / "gaia_files"
-    tmp_dir.mkdir(exist_ok=True)
+    tmp_dir.mkdir(parents=True, exist_ok=True)
     file_path = tmp_dir / filename
+
     with open(file_path, "wb") as f:
         f.write(resp.content)
+
     return str(file_path)
 
 def create_agent():