feat: add callable remote workflow capabilities

2026-08-31 01:01:40 +07:00
parent 00b51d2d96
commit d6f5a2e31d
6 changed files with 637 additions and 1 deletions
@@ -0,0 +1,132 @@
+from __future__ import annotations
+
+from typing import Any, cast
+
+import pytest
+
+from wf_client import CapabilityResult, RemoteCapability
+from wf_client.errors import InvalidResponse
+from wf_client.protocols import WorkflowClientPort
+from wf_platform import CapabilityRef, SourceRef
+
+
+def _inspect_payload() -> dict[str, Any]:
+    return {
+        "name": "app.default.search",
+        "source_id": "app.default",
+        "kind": "node_spec",
+        "description": "Search things",
+        "outcomes": ["ok", "error"],
+        "is_async": True,
+        "input_schema": {
+            "type": "object",
+            "properties": {"query": {"type": "string"}},
+            "required": ["query"],
+        },
+        "output_schema": {
+            "type": "object",
+            "properties": {"results": {"type": "array"}},
+            "required": ["results"],
+        },
+        "wrapper_hints": {},
+        "accepts_context": False,
+    }
+
+
+class _Port:
+    def __init__(self) -> None:
+        self.calls: list[dict[str, Any]] = []
+
+    async def call_capability(self, **params: Any) -> object:
+        self.calls.append(params)
+        return {
+            "qualified_name": "app.default.search",
+            "source_id": "app.default",
+            "kind": "node_spec",
+            "deployment_id": None,
+            "outcome": "ok",
+            "output": {"results": ["one"]},
+            "diagnostics": [],
+        }
+
+
+def _port() -> WorkflowClientPort:
+    return cast(WorkflowClientPort, _Port())
+
+
+def test_remote_capability_preserves_dotted_local_key() -> None:
+    capability = RemoteCapability(
+        _port=_port(),
+        ref=CapabilityRef(source=SourceRef.parse("app.default"), name="search.v2"),
+        qualified_name="app.default.search.v2",
+        description=None,
+        input_schema={"type": "object"},
+        output_schema={"type": "object"},
+        outcomes=("ok",),
+        is_async=False,
+    )
+
+    assert capability.ref.name == "search.v2"
+    assert capability.node_def().name == "app.default.search.v2"
+
+
+@pytest.mark.asyncio
+async def test_remote_capability_is_callable_and_validates_result() -> None:
+    port = _Port()
+    capability = RemoteCapability(
+        _port=cast(WorkflowClientPort, port),
+        ref=CapabilityRef.parse("app.default.search"),
+        qualified_name="app.default.search",
+        description=None,
+        input_schema={
+            "type": "object",
+            "properties": {"query": {"type": "string"}},
+            "required": ["query"],
+        },
+        output_schema={"type": "object", "required": ["results"]},
+        outcomes=("ok",),
+        is_async=False,
+    )
+
+    result = await capability(query="workflow")
+
+    assert isinstance(result, CapabilityResult)
+    assert result.output == {"results": ["one"]}
+    assert port.calls == [
+        {
+            "qualified_name": "app.default.search",
+            "payload": {"query": "workflow"},
+            "deployment_id": None,
+        }
+    ]
+
+
+@pytest.mark.asyncio
+async def test_remote_capability_rejects_mixed_payload_forms() -> None:
+    capability = RemoteCapability(
+        _port=_port(),
+        ref=CapabilityRef.parse("app.default.search"),
+        qualified_name="app.default.search",
+        description=None,
+        input_schema={"type": "object"},
+        output_schema={"type": "object"},
+        outcomes=("ok",),
+        is_async=False,
+    )
+
+    with pytest.raises(TypeError, match="not both"):
+        await capability({"query": "workflow"}, query="again")
+
+
+def test_remote_capability_rejects_invalid_inspected_schema() -> None:
+    with pytest.raises(InvalidResponse, match="invalid JSON Schema"):
+        RemoteCapability(
+            _port=_port(),
+            ref=CapabilityRef.parse("app.default.search"),
+            qualified_name="app.default.search",
+            description=None,
+            input_schema={"type": "not-a-json-schema-type"},
+            output_schema={"type": "object"},
+            outcomes=("ok",),
+            is_async=False,
+        )