fix: Workflow import, AI node MCP tool import type error

Author: shaohuzhang1

apps/application/serializers/application.py

@@ -31,7 +31,7 @@
 
 from application.flow.common import Workflow
 from application.models.application import Application, ApplicationTypeChoices, \
-    ApplicationFolder, ApplicationVersion
+    ApplicationFolder, ApplicationVersion, ApplicationKnowledgeMapping
 from application.models.application_access_token import ApplicationAccessToken
 from application.serializers.common import update_resource_mapping_by_application
 from common import result
@@ -641,6 +641,7 @@ def to_tool(tool, workspace_id, user_id):
                     input_field_list=tool.get('input_field_list'),
                     init_field_list=tool.get('init_field_list'),
                     is_active=False if len((tool.get('init_field_list') or [])) > 0 else tool.get('is_active'),
+                    tool_type=tool.get('tool_type', 'CUSTOM') or 'CUSTOM',
                     scope=ToolScope.WORKSPACE,
                     folder_id=workspace_id,
                     workspace_id=workspace_id)
@@ -790,6 +791,7 @@ def delete(self, with_valid=True):
         QuerySet(ResourceMapping).filter(
             Q(target_id=application_id) | Q(source_id=application_id)
         ).delete()
+        QuerySet(ApplicationKnowledgeMapping).filter(application_id=application_id).delete()
         QuerySet(Application).filter(id=application_id).delete()
         return True
 

apps/knowledge/serializers/knowledge_workflow.py

@@ -370,6 +370,7 @@ def to_tool(tool, workspace_id, user_id):
                         input_field_list=tool.get('input_field_list'),
                         init_field_list=tool.get('init_field_list'),
                         is_active=False if len((tool.get('init_field_list') or [])) > 0 else tool.get('is_active'),
+                        tool_type=tool.get('tool_type', 'CUSTOM') or 'CUSTOM',
                         scope=ToolScope.SHARED if workspace_id == 'None' else ToolScope.WORKSPACE,
                         folder_id='default' if workspace_id == 'None' else workspace_id,
                         workspace_id=workspace_id)