From 2db269da95cc67091cb1dd535912a4a5da64bafc Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Wed, 4 Jun 2025 11:50:15 +0800 Subject: [PATCH] chore(api): Rename import for `NodeRunMetadataKey`. `NodeRunMetadataKey` has been renamed to `WorkflowNodeExecutionMetadataKey`. --- api/core/workflow/workflow_entry.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/core/workflow/workflow_entry.py b/api/core/workflow/workflow_entry.py index 7133fd0c77..5c85d32f8e 100644 --- a/api/core/workflow/workflow_entry.py +++ b/api/core/workflow/workflow_entry.py @@ -10,7 +10,7 @@ from core.app.entities.app_invoke_entities import InvokeFrom from core.file.models import File from core.workflow.callbacks import WorkflowCallback from core.workflow.constants import ENVIRONMENT_VARIABLE_NODE_ID -from core.workflow.entities.node_entities import NodeRunMetadataKey +from core.workflow.entities.node_entities import WorkflowNodeExecutionMetadataKey from core.workflow.entities.variable_pool import VariablePool from core.workflow.errors import WorkflowNodeRunFailedError from core.workflow.graph_engine.entities.event import GraphEngineEvent, GraphRunFailedEvent, InNodeEvent @@ -390,7 +390,7 @@ _NodeOrInNodeEvent: TypeAlias = NodeEvent | InNodeEvent def _attach_execution_metadata( - extra_metadata: dict[NodeRunMetadataKey, Any], + extra_metadata: dict[WorkflowNodeExecutionMetadataKey, Any], ) -> Callable[[_NodeOrInNodeEvent], _NodeOrInNodeEvent]: def _execution_metadata_mapper(e: NodeEvent | InNodeEvent) -> NodeEvent | InNodeEvent: if not isinstance(e, RunCompletedEvent): @@ -416,7 +416,7 @@ def _attach_execution_metadata_based_on_node_config( raise Exception("invalid graph") return _attach_execution_metadata( { - NodeRunMetadataKey.LOOP_ID: loop_id, + WorkflowNodeExecutionMetadataKey.LOOP_ID: loop_id, } ) elif in_iteration: @@ -425,7 +425,7 @@ def _attach_execution_metadata_based_on_node_config( raise Exception("invalid graph") return _attach_execution_metadata( { - NodeRunMetadataKey.ITERATION_ID: iteration_id, + WorkflowNodeExecutionMetadataKey.ITERATION_ID: iteration_id, } ) else: