From 10372c89098709a3f7bc985bf8c800f40edbffdb Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Mon, 24 Nov 2025 17:26:35 +0800 Subject: [PATCH] fix: Add workspace ID as workflow execution parameter --- apps/application/flow/i_step_node.py | 4 +++- apps/knowledge/serializers/knowledge_workflow.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/application/flow/i_step_node.py b/apps/application/flow/i_step_node.py index cfb656674..e4c4374b1 100644 --- a/apps/application/flow/i_step_node.py +++ b/apps/application/flow/i_step_node.py @@ -79,7 +79,8 @@ class WorkFlowPostHandler: message_tokens=message_tokens, answer_tokens=answer_tokens, answer_text_list=answer_text_list, - run_time=time.time() - workflow.context.get('start_time') if workflow.context.get('start_time') is not None else 0, + run_time=time.time() - workflow.context.get('start_time') if workflow.context.get( + 'start_time') is not None else 0, index=0) self.chat_info.append_chat_record(chat_record) @@ -166,6 +167,7 @@ class FlowParamsSerializer(serializers.Serializer): class KnowledgeFlowParamsSerializer(serializers.Serializer): knowledge_id = serializers.UUIDField(required=True, label="知识库id") + workspace_id = serializers.CharField(required=True, label="工作空间id") knowledge_action_id = serializers.UUIDField(required=True, label="知识库任务执行器id") data_source = serializers.DictField(required=True, label="数据源") knowledge_base = serializers.DictField(required=False, label="知识库设置") diff --git a/apps/knowledge/serializers/knowledge_workflow.py b/apps/knowledge/serializers/knowledge_workflow.py index 09e3f94cf..53159ab77 100644 --- a/apps/knowledge/serializers/knowledge_workflow.py +++ b/apps/knowledge/serializers/knowledge_workflow.py @@ -45,6 +45,7 @@ class KnowledgeWorkflowActionSerializer(serializers.Serializer): work_flow_manage = KnowledgeWorkflowManage( Workflow.new_instance(knowledge_workflow.work_flow, WorkflowMode.KNOWLEDGE), {'knowledge_id': self.data.get("knowledge_id"), 'knowledge_action_id': knowledge_action_id, 'stream': True, + 'workspace_id': self.data.get("workspace_id"), **instance}, KnowledgeWorkflowPostHandler(None, knowledge_action_id)) work_flow_manage.run()