feat: add focused draft remove commands

src/wf_transport_rpc_http/__init__.py

@@ -32,6 +32,9 @@ from .models import (
     PatchDraftParams,
     PatchDraftWorkspaceParams,
     ReadRunTraceParams,
+    RemoveDraftBindingParams,
+    RemoveDraftRouteParams,
+    RemoveDraftStepParams,
     ResumeRunParams,
     SaveArtifactParams,
     SaveDeploymentParams,
@@ -75,6 +78,9 @@ __all__ = [
     "PatchDraftParams",
     "PatchDraftWorkspaceParams",
     "ReadRunTraceParams",
+    "RemoveDraftBindingParams",
+    "RemoveDraftRouteParams",
+    "RemoveDraftStepParams",
     "ResumeRunParams",
     "SaveArtifactParams",
     "SaveDeploymentParams",

src/wf_transport_rpc_http/client/drafts.py

@@ -225,6 +225,60 @@ class RpcDraftClientMixin:
             },
         )
 
+    async def remove_draft_route(
+        self: RpcCaller,
+        *,
+        workspace_id: str,
+        revision: int,
+        step_id: str,
+        outcome: str,
+    ) -> dict[str, Any]:
+        return await self._call(
+            "workflow.draft_workspaces.remove_route",
+            {
+                "workspace_id": workspace_id,
+                "revision": revision,
+                "step_id": step_id,
+                "outcome": outcome,
+            },
+        )
+
+    async def remove_draft_step(
+        self: RpcCaller,
+        *,
+        workspace_id: str,
+        revision: int,
+        step_id: str,
+    ) -> dict[str, Any]:
+        return await self._call(
+            "workflow.draft_workspaces.remove_step",
+            {
+                "workspace_id": workspace_id,
+                "revision": revision,
+                "step_id": step_id,
+            },
+        )
+
+    async def remove_draft_binding(
+        self: RpcCaller,
+        *,
+        workspace_id: str,
+        revision: int,
+        step_id: str,
+        inputs: Sequence[str] = (),
+        outputs: Sequence[str] = (),
+    ) -> dict[str, Any]:
+        return await self._call(
+            "workflow.draft_workspaces.remove_binding",
+            {
+                "workspace_id": workspace_id,
+                "revision": revision,
+                "step_id": step_id,
+                "inputs": list(inputs),
+                "outputs": list(outputs),
+            },
+        )
+
     async def validate_draft_workspace(
         self: RpcCaller,
         *,

src/wf_transport_rpc_http/methods/drafts.py

@@ -21,6 +21,9 @@ from ..models import (
     ListDraftWorkspacesParams,
     PatchDraftParams,
     PatchDraftWorkspaceParams,
+    RemoveDraftBindingParams,
+    RemoveDraftRouteParams,
+    RemoveDraftStepParams,
     SetDraftNameParams,
     SetDraftRouteParams,
     SetStepInputMapParams,
@@ -296,6 +299,54 @@ def register_methods(
         except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
             raise_workflow_rpc_error(exc)
 
+    @entrypoint.method(
+        name="workflow.draft_workspaces.remove_route", errors=[WorkflowRpcError]
+    )
+    async def workflow_draft_workspaces_remove_route(
+        params: RemoveDraftRouteParams = RpcParams(),
+    ) -> dict[str, Any]:
+        try:
+            return await server.api.remove_draft_route(
+                workspace_id=params.workspace_id,
+                revision=params.revision,
+                step_id=params.step_id,
+                outcome=params.outcome,
+            )
+        except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
+            raise_workflow_rpc_error(exc)
+
+    @entrypoint.method(
+        name="workflow.draft_workspaces.remove_step", errors=[WorkflowRpcError]
+    )
+    async def workflow_draft_workspaces_remove_step(
+        params: RemoveDraftStepParams = RpcParams(),
+    ) -> dict[str, Any]:
+        try:
+            return await server.api.remove_draft_step(
+                workspace_id=params.workspace_id,
+                revision=params.revision,
+                step_id=params.step_id,
+            )
+        except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
+            raise_workflow_rpc_error(exc)
+
+    @entrypoint.method(
+        name="workflow.draft_workspaces.remove_binding", errors=[WorkflowRpcError]
+    )
+    async def workflow_draft_workspaces_remove_binding(
+        params: RemoveDraftBindingParams = RpcParams(),
+    ) -> dict[str, Any]:
+        try:
+            return await server.api.remove_draft_binding(
+                workspace_id=params.workspace_id,
+                revision=params.revision,
+                step_id=params.step_id,
+                inputs=params.inputs,
+                outputs=params.outputs,
+            )
+        except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
+            raise_workflow_rpc_error(exc)
+
     @entrypoint.method(
         name="workflow.draft_workspaces.create_artifact", errors=[WorkflowRpcError]
     )

src/wf_transport_rpc_http/models.py

@@ -180,6 +180,27 @@ class HandleDraftParams(RpcParamsModel):
     target: str = Field(min_length=1)
 
 
+class RemoveDraftRouteParams(RpcParamsModel):
+    workspace_id: str = Field(min_length=1)
+    revision: int = Field(ge=1)
+    step_id: str = Field(min_length=1)
+    outcome: str = Field(min_length=1)
+
+
+class RemoveDraftStepParams(RpcParamsModel):
+    workspace_id: str = Field(min_length=1)
+    revision: int = Field(ge=1)
+    step_id: str = Field(min_length=1)
+
+
+class RemoveDraftBindingParams(RpcParamsModel):
+    workspace_id: str = Field(min_length=1)
+    revision: int = Field(ge=1)
+    step_id: str = Field(min_length=1)
+    inputs: list[str] = Field(default_factory=list)
+    outputs: list[str] = Field(default_factory=list)
+
+
 class ValidateDraftWorkspaceParams(RpcParamsModel):
     workspace_id: str = Field(min_length=1)