feat: bind draft outputs to state

src/wf_transport_rpc_http/client/drafts.py

@@ -161,6 +161,26 @@ class RpcDraftClientMixin:
             },
         )
 
+    async def bind_output_to_state(
+        self: RpcCaller,
+        *,
+        workspace_id: str,
+        revision: int,
+        step_id: str,
+        output_field: str,
+        state_path: str,
+    ) -> dict[str, Any]:
+        return await self._call(
+            "workflow.draft_workspaces.bind_output_to_state",
+            {
+                "workspace_id": workspace_id,
+                "revision": revision,
+                "step_id": step_id,
+                "output_field": output_field,
+                "state_path": state_path,
+            },
+        )
+
     async def validate_draft_workspace(
         self: RpcCaller,
         *,

src/wf_transport_rpc_http/methods/drafts.py

@@ -9,6 +9,7 @@ from wf_server import WorkflowServer
 from ..errors import WorkflowRpcError, raise_workflow_rpc_error
 from ..models import (
     AddStateFromOutputParams,
+    BindOutputToStateParams,
     CreateArtifactFromWorkspaceParams,
     CreateDraftFromCapabilityParams,
     CreateWrapperFromWorkspaceParams,
@@ -196,6 +197,24 @@ def register_methods(
         except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
             raise_workflow_rpc_error(exc)
 
+    @entrypoint.method(
+        name="workflow.draft_workspaces.bind_output_to_state",
+        errors=[WorkflowRpcError],
+    )
+    async def workflow_draft_workspaces_bind_output_to_state(
+        params: BindOutputToStateParams = RpcParams(),
+    ) -> dict[str, Any]:
+        try:
+            return await server.api.bind_output_to_state(
+                workspace_id=params.workspace_id,
+                revision=params.revision,
+                step_id=params.step_id,
+                output_field=params.output_field,
+                state_path=params.state_path,
+            )
+        except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
+            raise_workflow_rpc_error(exc)
+
     @entrypoint.method(
         name="workflow.draft_workspaces.validate", errors=[WorkflowRpcError]
     )

src/wf_transport_rpc_http/models.py

@@ -149,6 +149,14 @@ class AddStateFromOutputParams(RpcParamsModel):
     state_path: str = Field(min_length=1)
 
 
+class BindOutputToStateParams(RpcParamsModel):
+    workspace_id: str = Field(min_length=1)
+    revision: int = Field(ge=1)
+    step_id: str = Field(min_length=1)
+    output_field: str = Field(min_length=1)
+    state_path: str = Field(min_length=1)
+
+
 class ValidateDraftWorkspaceParams(RpcParamsModel):
     workspace_id: str = Field(min_length=1)