feat: merge draft step maps

src/wf_transport_rpc_http/client/drafts.py

@@ -108,6 +108,7 @@ class RpcDraftClientMixin:
         revision: int,
         step_id: str,
         input_map: dict[str, str],
+        merge: bool = False,
     ) -> dict[str, Any]:
         return await self._call(
             "workflow.draft_workspaces.set_step_input_map",
@@ -116,6 +117,7 @@ class RpcDraftClientMixin:
                 "revision": revision,
                 "step_id": step_id,
                 "input_map": input_map,
+                "merge": merge,
             },
         )
 
@@ -126,6 +128,7 @@ class RpcDraftClientMixin:
         revision: int,
         step_id: str,
         output_map: dict[str, str],
+        merge: bool = False,
     ) -> dict[str, Any]:
         return await self._call(
             "workflow.draft_workspaces.set_step_output_map",
@@ -134,6 +137,7 @@ class RpcDraftClientMixin:
                 "revision": revision,
                 "step_id": step_id,
                 "output_map": output_map,
+                "merge": merge,
             },
         )
 

src/wf_transport_rpc_http/methods/drafts.py

@@ -154,6 +154,7 @@ def register_methods(
                 revision=params.revision,
                 step_id=params.step_id,
                 input_map=params.input_map,
+                merge=params.merge,
             )
         except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
             raise_workflow_rpc_error(exc)
@@ -171,6 +172,7 @@ def register_methods(
                 revision=params.revision,
                 step_id=params.step_id,
                 output_map=params.output_map,
+                merge=params.merge,
             )
         except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
             raise_workflow_rpc_error(exc)

src/wf_transport_rpc_http/models.py

@@ -130,6 +130,7 @@ class SetStepInputMapParams(RpcParamsModel):
     revision: int = Field(ge=1)
     step_id: str = Field(min_length=1)
     input_map: dict[str, str]
+    merge: bool = False
 
 
 class SetStepOutputMapParams(RpcParamsModel):
@@ -137,6 +138,7 @@ class SetStepOutputMapParams(RpcParamsModel):
     revision: int = Field(ge=1)
     step_id: str = Field(min_length=1)
     output_map: dict[str, str]
+    merge: bool = False
 
 
 class ValidateDraftWorkspaceParams(RpcParamsModel):