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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: harness-sdk-python
-Version: 0.29.1
+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.1 → harness_sdk_python-0.29.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "harness-sdk-python"
-version = "0.29.1"
+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.1 → harness_sdk_python-0.29.2}/src/harness_sdk/voice.py

@@ -18,7 +18,8 @@ events render live in ``state["messages"]`` through
 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),
@@ -189,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.1 → harness_sdk_python-0.29.2}/tests/test_voice.py

@@ -186,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})
@@ -199,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})
@@ -234,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")