Update app.py

app.py

@@ -18,7 +18,7 @@ class AttachmentDownloadTool(Tool):
             "description": "task_id that needs to download attachment files."
         }
     }
-    output_type = "bytes"
+    output_type = "any"
 
     def forward(self, task_id):
         download_url = f"{DEFAULT_API_URL}/files/"
@@ -40,7 +40,7 @@ class ImageCaptionTool(Tool):
             "description": "Images that need to be identified and described"
         }
     }
-    output_type = "str"
+    output_type = "string"
 
     def setup(self):
         self.model = OpenAIServerModel(
@@ -68,7 +68,7 @@ class AudioToTextTool(Tool):
             "description": "The audio file that needs to be transcribed"
         }
     }
-    output_type = "str"
+    output_type = "string"
 
     def setup(self):
         # 使用 HuggingFace Hub 上的 Whisper 大模型
@@ -114,20 +114,20 @@ class BasicAgent:
         attachment_tool=AttachmentDownloadTool()
         image_tool=ImageCaptionTool()
         audio_tool=AudioToTextTool()
-        self.tools = [attachment_tool,image_tool,audio_tool,DuckDuckGoSearchTool(), VisitWebpageTool()]
-
-        # web_agent = ToolCallingAgent(
-        #     tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
-        #     model=self.base_model,
-        #     max_steps=10,
-        #     name="web_search_agent",
-        #     description="Runs web searches for you.",
-        # )
+        # self.tools = [attachment_tool,image_tool,audio_tool,DuckDuckGoSearchTool(), VisitWebpageTool()]
+        self.tools = [attachment_tool,image_tool,audio_tool]
+        web_agent = ToolCallingAgent(
+            tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
+            model=self.base_model,
+            max_steps=10,
+            name="web_search_agent",
+            description="Runs web searches for you.",
+        )
         
         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
         )