ts for some reason, moved to wf_api

tests/wf_mcp/service/conftest.py

@@ -5,7 +5,8 @@ from typing import Any
 from wf_authoring import node
 from wf_core import END
 from wf_mcp.capabilities import DiscoveredTool
-from wf_mcp.models import AuthRecord, ConnectionConfig, RawWorkflowPlan
+from wf_api.models import RawWorkflowPlan
+from wf_mcp.models import AuthRecord, ConnectionConfig
 from wf_mcp.sdk import ToolCallResult
 
 from ..test_support import (

tests/wf_mcp/workflow_surface/test_runs.py

@@ -29,7 +29,7 @@ from .conftest import (
 
 
 def test_raw_workflow_plan_uses_core_step_and_edge_models() -> None:
-    from wf_mcp.models import RawWorkflowPlan
+    from wf_api.models import RawWorkflowPlan
 
     plan = RawWorkflowPlan.model_validate(echo_artifact().plan)