Update app.py

app.py

@@ -18,7 +18,7 @@ class AttachmentDownloadTool(Tool):
             "description": "task_id that needs to download attachment files."
         }
     }
-    output_type = io.BytesIO
+    output_type = bytes
 
     def forward(self, task_id):
         download_url = f"{api_url}/files/"
@@ -26,12 +26,10 @@ class AttachmentDownloadTool(Tool):
         try:
             response = requests.get(download_url + task_id, stream=True, timeout=15)
             if response.status_code != 200:
-                return io.BytesIO()
-            file_obj = io.BytesIO(response.content)
-            file_obj.seek(0)
-            return file_obj
+                return None
+            return response.content
         except Exception as e:
-            return io.BytesIO()
+            return None
 
 class ImageCaptionTool(Tool):
     name = "image-captioner"
@@ -129,7 +127,7 @@ class BasicAgent:
         self.agent = CodeAgent(
             tools=self.tools,
             model=self.think_model,
-            managed_agents=[web_agent,],
+            managed_agents=[web_agent],
             additional_authorized_imports=["time", "numpy", "pandas"],
             max_steps=20
         )