feat: expose workflow runs over json rpc

2026-06-03 07:11:29 +07:00
parent 1130d1f1ed
commit 66750a8386
2 changed files with 167 additions and 0 deletions
@@ -5,6 +5,8 @@ from typing import Any

 import httpx

+from wf_api.models import RawWorkflowPlan
+from wf_core import END
 from wf_server import build_local_static_workflow_server
 from wf_transport_rpc_http.app import create_rpc_app

@@ -175,3 +177,108 @@ def test_rpc_draft_artifact_deployment_lifecycle(tmp_path) -> None:
        assert validate_deployment["result"]["status"] == "runnable"

    asyncio.run(scenario())
+
+
+def _constant_plan() -> RawWorkflowPlan:
+    return RawWorkflowPlan.model_validate(
+        {
+            "name": "rpc_constant",
+            "input_schema": {"type": "object", "properties": {}},
+            "state_schema": {
+                "type": "object",
+                "properties": {
+                    "result": {"type": "string", "reducer": "wf.std.replace"}
+                },
+            },
+            "output_schema": {
+                "type": "object",
+                "properties": {"result": {"type": "string"}},
+                "required": ["result"],
+            },
+            "outcomes": ["ok"],
+            "start": "constant",
+            "nodes": [
+                {
+                    "id": "constant",
+                    "type": "node",
+                    "node": "wf.std.constant",
+                    "input": [
+                        {
+                            "value": "hello over rpc",
+                            "target": {"root": "local", "parts": ["value"]},
+                        }
+                    ],
+                    "output": [
+                        {
+                            "source": {"root": "local", "parts": ["value"]},
+                            "target": {"root": "state", "parts": ["result"]},
+                        }
+                    ],
+                }
+            ],
+            "edges": [{"from": "constant", "outcome": "ok", "to": END}],
+            "output": [
+                {
+                    "path": {"root": "state", "parts": ["result"]},
+                    "target": {"root": "local", "parts": ["result"]},
+                }
+            ],
+        }
+    )
+
+
+def test_rpc_runs_deployment_and_reads_bounded_trace(tmp_path) -> None:
+    async def scenario() -> None:
+        server = build_local_static_workflow_server(tmp_path / "store")
+        await server.api.create_artifact_from_plan(
+            artifact_id="rpc_constant",
+            version=1,
+            title="RPC Constant",
+            plan=_constant_plan(),
+            outcomes=["ok"],
+            source_bindings={"wf.std": "wf.std"},
+        )
+        await server.api.save_deployment(
+            {
+                "id": "rpc_constant.default",
+                "artifact_id": "rpc_constant",
+                "artifact_version": 1,
+                "bindings": [{"logical_source": "wf.std", "concrete_source": "wf.std"}],
+            }
+        )
+
+        app = create_rpc_app(server)
+        transport = httpx.ASGITransport(app=app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://test") as client:
+            run = await _rpc(
+                client,
+                "workflow.runs.start",
+                {
+                    "deployment_id": "rpc_constant.default",
+                    "workflow_input": {},
+                    "trace_range": {"start": 0, "limit": 1},
+                },
+            )
+            inspected = await _rpc(
+                client,
+                "workflow.runs.inspect",
+                {"run_id": run["result"]["run_id"]},
+            )
+            trace = await _rpc(
+                client,
+                "workflow.runs.trace",
+                {
+                    "run_id": run["result"]["run_id"],
+                    "trace_range": {"start": 0, "limit": 1},
+                },
+            )
+
+        assert run["result"]["status"] == "completed"
+        assert run["result"]["output"]["result"] == "hello over rpc"
+        assert "trace" not in inspected["result"]
+        assert inspected["result"]["trace_count"] >= 1
+        assert trace["result"]["trace_start"] == 0
+        assert trace["result"]["trace_limit"] == 1
+        assert len(trace["result"]["trace"]) == 1
+
+    asyncio.run(scenario())