statewire 0.3.3__tar.gz → 0.4.0__tar.gz

{statewire-0.3.3 → statewire-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: statewire
-Version: 0.3.3
+Version: 0.4.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

{statewire-0.3.3 → statewire-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "statewire"
-version = "0.3.3"
+version = "0.4.0"
 description = "Replicate one JSON object over an SSE op stream + a command endpoint"
 readme = "README.md"
 license = "MIT"

{statewire-0.3.3 → statewire-0.4.0}/src/statewire/assistant_transport.py

@@ -17,8 +17,9 @@ the legacy commands, which are normalized to the canonical ``run/*`` dialect
 (a host-registered handler of the same legacy name always wins and receives
 the command whole):
 
-- ``add-message`` becomes ``run/steer`` with ``{"message": <wire message>}``:
-  a fresh ``legacy_`` id is synthesized, image parts become
+- ``add-message`` becomes ``run/steer`` with ``{"message": <wire message>,
+  "anchorMessageId": <body parentId>}`` (body ``parentId`` null or absent →
+  ``null``): a fresh ``legacy_`` id is synthesized, image parts become
   ``{"type": "file", "mediaType": "image/*", "url"}`` parts, and command-level
   anchors are dropped. A command-level non-null ``sourceId`` makes it
   ``run/edit`` with that ``sourceId`` directly; otherwise, when the body
@@ -31,8 +32,8 @@ the command whole):
   unanswered tool-call request in ``state["inputRequests"]`` (None rejects),
   and a ``modelContent`` field rejects (unsupported in ``run/input``).
 - A body with a ``parentId`` and no commands is the legacy reload: it becomes
-  ``run/reload`` with ``{"sourceId": <child>}`` via ``get_message_child_id``
-  (None rejects).
+  ``run/reload`` with ``{"sourceId": <child>, "messageId": <fresh legacy_
+  id>}`` via ``get_message_child_id`` (None rejects).
 
 A host without the target ``run/*`` handler rejects the batch via the
 ordinary unknown-command path. The response streams state changes as legacy
@@ -464,7 +465,10 @@ class AssistantTransport(Statewire):
                     "method": "run/edit",
                     "params": [{"sourceId": source_id, "message": message}],
                 }
-            return {"method": "run/steer", "params": [{"message": message}]}
+            return {
+                "method": "run/steer",
+                "params": [{"message": message, "anchorMessageId": parent_id}],
+            }
         if kind == "add-tool-result" and "add-tool-result" not in registered:
             return {
                 "method": "run/input",
@@ -494,7 +498,12 @@ class AssistantTransport(Statewire):
         source_id = await self.get_message_child_id(parent_id)
         if source_id is None:
             raise _RunError(f"reload: no child message for parentId {parent_id!r}")
-        return {"method": "run/reload", "params": [{"sourceId": source_id}]}
+        return {
+            "method": "run/reload",
+            "params": [
+                {"sourceId": source_id, "messageId": f"legacy_{uuid4().hex}"}
+            ],
+        }
 
     async def _assistant_transport_dispatch(self, method: str, identity: Any) -> None:
         client_id = "at-" + secrets.token_urlsafe(9)

{statewire-0.3.3 → statewire-0.4.0}/tests/test_assistant_transport.py

@@ -27,7 +27,7 @@ class Harness(AssistantTransport):
         self.state = {"messages": [], "text": "", "meta": {"title": None}}
         yield
 
-    @command("message/add")
+    @command("demo/echo")
     async def add_message(self, cmd):
         self.seen.append(cmd)
         self.state["messages"].append(cmd["message"])
@@ -120,7 +120,7 @@ async def test_data_stream_response_streams_snapshot_then_ops_until_idle():
             {
                 "commands": [
                     {
-                        "type": "message/add",
+                        "type": "demo/echo",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -229,7 +229,7 @@ async def test_splice_and_remove_fall_back_to_parent_set():
             {
                 "commands": [
                     {
-                        "type": "message/add",
+                        "type": "demo/echo",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -266,7 +266,7 @@ async def test_assistant_transport_sse_mode():
             {
                 "commands": [
                     {
-                        "type": "message/add",
+                        "type": "demo/echo",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -428,7 +428,7 @@ async def test_resume_after_completed_run_replays_with_completed_header():
             {
                 "commands": [
                     {
-                        "type": "message/add",
+                        "type": "demo/echo",
                         "message": {
                             "role": "user",
                             "parts": [{"type": "text", "text": "hi"}],
@@ -453,7 +453,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": "message/add"}]},
+            json={"commands": [{"type": "demo/echo"}]},
         )
         assert response.status_code == 400
         assert (
@@ -840,7 +840,8 @@ async def test_add_message_is_normalized_to_run_steer():
         assert response.status_code == 200
         instance = (await app.state.pinned_host.directory.get("t1")).instance
         [(params, caller_id)] = instance.steered
-        assert set(params) == {"message"}
+        assert set(params) == {"message", "anchorMessageId"}
+        assert params["anchorMessageId"] is None
         message = params["message"]
         assert message["id"].startswith("legacy_")
         assert message["role"] == "user"
@@ -1094,7 +1095,9 @@ async def test_parent_id_without_commands_is_normalized_to_run_reload():
         frames = data_stream_frames(response.text)
         assert all(prefix == "aui-state" for prefix, _ in frames)
         instance = (await app.state.pinned_host.directory.get("t1")).instance
-        assert instance.reloaded == [{"sourceId": "m2"}]
+        [params] = instance.reloaded
+        assert params["sourceId"] == "m2"
+        assert params["messageId"].startswith("legacy_")
 
 
 async def test_reload_without_a_resolvable_child_is_an_error_frame():
@@ -1184,7 +1187,8 @@ async def test_add_message_with_a_null_source_id_and_no_parent_is_run_steer():
         instance = (await app.state.pinned_host.directory.get("t1")).instance
         assert instance.edited == []
         [(params, _)] = instance.steered
-        assert set(params) == {"message"}
+        assert set(params) == {"message", "anchorMessageId"}
+        assert params["anchorMessageId"] is None
 
 
 async def test_add_message_with_an_unresolvable_parent_falls_back_to_run_steer():
@@ -1209,4 +1213,5 @@ async def test_add_message_with_an_unresolvable_parent_falls_back_to_run_steer()
         instance = (await app.state.pinned_host.directory.get("t1")).instance
         assert instance.edited == []
         [(params, _)] = instance.steered
-        assert set(params) == {"message"}
+        assert set(params) == {"message", "anchorMessageId"}
+        assert params["anchorMessageId"] == "unknown"

{statewire-0.3.3 → statewire-0.4.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": "message/add", "message": "hi"}]) is True
+        assert await client.send([{"type": "demo/echo", "message": "hi"}]) is True
         assert client.state == {"text": "hello"}
     assert received == [
         {
-            "commands": [{"type": "message/add", "message": "hi"}],
+            "commands": [{"type": "demo/echo", "message": "hi"}],
             "state": {"text": ""},
             "threadId": "t1",
             "system": "be nice",