Update agent.py

agent.py

@@ -90,7 +90,8 @@ class VisitWebpageTool(Tool):
         try:
             import requests
             from markdownify import markdownify
-            from requests.exceptions import RequestException
+            from requests.exceptions
+ import RequestException
 
             from smolagents.utils import truncate_content
         except ImportError as e:
@@ -126,6 +127,7 @@ class DownloadTaskAttachmentTool(Tool):
     inputs = {'task_id': {'type': 'string', 'description': 'The task id to download attachment from.'}}
     output_type = "string"
 
+    
     def forward(self, task_id: str) -> str:
         """
         Downloads a file associated with the given task ID.
@@ -136,6 +138,10 @@ class DownloadTaskAttachmentTool(Tool):
 
         print(f"Downloading file for task ID {task_id} from {file_url}...")
         try:
+
+            file_url = f"{DEFAULT_API_URL}/files/{task_id}"
+            local_path = f"downloads/{task_id}.xlsx"
+                        
             response = requests.get(file_url, stream=True, timeout=15)
             response.raise_for_status()