harness-sdk-python 0.29.0__tar.gz → 0.29.2__tar.gz

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: harness-sdk-python
-Version: 0.29.0
+Version: 0.29.2
 Summary: RunManager: the harness-sdk runs subsystem for Python Statewire hosts
 Project-URL: Repository, https://github.com/assistant-ui/harness-sdk
 License-Expression: MIT

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "harness-sdk-python"
-version = "0.29.0"
+version = "0.29.2"
 description = "RunManager: the harness-sdk runs subsystem for Python Statewire hosts"
 readme = "README.md"
 license = "MIT"

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/src/harness_sdk/deepagents.py

@@ -1,7 +1,8 @@
 """DeepagentsMessages: the ``messages`` slice of a Statewire host, projected
 from a langgraph checkpointer in the deepagents dialect. The checkpointer is
 the source of truth and the wire state is its projection, restamped with
-``branch: {id, siblings}`` metadata after every ingest. A rewind restores the
+``branch: {id, siblings}`` metadata after every ingest; ephemeral overlay
+entries carry ``ephemeral: true`` instead. A rewind restores the
 projection from the target message's anchor checkpoint; a branch switch clones
 the target branch's tip as the checkpointer's latest checkpoint, so resumption
 and reboot both land on the switched world.
@@ -243,7 +244,7 @@ class DeepagentsMessages:
         return snapshot
 
     def upsert_ephemeral_message(self, message: dict[str, Any]) -> None:
-        """Render an in-flight ``{id, role, text}`` entry in ``state["messages"]``; an ``append_messages`` of the same id replaces it durably."""
+        """Render an in-flight ``{id, role, text}`` entry in ``state["messages"]`` tagged ``ephemeral: true``; an ``append_messages`` of the same id replaces it durably."""
         if message["role"] not in ("user", "assistant"):
             raise ValueError(f"unsupported message role: {message['role']!r}")
         if message["id"] not in self._ephemeral and any(
@@ -280,6 +281,8 @@ class DeepagentsMessages:
             stamped.append({**message, "branch": {"id": branch, "siblings": siblings}})
         durable = {message["id"] for message in stamped}
         stamped.extend(
-            entry for id_, entry in self._ephemeral.items() if id_ not in durable
+            {**entry, "ephemeral": True}
+            for id_, entry in self._ephemeral.items()
+            if id_ not in durable
         )
         self._replace(stamped)

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/src/harness_sdk/voice.py

@@ -13,11 +13,13 @@ status}`` — the SDP answer rides only the command's result payload
 ``{answer}``. The session is bound to the issuing client's connection: ``voice/end`` from the owner, a
 provider ``closed`` event, or a connection drop ends it. Provider turn
 events render live in ``state["messages"]`` through
-``Host.upsert_ephemeral_message`` (message id = turn id); each completed
+``Host.upsert_ephemeral_message`` (message id = turn id), tagged
+``ephemeral: true`` on the wire; each completed
 turn commits through ``Host.commit_voice_turns`` (serialized, in
 done-order) and the durable entry replaces the ephemeral one under the
 same id, a failed commit keeps its turns for the next flush; session end
-flushes the remaining done turns and drops in-progress turns.
+flushes the remaining done turns, then commits in-progress turns with
+their partial text; in-progress turns with no text yet are dropped.
 
 Rejections: ``voice-active`` (a session is already live), ``no-voice``
 (no voice manager declared, or ``voice/end`` with no session),
@@ -188,6 +190,10 @@ class VoiceManager(Tray):
                 await provider.close()
             except Exception:
                 logger.exception("voice provider close failed for session %r", self.id)
+        self._turns.sort(key=lambda turn: not turn["done"])
+        for turn in self._turns:
+            if turn["text"]:
+                turn["done"] = True
         await self._flush()
         turns, self._turns = self._turns, []
         for turn in turns:

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/tests/test_deepagents_messages.py

@@ -170,10 +170,16 @@ async def test_ingest_turn_refetch_discards_the_capture():
 async def test_ephemeral_message_renders_and_updates_in_place():
     _, _, _, messages, state = await seeded()
     messages.upsert_ephemeral_message({"id": "v1", "role": "user", "text": "hel"})
-    assert state["messages"][-1] == {"id": "v1", "type": "human", "content": "hel"}
+    assert state["messages"][-1] == {
+        "id": "v1",
+        "type": "human",
+        "content": "hel",
+        "ephemeral": True,
+    }
     messages.upsert_ephemeral_message({"id": "v1", "role": "user", "text": "hello"})
     assert ids(state) == ["u1", "a0", "u2", "a1", "v1"]
     assert state["messages"][-1]["content"] == "hello"
+    assert all("ephemeral" not in m for m in state["messages"][:-1])
 
 
 async def test_ephemeral_message_survives_a_turn_ingest():
@@ -181,7 +187,12 @@ async def test_ephemeral_message_survives_a_turn_ingest():
     messages.upsert_ephemeral_message({"id": "v1", "role": "assistant", "text": "hi"})
     await turn(graph, messages, Ctx("message-send", (user("u3", "three"),)))
     assert ids(state) == ["u1", "a0", "u2", "a1", "u3", "a2", "v1"]
-    assert state["messages"][-1] == {"id": "v1", "type": "ai", "content": "hi"}
+    assert state["messages"][-1] == {
+        "id": "v1",
+        "type": "ai",
+        "content": "hi",
+        "ephemeral": True,
+    }
 
 
 async def test_append_messages_replaces_the_ephemeral_entry_durably():
@@ -190,6 +201,7 @@ async def test_append_messages_replaces_the_ephemeral_entry_durably():
     await messages.append_messages([{"id": "v1", "role": "user", "text": "spoken"}])
     assert ids(state) == ["u1", "a0", "u2", "a1", "v1"]
     assert "branch" in state["messages"][-1]
+    assert "ephemeral" not in state["messages"][-1]
     messages.drop_ephemeral_message("v1")
     assert ids(state) == ["u1", "a0", "u2", "a1", "v1"]
 

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/tests/test_voice.py

@@ -88,7 +88,14 @@ async def test_turn_streams_into_messages_ephemerally():
         events.turn_delta("t1", "there")
         await drv.wait(
             lambda s: messages_of(s)
-            == [{"id": "t1", "type": "human", "content": "hello there"}]
+            == [
+                {
+                    "id": "t1",
+                    "type": "human",
+                    "content": "hello there",
+                    "ephemeral": True,
+                }
+            ]
         )
         assert bench.commits == []
 
@@ -141,7 +148,14 @@ async def test_commit_failure_keeps_the_turn_until_the_next_flush():
         assert bench.commits == []
         await drv.wait(
             lambda s: messages_of(s)
-            == [{"id": "t1", "type": "human", "content": "question"}]
+            == [
+                {
+                    "id": "t1",
+                    "type": "human",
+                    "content": "question",
+                    "ephemeral": True,
+                }
+            ]
         )
         # session end retries the kept turn
         await drv.command("voice/end")
@@ -172,7 +186,7 @@ async def test_commit_failure_retries_in_order_with_the_next_turn():
         ]
 
 
-async def test_end_drops_in_progress_turns():
+async def test_end_commits_in_progress_turns_with_partial_text():
     bench = Bench()
     async with voice_host(bench) as (drv, _app):
         await drv.command("voice/start", {"offer": OFFER})
@@ -185,15 +199,41 @@ async def test_end_drops_in_progress_turns():
         rsp = await drv.command("voice/end")
         assert rsp["type"] == "accepted", rsp
         assert bench.provider.closed
-        assert bench.commits == [[{"id": "t1", "role": "user", "text": "question"}]]
+        assert bench.commits == [
+            [{"id": "t1", "role": "user", "text": "question"}],
+            [{"id": "t2", "role": "assistant", "text": "half an ans"}],
+        ]
+        # the durable entries replaced the ephemeral ones under the same ids
         await drv.wait(
             lambda s: voice_of(s) is None
             and messages_of(s)
-            == [{"id": "t1", "type": "human", "content": "question"}]
+            == [
+                {"id": "t1", "type": "human", "content": "question"},
+                {"id": "t2", "type": "ai", "content": "half an ans"},
+            ]
         )
 
 
-async def test_end_with_no_done_turns_commits_nothing():
+async def test_end_commits_partial_turns_after_kept_done_turns():
+    bench = Bench(commit_failures=[RuntimeError("checkpointer down")])
+    async with voice_host(bench) as (drv, _app):
+        await drv.command("voice/start", {"offer": OFFER})
+        events = bench.events()
+        events.turn_started("t1", "user")
+        events.turn_done("t1", "question")
+        await until(lambda: bench.commit_failures == [])
+        events.turn_started("t2", "assistant")
+        events.turn_delta("t2", "half an ans")
+        await drv.command("voice/end")
+        assert bench.commits == [
+            [
+                {"id": "t1", "role": "user", "text": "question"},
+                {"id": "t2", "role": "assistant", "text": "half an ans"},
+            ]
+        ]
+
+
+async def test_end_drops_started_turns_with_no_text():
     bench = Bench()
     async with voice_host(bench) as (drv, _app):
         await drv.command("voice/start", {"offer": OFFER})
@@ -220,9 +260,14 @@ async def test_provider_closed_event_ends_and_commits():
         events = bench.events()
         events.turn_started("t1", "assistant")
         events.turn_done("t1", "goodbye")
+        events.turn_started("t2", "user")
+        events.turn_delta("t2", "wai")
         events.closed("server")
         await drv.wait(lambda s: voice_of(s) is None)
-        assert bench.commits == [[{"id": "t1", "role": "assistant", "text": "goodbye"}]]
+        assert bench.commits == [
+            [{"id": "t1", "role": "assistant", "text": "goodbye"}],
+            [{"id": "t2", "role": "user", "text": "wai"}],
+        ]
         assert_rejected(await drv.command("voice/end"), "no-voice")
 
 

{harness_sdk_python-0.29.0 → harness_sdk_python-0.29.2}/tests/voice_helpers.py

@@ -83,7 +83,11 @@ def make_voice_host(bench: Bench, *, declare: bool = True):
             durable_ids = {m["id"] for m in self._durable}
             self.state["messages"] = [
                 *self._durable,
-                *(e for i, e in self._ephemeral.items() if i not in durable_ids),
+                *(
+                    {**e, "ephemeral": True}
+                    for i, e in self._ephemeral.items()
+                    if i not in durable_ids
+                ),
             ]
 
         async def get_voice_seed(self):