From 80372b954c95a295a458ae3d9d9c13d00049ec46 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Mon, 30 Mar 2026 18:43:46 +0800
Subject: [PATCH] fix: [Workflow Tool] Missing Document Segmentation Component

---
 .../i_document_split_node.py                    |  8 +++++++-
 ui/src/workflow/common/data.ts                  | 17 +++++++++++------
 2 files changed, 18 insertions(+), 7 deletions(-)

diff --git a/apps/application/flow/step_node/document_split_node/i_document_split_node.py b/apps/application/flow/step_node/document_split_node/i_document_split_node.py
index e4a1c809fb1..eae15861ca7 100644
--- a/apps/application/flow/step_node/document_split_node/i_document_split_node.py
+++ b/apps/application/flow/step_node/document_split_node/i_document_split_node.py
@@ -80,7 +80,13 @@ def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
         return DocumentSplitNodeSerializer
 
     def _run(self):
-        return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
+        if [WorkflowMode.KNOWLEDGE, WorkflowMode.KNOWLEDGE_LOOP, WorkflowMode.TOOL,
+            WorkflowMode.TOOL_LOOP].__contains__(
+            self.workflow_manage.flow.workflow_mode):
+            return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data,
+                                **{'knowledge_id': None})
+        else:
+            return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
 
     def execute(self, document_list, knowledge_id, split_strategy, paragraph_title_relate_problem_type,
                 paragraph_title_relate_problem, paragraph_title_relate_problem_reference,
diff --git a/ui/src/workflow/common/data.ts b/ui/src/workflow/common/data.ts
index e44a7cb24a0..46ac1fb1b83 100644
--- a/ui/src/workflow/common/data.ts
+++ b/ui/src/workflow/common/data.ts
@@ -976,7 +976,13 @@ const toolMenuNodes = [
   },
   {
     label: t('views.knowledge.title'),
-    list: [searchKnowledgeNode, searchDocumentNode, rerankerNode, documentExtractNode],
+    list: [
+      searchKnowledgeNode,
+      searchDocumentNode,
+      rerankerNode,
+      documentExtractNode,
+      documentSplitNode,
+    ],
   },
   {
     label: t('workflow.nodes.classify.businessLogic'),
@@ -1043,14 +1049,13 @@ export const toolLibNode = {
  */
 export const toolWorkflowLibNode = {
   type: WorkflowType.ToolWorkflowLib,
-  text: t('workflow.nodes.toolWorlflowNode.text','工作流工具'),
-  label: t('workflow.nodes.toolWorlflowNode.label','工作流工具'),
+  text: t('workflow.nodes.toolWorlflowNode.text', '工作流工具'),
+  label: t('workflow.nodes.toolWorlflowNode.label', '工作流工具'),
   height: 170,
   properties: {
-    stepName: t('workflow.nodes.toolWorlflowNode.label','工作流工具'),
+    stepName: t('workflow.nodes.toolWorlflowNode.label', '工作流工具'),
     config: {
-      fields: [
-      ],
+      fields: [],
     },
   },
 }