fix: opt draft storage into real callers

2026-08-31 14:07:53 +07:00
parent ce7e3ed761
commit b377311a6e
21 changed files with 194 additions and 50 deletions
@@ -131,7 +131,7 @@ def _artifact_api(
        )
        service.register_specs("demo.personal", echo_tool)
    context = context_from_service(service)
-    return WorkflowArtifactApi(context), service
+    return WorkflowArtifactApi(context, drafts=True), service


@pytest.mark.asyncio
@@ -454,7 +454,7 @@ def _api(root: Path) -> WorkflowApi:
        artifact_store=FileWorkflowArtifactStore(root),
        draft_workspace_store=FileDraftWorkspaceStore(mcp_root),
    )
-    return WorkflowApi(context_from_service(service))
+    return WorkflowApi(context_from_service(service), drafts=True)


@pytest.mark.asyncio