Update agent.py

agent.py

@@ -124,7 +124,7 @@ class VisitWebpageTool(Tool):
 
 class DownloadTaskAttachmentTool(Tool):
     name = "download_file"
-    description = "Downloads the file attached to the task ID and returns the local file path. Supports Excel (.xlsx), image (.png), and audio (.mp3) files."
+    description = "Downloads the file attached to the task ID and returns the local file path. Supports Excel (.xlsx), image (.png, .jpg), and audio (.mp3) files."
     inputs = {'task_id': {'type': 'string', 'description': 'The task id to download attachment from.'}}
     output_type = "string"
 
@@ -133,14 +133,14 @@ class DownloadTaskAttachmentTool(Tool):
         self.rate_limiter = rate_limiter
         self.default_api_url = default_api_url
 
-    async def forward(self, task_id: str) -> str:
+    def forward(self, task_id: str) -> str:
         file_url = f"{self.default_api_url}/files/{task_id}"
         print(f"Downloading file for task ID {task_id} from {file_url}...")
         try:
             if self.rate_limiter:
                 while not self.rate_limiter.consume(1):
                     print(f"Rate limit reached for downloading file for task {task_id}. Waiting...")
-                    await asyncio.sleep(60 / 15)  # Assuming 15 RPM
+                    time.sleep(60 / 15)  # Assuming 15 RPM
             response = requests.get(file_url, stream=True, timeout=15)
             response.raise_for_status()