Update tools.py

tools.py

@@ -210,3 +210,31 @@ def search_papers(query: str) -> str:
 
     except Exception as e:
         return f"Error fetching papers: {e}"
+
+@tool
+def download_file(task_id: str) -> str:
+    """
+    Downloads a file associated with the given task ID.
+    Returns the file path where the file is saved locally.
+
+    Args:
+        task_id: The task id to download attachment from.
+    """
+    file_url = f"{DEFAULT_API_URL}/files/{task_id}"
+    local_file_path = f"downloads/{task_id}.file"
+
+    print(f"Downloading file for task ID {task_id} from {file_url}...")
+    try:
+        response = requests.get(file_url, stream=True, timeout=15)
+        response.raise_for_status()
+
+        os.makedirs("downloads", exist_ok=True)
+        with open(local_file_path, "wb") as file:
+            for chunk in response.iter_content(chunk_size=8192):
+                file.write(chunk)
+
+        print(f"File downloaded successfully: {local_file_path}")
+        return local_file_path
+    except requests.exceptions.RequestException as e:
+        print(f"Error downloading file for task {task_id}: {e}")
+        raise