Skip to content

Commit 8d47aea

Browse files
authored
fix: [Workflow Tool] Missing Document Segmentation Component (#4982)
1 parent 4feae4e commit 8d47aea

File tree

2 files changed

+18
-7
lines changed

2 files changed

+18
-7
lines changed

apps/application/flow/step_node/document_split_node/i_document_split_node.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,13 @@ def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
8080
return DocumentSplitNodeSerializer
8181

8282
def _run(self):
83-
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
83+
if [WorkflowMode.KNOWLEDGE, WorkflowMode.KNOWLEDGE_LOOP, WorkflowMode.TOOL,
84+
WorkflowMode.TOOL_LOOP].__contains__(
85+
self.workflow_manage.flow.workflow_mode):
86+
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data,
87+
**{'knowledge_id': None})
88+
else:
89+
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
8490

8591
def execute(self, document_list, knowledge_id, split_strategy, paragraph_title_relate_problem_type,
8692
paragraph_title_relate_problem, paragraph_title_relate_problem_reference,

ui/src/workflow/common/data.ts

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -976,7 +976,13 @@ const toolMenuNodes = [
976976
},
977977
{
978978
label: t('views.knowledge.title'),
979-
list: [searchKnowledgeNode, searchDocumentNode, rerankerNode, documentExtractNode],
979+
list: [
980+
searchKnowledgeNode,
981+
searchDocumentNode,
982+
rerankerNode,
983+
documentExtractNode,
984+
documentSplitNode,
985+
],
980986
},
981987
{
982988
label: t('workflow.nodes.classify.businessLogic'),
@@ -1043,14 +1049,13 @@ export const toolLibNode = {
10431049
*/
10441050
export const toolWorkflowLibNode = {
10451051
type: WorkflowType.ToolWorkflowLib,
1046-
text: t('workflow.nodes.toolWorlflowNode.text','工作流工具'),
1047-
label: t('workflow.nodes.toolWorlflowNode.label','工作流工具'),
1052+
text: t('workflow.nodes.toolWorlflowNode.text', '工作流工具'),
1053+
label: t('workflow.nodes.toolWorlflowNode.label', '工作流工具'),
10481054
height: 170,
10491055
properties: {
1050-
stepName: t('workflow.nodes.toolWorlflowNode.label','工作流工具'),
1056+
stepName: t('workflow.nodes.toolWorlflowNode.label', '工作流工具'),
10511057
config: {
1052-
fields: [
1053-
],
1058+
fields: [],
10541059
},
10551060
},
10561061
}

0 commit comments

Comments
 (0)