feat: add draft state schema projection helper

2026-06-26 11:34:37 +07:00
parent bcbdb81228
commit c6a1dd336e
23 changed files with 1545 additions and 0 deletions
@@ -141,6 +141,26 @@ class RpcDraftClientMixin:
            },
        )

+    async def add_state_schema_from_output(
+        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.add_state_from_output",
+            {
+                "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,
        *,