refactor: move mcp auth record ownership

src/wf_api/__init__.py

@@ -1,7 +1,5 @@
 from __future__ import annotations
 
-from .auth import AUTH_ID_PATTERN, AuthRecord, AuthStore, validate_auth_id
-from .listing import matches_query, paged_list_payload
 from .admin import (
     WorkflowAdminApi,
     WorkflowAdminAuthProvider,
@@ -9,8 +7,7 @@ from .admin import (
     WorkflowAdminEventProvider,
 )
 from .artifacts import WorkflowArtifactApi
-from .local_sources import builtin_sources, get_qualified_spec, qualify_spec
-from .models import RawWorkflowPlan, TraceRange
+from .auth import AUTH_ID_PATTERN, AuthRecord, AuthStore, validate_auth_id
 from .capabilities import WorkflowCapabilityApi
 from .constants import (
     DEFAULT_CALL_STEP_ID,
@@ -21,9 +18,21 @@ from .constants import (
 )
 from .deployments import WorkflowDeploymentApi
 from .drafts import WorkflowDraftApi
-from .next_actions import NextActionPatchExample, NextActionTool, NextActions
+from .durable_context import durable_workflow_api, require_workflow_stores
+from .listing import matches_query, paged_list_payload
+from .local_sources import builtin_sources, get_qualified_spec, qualify_spec
+from .models import RawWorkflowPlan, TraceRange
+from .next_actions import NextActionPatchExample, NextActions, NextActionTool
+from .operation_context import (
+    WorkflowEventRecorder,
+    WorkflowLiveSourceChecker,
+    WorkflowOperationContext,
+    WorkflowRuntimeRunner,
+    WorkflowSpecProvider,
+)
 from .refs import WorkflowSurfaceCapabilityId, parse_workflow_surface_capability_id
 from .runs import WorkflowRunApi
+from .runtime_dependencies import RuntimeDependencies, resolve_runtime_dependencies
 from .service import WorkflowApi
 from .source_admin import WorkflowSourceAdminApi
 from .source_registry_admin import (
@@ -32,6 +41,7 @@ from .source_registry_admin import (
     WorkflowSourceRegistryMutationProvider,
     WorkflowSourceRegistryProvider,
 )
+from .stores import WorkflowStores, file_workflow_stores
 from .surface import (
     WorkflowAdminSurface,
     WorkflowApiSurface,
@@ -55,18 +65,6 @@ from .wrapper_hints import (
     wrapper_hints_for_capability,
 )
 
-from .operation_context import (
-    WorkflowEventRecorder,
-    WorkflowLiveSourceChecker,
-    WorkflowOperationContext,
-    WorkflowRuntimeRunner,
-    WorkflowSpecProvider,
-)
-
-from .runtime_dependencies import RuntimeDependencies, resolve_runtime_dependencies
-from .stores import WorkflowStores, file_workflow_stores
-from .durable_context import durable_workflow_api, require_workflow_stores
-
 __all__ = [
     "AUTH_ID_PATTERN",
     "DEFAULT_CALL_STEP_ID",