chore(api): Rename import for `NodeRunMetadataKey`.

`NodeRunMetadataKey` has been renamed to
`WorkflowNodeExecutionMetadataKey`.
pull/20699/head
QuantumGhost 12 months ago
parent ffa9bc8998
commit 2db269da95

@ -10,7 +10,7 @@ from core.app.entities.app_invoke_entities import InvokeFrom
from core.file.models import File from core.file.models import File
from core.workflow.callbacks import WorkflowCallback from core.workflow.callbacks import WorkflowCallback
from core.workflow.constants import ENVIRONMENT_VARIABLE_NODE_ID 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.entities.variable_pool import VariablePool
from core.workflow.errors import WorkflowNodeRunFailedError from core.workflow.errors import WorkflowNodeRunFailedError
from core.workflow.graph_engine.entities.event import GraphEngineEvent, GraphRunFailedEvent, InNodeEvent from core.workflow.graph_engine.entities.event import GraphEngineEvent, GraphRunFailedEvent, InNodeEvent
@ -390,7 +390,7 @@ _NodeOrInNodeEvent: TypeAlias = NodeEvent | InNodeEvent
def _attach_execution_metadata( def _attach_execution_metadata(
extra_metadata: dict[NodeRunMetadataKey, Any], extra_metadata: dict[WorkflowNodeExecutionMetadataKey, Any],
) -> Callable[[_NodeOrInNodeEvent], _NodeOrInNodeEvent]: ) -> Callable[[_NodeOrInNodeEvent], _NodeOrInNodeEvent]:
def _execution_metadata_mapper(e: NodeEvent | InNodeEvent) -> NodeEvent | InNodeEvent: def _execution_metadata_mapper(e: NodeEvent | InNodeEvent) -> NodeEvent | InNodeEvent:
if not isinstance(e, RunCompletedEvent): if not isinstance(e, RunCompletedEvent):
@ -416,7 +416,7 @@ def _attach_execution_metadata_based_on_node_config(
raise Exception("invalid graph") raise Exception("invalid graph")
return _attach_execution_metadata( return _attach_execution_metadata(
{ {
NodeRunMetadataKey.LOOP_ID: loop_id, WorkflowNodeExecutionMetadataKey.LOOP_ID: loop_id,
} }
) )
elif in_iteration: elif in_iteration:
@ -425,7 +425,7 @@ def _attach_execution_metadata_based_on_node_config(
raise Exception("invalid graph") raise Exception("invalid graph")
return _attach_execution_metadata( return _attach_execution_metadata(
{ {
NodeRunMetadataKey.ITERATION_ID: iteration_id, WorkflowNodeExecutionMetadataKey.ITERATION_ID: iteration_id,
} }
) )
else: else:

Loading…
Cancel
Save