feat: expose output bindings over rpc

src/wf_transport_rpc_http/__init__.py

@@ -45,6 +45,7 @@ from .models import (
     SetDraftStartParams,
     SetStepInputBindingsParams,
     SetStepInputMapParams,
+    SetStepOutputBindingsParams,
     SetStepOutputMapParams,
     SetWorkflowOutputMapParams,
     StartRunParams,
@@ -96,6 +97,7 @@ __all__ = [
     "SetDraftStartParams",
     "SetStepInputBindingsParams",
     "SetStepInputMapParams",
+    "SetStepOutputBindingsParams",
     "SetStepOutputMapParams",
     "SetWorkflowOutputMapParams",
     "StartRunParams",

src/wf_transport_rpc_http/client/drafts.py

@@ -5,7 +5,7 @@ from typing import Any, Literal
 
 from wf_api.surface import RouteSource
 from wf_artifacts.drafts.models import DraftStep
-from wf_core.models.steps import InputBinding
+from wf_core.models.steps import InputBinding, OutputBinding
 
 from .base import RpcCaller
 
@@ -201,6 +201,24 @@ class RpcDraftClientMixin:
             },
         )
 
+    async def set_step_output_bindings(
+        self: RpcCaller,
+        *,
+        workspace_id: str,
+        revision: int,
+        step_id: str,
+        bindings: Sequence[OutputBinding],
+    ) -> dict[str, Any]:
+        return await self._call(
+            "workflow.draft_workspaces.set_step_output_bindings",
+            {
+                "workspace_id": workspace_id,
+                "revision": revision,
+                "step_id": step_id,
+                "bindings": [binding.model_dump(mode="json") for binding in bindings],
+            },
+        )
+
     async def set_step_output_map(
         self: RpcCaller,
         *,

src/wf_transport_rpc_http/methods/drafts.py

@@ -33,6 +33,7 @@ from ..models import (
     SetDraftStartParams,
     SetStepInputBindingsParams,
     SetStepInputMapParams,
+    SetStepOutputBindingsParams,
     SetStepOutputMapParams,
     SetWorkflowOutputMapParams,
     ValidateDraftParams,
@@ -244,6 +245,23 @@ def register_methods(
         except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
             raise_workflow_rpc_error(exc)
 
+    @entrypoint.method(
+        name="workflow.draft_workspaces.set_step_output_bindings",
+        errors=[WorkflowRpcError],
+    )
+    async def workflow_draft_workspaces_set_step_output_bindings(
+        params: SetStepOutputBindingsParams = RpcParams(),
+    ) -> dict[str, Any]:
+        try:
+            return await server.api.set_step_output_bindings(
+                workspace_id=params.workspace_id,
+                revision=params.revision,
+                step_id=params.step_id,
+                bindings=params.bindings,
+            )
+        except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
+            raise_workflow_rpc_error(exc)
+
     @entrypoint.method(
         name="workflow.draft_workspaces.set_step_output_map",
         errors=[WorkflowRpcError],

src/wf_transport_rpc_http/models.py

@@ -6,7 +6,7 @@ from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from wf_api.models import TraceRange
 from wf_artifacts.drafts.models import DraftStep
-from wf_core.models.steps import InputBinding
+from wf_core.models.steps import InputBinding, OutputBinding
 
 
 class RpcParamsModel(BaseModel):
@@ -216,6 +216,13 @@ class SetStepInputBindingsParams(RpcParamsModel):
     bindings: list[InputBinding]
 
 
+class SetStepOutputBindingsParams(RpcParamsModel):
+    workspace_id: str = Field(min_length=1)
+    revision: int = Field(ge=1)
+    step_id: str = Field(min_length=1)
+    bindings: list[OutputBinding]
+
+
 class SetStepOutputMapParams(RpcParamsModel):
     workspace_id: str = Field(min_length=1)
     revision: int = Field(ge=1)