refactor: move mcp auth record ownership

src/wf_core/runtime/__init__.py

@@ -15,8 +15,8 @@ from .engine import (
     resume_workflow_async,
     resume_workflow_result_async,
 )
-from .subgraphs import PreparedSubgraph
 from .step import complete_step, step_workflow, step_workflow_async
+from .subgraphs import PreparedSubgraph
 
 __all__ = [
     "AsyncNodeHandler",

src/wf_core/runtime/engine.py

@@ -5,12 +5,12 @@ from typing import Any
 
 from wf_core.errors import WorkflowExecutionError
 from wf_core.models.workflow import Workflow
+from wf_core.run_state import ROOT_SCOPE_ID, RunState, RunStatus
 from wf_core.runtime.ops.flow import finalize_run
 from wf_core.runtime.ops.merges import ReducerDefinition
 from wf_core.runtime.ops.nodes import AsyncNodeHandler, NodeHandler
 from wf_core.runtime.ops.runs import create_run_state
 from wf_core.runtime.scheduler import resolve_no_ready_frames, select_next_frame
-from wf_core.run_state import ROOT_SCOPE_ID, RunState, RunStatus
 from wf_core.tokens import END
 
 from .preparation import prepare_new_run, prepare_resume

src/wf_core/runtime/lineage.py

@@ -8,8 +8,11 @@ from typing import Any
 from wf_core.errors import WorkflowExecutionError
 from wf_core.run_state import ExecutionFrame, LineageState, RunState, StateWrite
 from wf_core.runtime.foreach_state import ForeachBarrierState, item_frame_owner
-from wf_core.runtime.ops.state import StatePatch
-from wf_core.runtime.ops.state import commit_state_patch, safe_set_nested_value
+from wf_core.runtime.ops.state import (
+    StatePatch,
+    commit_state_patch,
+    safe_set_nested_value,
+)
 
 
 @dataclass(slots=True)

src/wf_core/runtime/ops/runs.py

@@ -5,12 +5,12 @@ from copy import deepcopy
 from wf_core.models.workflow import Workflow
 from wf_core.paths import set_nested_value
 from wf_core.run_state import (
-    ExecutionFrame,
-    FrameStatus,
-    LineageState,
     ROOT_FRAME_ID,
     ROOT_LINEAGE_ID,
     ROOT_SCOPE_ID,
+    ExecutionFrame,
+    FrameStatus,
+    LineageState,
     RunState,
     RunStatus,
     RuntimeScope,

src/wf_core/runtime/ops/schemas.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Any
 
-from jsonschema import ValidationError, SchemaError, validators
+from jsonschema import SchemaError, ValidationError, validators
 
 from wf_core.errors import WorkflowExecutionError
 from wf_core.models.schemas import SchemaRef

src/wf_core/runtime/ops/state.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 from collections.abc import Mapping, Sequence
 from copy import deepcopy
-from dataclasses import dataclass, field as dataclass_field
+from dataclasses import dataclass
+from dataclasses import field as dataclass_field
 from typing import Any
 
 from wf_core.conditions import safe_resolve_path

src/wf_core/runtime/preparation.py

@@ -5,12 +5,12 @@ from typing import Any
 
 from wf_core.errors import WorkflowExecutionError
 from wf_core.models.workflow import Workflow
+from wf_core.run_state import FrameStatus, RunState, RunStatus
 from wf_core.runtime.ops.index import WorkflowIndex, build_workflow_index
 from wf_core.runtime.ops.interrupts import resume_interrupt
 from wf_core.runtime.ops.merges import ReducerDefinition
 from wf_core.runtime.ops.schemas import validate_payload_against_schema
 from wf_core.runtime.scheduler import wake_frame
-from wf_core.run_state import FrameStatus, RunState, RunStatus
 from wf_core.tokens import END
 
 

src/wf_core/runtime/step.py

@@ -15,6 +15,8 @@ from wf_core.models.steps import (
     SubgraphNode,
 )
 from wf_core.models.workflow import Workflow
+from wf_core.run_state import ExecutionFrame, FrameStatus, RunState, StepExecutionResult
+from wf_core.runtime.foreach_state import ForeachBarrierState, item_frame_owner
 from wf_core.runtime.ops.flow import advance_frame, append_step_result_trace
 from wf_core.runtime.ops.foreach import step_foreach
 from wf_core.runtime.ops.handlers import (
@@ -32,14 +34,12 @@ from wf_core.runtime.ops.nodes import (
     finalize_pending_async_node_result,
     invoke_node_use_async_for_frame,
 )
-from wf_core.runtime.foreach_state import ForeachBarrierState, item_frame_owner
 from wf_core.runtime.scheduler import (
     ForeachIterationMetadata,
     select_next_frame,
     wake_parent_for_child_progress,
 )
 from wf_core.runtime.subgraphs import PreparedSubgraph, step_subgraph
-from wf_core.run_state import ExecutionFrame, FrameStatus, RunState, StepExecutionResult
 from wf_core.tokens import END
 
 from .preparation import prepare_step