15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-30 01:32:49 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: Add workspace ID as workflow execution parameter

This commit is contained in:
shaohuzhang1 2025-11-24 17:26:35 +08:00
parent a3c0cc1115
commit 10372c8909
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/flow/i_step_node.py
View File

@ -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="知识库设置")

1
apps/knowledge/serializers/knowledge_workflow.py
View File

@ -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()