harness-sdk-python 0.36.4__tar.gz → 0.36.5__tar.gz

{harness_sdk_python-0.36.4 → harness_sdk_python-0.36.5}/PKG-INFO

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

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

{harness_sdk_python-0.36.4 → harness_sdk_python-0.36.5}/src/harness_sdk/run_manager.py

@@ -455,6 +455,12 @@ class RunManager(Tray):
                 self._replay_settle()
                 self._outcome = settled
         recorded = self.outbox.value
+        if recorded is not None and not recorded["item"].get("durableAdmission", False):
+            self.steerQueue.untake(self.confirmedTakes.value)
+            self.confirmedTakes.value = 0
+            self.batchApplied.value = False
+            recorded = {**recorded, "item": {**recorded["item"], "durableAdmission": True}}
+            self.outbox.value = recorded
         if recorded is not None and not self.batchApplied.value:
             record = recorded["item"]["record"]
             lanes = recorded["item"]["lanes"]
@@ -801,7 +807,11 @@ class RunManager(Tray):
             root_meta=root_meta,
             input_outcomes=input_outcomes,
         )
-        item = {"record": record, "lanes": [lane for lane, _ in self._batch]}
+        item = {
+            "record": record,
+            "lanes": [lane for lane, _ in self._batch],
+            "durableAdmission": True,
+        }
         self.kick()
         self.stopReason.value = None
         self.epoch.value = self.epoch.value + 1

{harness_sdk_python-0.36.4 → harness_sdk_python-0.36.5}/tests/test_run_admission.py

@@ -144,3 +144,52 @@ async def test_unconfirmed_rewind_replacement_can_be_enqueued_after_stop():
         assert [m["id"] for m in retry.ctx.messages] == ["m2"]
         retry.applied()
         retry.finish(RunManager.Complete())
+
+
+@pytest.mark.parametrize("takes", [0, 2])
+@pytest.mark.parametrize("checkpoint", [None, {"graph_input_checkpointed": True}])
+async def test_pre_split_records_replay_inputs_and_restore_confirmed_steers(
+    takes, checkpoint
+):
+    store = MemoryStateStore()
+    script = Script()
+    host_type = make_host(script)
+    async with instance(host_type, state_store=store) as host:
+        await host.runs.enqueue(
+            add("m1"), meta={"author": "first"}, applied=lambda: None, auth_result=None
+        )
+        call = await script.next_call()
+        call.ctx.set_state(checkpoint)
+        call.applied()
+        for index in range(takes):
+            await host.runs.steer(
+                add(f"s{index}"),
+                meta={"author": f"author{index}"},
+                applied=lambda: None,
+                auth_result=None,
+            )
+        if takes:
+            call.ctx.steering.take()
+            call.applied()
+        await host.frontier.wait()
+    for _, doc in store.trays.values():
+        if "outbox" in doc:
+            doc["outbox"]["item"].pop("durableAdmission")
+
+    async with instance(host_type, state_store=store) as host:
+        replay = await script.next_call()
+        assert not replay.ctx.resume
+        assert [m["id"] for m in replay.ctx.messages] == ["m1"]
+        assert replay.ctx.meta("m1") == {"author": "first"}
+        replay.applied()
+        if takes:
+            assert [m["id"] for m in replay.ctx.steering.take()] == ["s0", "s1"]
+            assert replay.ctx.meta("s0") == {"author": "author0"}
+            assert replay.ctx.meta("s1") == {"author": "author1"}
+            replay.applied()
+        await host.frontier.wait()
+    async with instance(host_type, state_store=store):
+        confirmed = await script.next_call()
+        assert confirmed.ctx.resume
+        assert confirmed.ctx.steering.take() == ()
+        confirmed.finish(RunManager.Complete())