statewire 0.1.2__tar.gz → 0.2.0__tar.gz

{statewire-0.1.2 → statewire-0.2.0}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: statewire
-Version: 0.1.2
+Version: 0.2.0
 Summary: Replicate one JSON object over an SSE op stream + a command endpoint
 Project-URL: Repository, https://github.com/assistant-ui/harness-sdk
 License-Expression: MIT
 Requires-Python: <4.0,>=3.11
 Requires-Dist: fastapi>=0.115
 Requires-Dist: httpx>=0.27
-Requires-Dist: pinned<0.2,>=0.1.0
+Requires-Dist: pinned<0.3,>=0.2.0
 Provides-Extra: langgraph
 Requires-Dist: langchain-core>=0.3; extra == 'langgraph'
 Description-Content-Type: text/markdown

{statewire-0.1.2 → statewire-0.2.0}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "statewire"
-version = "0.1.2"
+version = "0.2.0"
 description = "Replicate one JSON object over an SSE op stream + a command endpoint"
 readme = "README.md"
 license = "MIT"
 requires-python = ">=3.11,<4.0"
 dependencies = [
-    "pinned>=0.1.0,<0.2",
+    "pinned>=0.2.0,<0.3",
     "fastapi>=0.115",
     "httpx>=0.27",
 ]

{statewire-0.1.2 → statewire-0.2.0}/src/statewire/api.py

@@ -168,6 +168,10 @@ class StatewireClientHandle:
         self._host = host
         self._client = client
 
+    @property
+    def client_id(self) -> str:
+        return self._client.id
+
     @property
     def context(self) -> dict[str, Any] | None:
         return self._client.context
@@ -297,18 +301,23 @@ def _settled_response(settled: _Settled) -> dict[str, Any]:
     return {"type": "rejected", "message": str(value) or type(value).__name__}
 
 
+def _wants_caller(fn: Handler, desc: str) -> bool:
+    caller = next(
+        (p for p in inspect.signature(fn).parameters.values() if p.name == "caller"),
+        None,
+    )
+    if caller is not None and caller.kind is not inspect.Parameter.KEYWORD_ONLY:
+        raise TypeError(f"{desc} must declare 'caller' keyword-only")
+    return caller is not None
+
+
 def _mark(fn: Handler, method: str) -> Handler:
     if not inspect.iscoroutinefunction(fn):
         raise TypeError(f"@command handler {fn.__qualname__!r} must be 'async def'")
     params = list(inspect.signature(fn).parameters.values())
     if not params or params[0].name != "self":
         raise TypeError(f"@command handler {fn.__qualname__!r} must be an instance method")
-    caller = next((p for p in params if p.name == "caller"), None)
-    if caller is not None and caller.kind is not inspect.Parameter.KEYWORD_ONLY:
-        raise TypeError(
-            f"@command handler {fn.__qualname__!r} must declare 'caller' keyword-only"
-        )
-    fn._statewire_wants_caller = caller is not None  # type: ignore[attr-defined]
+    fn._statewire_wants_caller = _wants_caller(fn, f"@command handler {fn.__qualname__!r}")  # type: ignore[attr-defined]
     fn._statewire_command_method = method  # type: ignore[attr-defined]
     return fn
 
@@ -788,10 +797,9 @@ class Statewire(PinnedAPI):
                 )
                 return
         handler = getattr(self, attr)
+        wants_caller = getattr(handler, "_statewire_wants_caller", False)
         kwargs = (
-            {"caller": StatewireClientHandle(self, client)}
-            if getattr(handler, "_statewire_wants_caller", False)
-            else {}
+            {"caller": StatewireClientHandle(self, client)} if wants_caller else {}
         )
         try:
             inspect.signature(handler).bind(*params, **kwargs)

{statewire-0.1.2 → statewire-0.2.0}/tests/test_assistant_transport.py

@@ -24,14 +24,14 @@ class Harness(AssistantTransport):
         self.state = {"messages": [], "text": "", "meta": {"title": None}}
         yield
 
-    @command("add-message")
+    @command("message/add")
     async def add_message(self, cmd):
         self.seen.append(cmd)
         self.state["messages"].append(cmd["message"])
         self.state["text"] += "echo: "
         self.state["text"] += cmd["message"]["parts"][0]["text"]
 
-    @command("add-tool-result")
+    @command("tool-result/add")
     async def add_tool_result(self, cmd):
         self.seen.append(cmd)
         self.state["meta"]["title"] = cmd["toolCallId"]
@@ -120,7 +120,7 @@ async def test_data_stream_response_streams_snapshot_then_ops_until_settle():
             {
                 "commands": [
                     {
-                        "type": "add-message",
+                        "type": "message/add",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -158,7 +158,7 @@ async def test_data_stream_response_streams_snapshot_then_ops_until_settle():
 async def test_unknown_fields_are_tolerated_and_commands_pass_through_whole():
     async with statewire_client(Harness) as (app, client):
         command = {
-            "type": "add-tool-result",
+            "type": "tool-result/add",
             "toolCallId": "call_1",
             "result": {"ok": True},
             "custom-athena-field": {"nested": [1, 2]},
@@ -229,7 +229,7 @@ async def test_splice_and_remove_fall_back_to_parent_set():
             {
                 "commands": [
                     {
-                        "type": "add-message",
+                        "type": "message/add",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -266,7 +266,7 @@ async def test_assistant_transport_sse_mode():
             {
                 "commands": [
                     {
-                        "type": "add-message",
+                        "type": "message/add",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -424,7 +424,7 @@ async def test_resume_after_completed_run_replays_with_completed_header():
             {
                 "commands": [
                     {
-                        "type": "add-message",
+                        "type": "message/add",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -449,7 +449,7 @@ async def test_resume_rejects_commands_and_malformed_bodies():
     async with statewire_client(Harness) as (app, client):
         response = await client.post(
             "/threads/t1/assistant-transport/api/resume",
-            json={"commands": [{"type": "add-message"}]},
+            json={"commands": [{"type": "message/add"}]},
         )
         assert response.status_code == 400
         assert (

{statewire-0.1.2 → statewire-0.2.0}/tests/test_assistant_transport_client.py

@@ -124,11 +124,11 @@ async def test_client_posts_the_legacy_body_and_applies_state_ops():
             body={"system": "be nice"},
             on_chunk=seen.append,
         )
-        assert await client.send([{"type": "add-message", "message": "hi"}]) is True
+        assert await client.send([{"type": "message/add", "message": "hi"}]) is True
         assert client.state == {"text": "hello"}
     assert received == [
         {
-            "commands": [{"type": "add-message", "message": "hi"}],
+            "commands": [{"type": "message/add", "message": "hi"}],
             "state": {"text": ""},
             "threadId": "t1",
             "system": "be nice",