fix: Add workspace ID as workflow execution parameter

Author: shaohuzhang1

apps/application/flow/i_step_node.py

@@ -79,7 +79,8 @@ def handler(self, workflow):
                                      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="知识库设置")

apps/knowledge/serializers/knowledge_workflow.py

@@ -45,6 +45,7 @@ def action(self, instance: Dict, with_valid=True):
         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()