From c70d0546aedc4ee31612a59a70bd3b05fd8be875 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Wed, 27 Mar 2024 16:00:54 +0800 Subject: [PATCH] sign single step files --- api/core/workflow/workflow_engine_manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/core/workflow/workflow_engine_manager.py b/api/core/workflow/workflow_engine_manager.py index a9fa646bb5..8a9d3fc36f 100644 --- a/api/core/workflow/workflow_engine_manager.py +++ b/api/core/workflow/workflow_engine_manager.py @@ -4,6 +4,7 @@ from typing import Optional from core.app.apps.base_app_queue_manager import GenerateTaskStoppedException from core.file.file_obj import FileVar +from core.tools.tool_file_manager import ToolFileManager from core.workflow.callbacks.base_workflow_callback import BaseWorkflowCallback from core.workflow.entities.node_entities import NodeRunMetadataKey, NodeRunResult, NodeType from core.workflow.entities.variable_pool import VariablePool, VariableValue @@ -258,11 +259,13 @@ class WorkflowEngineManager: variable_key_list=variable_key_list, value=user_inputs.get(variable_key) ) - # run node node_run_result = node_instance.run( variable_pool=variable_pool ) + + # sign output files + node_run_result.outputs = self.handle_special_values(node_run_result.outputs) except Exception as e: raise WorkflowNodeRunFailedError( node_id=node_instance.node_id,