create-a-workflow! end to end usage!

2026-05-12 18:33:52 +07:00
parent 2ec956d2ba
commit f90367d471
7 changed files with 740 additions and 0 deletions
@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from wf_artifacts import RequiredCapability, create_workflow_artifact_from_plan
+
+
+def test_create_workflow_artifact_from_plan_derives_boundary_schemas() -> None:
+    artifact = create_workflow_artifact_from_plan(
+        artifact_id="echo",
+        version=1,
+        title="Echo",
+        description="Echo through a saved plan.",
+        plan=_plan(),
+        outcomes=("done",),
+        required_capabilities={
+            "demo.echo_tool": RequiredCapability(
+                logical_source="demo",
+                capability_name="echo_tool",
+                kind="node_spec",
+            )
+        },
+        created_from_catalog_version="catalog-1",
+    )
+
+    assert artifact.id == "echo"
+    assert artifact.version == 1
+    assert artifact.title == "Echo"
+    assert artifact.input_schema["properties"]["text"]["type"] == "string"
+    assert artifact.output_schema["properties"]["echoed"]["type"] == "string"
+    assert artifact.outcomes == ("done",)
+    assert artifact.plan["name"] == "echo"
+    assert "demo.echo_tool" in artifact.required_capabilities
+    assert artifact.created_from_catalog_version == "catalog-1"
+
+
+def test_create_workflow_artifact_from_plan_rejects_missing_boundary_schema() -> None:
+    plan = _plan()
+    plan.pop("output_schema")
+
+    try:
+        create_workflow_artifact_from_plan(
+            artifact_id="echo",
+            version=1,
+            title="Echo",
+            plan=plan,
+            outcomes=("done",),
+        )
+    except ValueError as exc:
+        assert "output_schema" in str(exc)
+    else:
+        raise AssertionError("expected missing output_schema to be rejected")
+
+
+def _plan() -> dict[str, object]:
+    return {
+        "name": "echo",
+        "input_schema": {
+            "type": "object",
+            "properties": {"text": {"type": "string"}},
+            "required": ["text"],
+        },
+        "state_schema": {"fields": {"echoed": {"type": "string"}}},
+        "output_schema": {
+            "type": "object",
+            "properties": {"echoed": {"type": "string"}},
+            "required": ["echoed"],
+        },
+        "start": "echo",
+        "nodes": [],
+        "edges": [],
+    }