🐛 Repairing Multimodal Tools Cannot Export

Author: Phinease

backend/services/agent_service.py

@@ -844,7 +844,7 @@ async def export_agent_by_agent_id(agent_id: int, tenant_id: str, user_id: str)
 
     # Check if any tool is KnowledgeBaseSearchTool and set its metadata to empty dict
     for tool in tool_list:
-        if tool.class_name == "KnowledgeBaseSearchTool":
+        if tool.class_name in ["KnowledgeBaseSearchTool", "AnalyzeTextFileTool", "AnalyzeImageTool"]:
             tool.metadata = {}
 
     # Get model_id and model display name from agent_info

test/backend/services/test_agent_service.py

@@ -1781,6 +1781,28 @@ async def test_export_agent_by_agent_id_success(mock_search_agent_info, mock_cre
             output_type="search results",
             usage=None
         ),
+        ToolConfig(
+            class_name="AnalyzeTextFileTool",
+            name="Text Analyzer",
+            source="source3",
+            params={"param4": "value4"},
+            metadata={"text": "data"},
+            description="Text analysis tool",
+            inputs="text file",
+            output_type="analysis",
+            usage=None
+        ),
+        ToolConfig(
+            class_name="AnalyzeImageTool",
+            name="Image Analyzer",
+            source="source4",
+            params={"param5": "value5"},
+            metadata={"image": "data"},
+            description="Image analysis tool",
+            inputs="image file",
+            output_type="analysis result",
+            usage=None
+        ),
         ToolConfig(
             class_name="MCPTool",
             name="MCP Tool",
@@ -1810,14 +1832,22 @@ async def test_export_agent_by_agent_id_success(mock_search_agent_info, mock_cre
     assert result.agent_id == 123
     assert result.name == "Test Agent"
     assert result.business_description == "For testing purposes"
-    assert len(result.tools) == 3
+    assert len(result.tools) == 5
     assert result.managed_agents == mock_sub_agent_ids
 
     # Verify KnowledgeBaseSearchTool metadata is empty
     knowledge_tool = next(
         tool for tool in result.tools if tool.class_name == "KnowledgeBaseSearchTool")
     assert knowledge_tool.metadata == {}
 
+    analyze_text_tool = next(
+        tool for tool in result.tools if tool.class_name == "AnalyzeTextFileTool")
+    assert analyze_text_tool.metadata == {}
+
+    analyze_image_tool = next(
+        tool for tool in result.tools if tool.class_name == "AnalyzeImageTool")
+    assert analyze_image_tool.metadata == {}
+
     # Verify MCP tool has usage field
     mcp_tool = next(
         tool for tool in result.tools if tool.class_name == "MCPTool")