fix: Workflow import, AI node MCP tool import type error (#4636)

Author: shaohuzhang1

apps/application/flow/tools.py

@@ -546,4 +546,8 @@ def get_tool_id_list(workflow):
             tool_ids = node_data.get('tool_ids') or []
             for _id in mcp_tool_ids + tool_ids:
                 _result.append(_id)
+        elif node.get('type') == 'mcp-node':
+            mcp_tool_id = node.get('properties', {}).get('node_data', {}).get('mcp_tool_id')
+            if mcp_tool_id:
+                _result.append(mcp_tool_id)
     return _result

apps/application/serializers/application.py

@@ -81,6 +81,10 @@ def hand_node(node, update_tool_map):
         tool_ids = node_data.get('tool_ids') or []
         node_data['tool_ids'] = [update_tool_map.get(tool_id,
                                                      tool_id) for tool_id in tool_ids]
+    if node.get('type') == 'mcp-node':
+        mcp_tool_id = (node.get('properties', {}).get('node_data', {}).get('mcp_tool_id') or '')
+        node.get('properties', {}).get('node_data', {})['mcp_tool_id'] = update_tool_map.get(mcp_tool_id,
+                                                                                             mcp_tool_id)
 
 
 class MKInstance:

apps/knowledge/serializers/knowledge_workflow.py

@@ -60,6 +60,10 @@ def hand_node(node, update_tool_map):
         tool_ids = node_data.get('tool_ids') or []
         node_data['tool_ids'] = [update_tool_map.get(tool_id,
                                                      tool_id) for tool_id in tool_ids]
+    if node.get('type') == 'mcp-node':
+        mcp_tool_id = (node.get('properties', {}).get('node_data', {}).get('mcp_tool_id') or '')
+        node.get('properties', {}).get('node_data', {})['mcp_tool_id'] = update_tool_map.get(mcp_tool_id,
+                                                                                             mcp_tool_id)
 
 
 class KnowledgeWorkflowModelSerializer(serializers.ModelSerializer):