harness-sdk-python 0.36.0__py3-none-any.whl → 0.36.2__py3-none-any.whl

harness_sdk/run_manager.py

@@ -67,6 +67,7 @@ import logging
 from dataclasses import dataclass, field
 from typing import Any, AsyncIterator, Callable, Coroutine, Iterable, Protocol
 
+from pinned import EvictionRequested
 from pinned.trays import Tray, TrayDescriptor, TrayCtx, TrayMapNamespace, inbox
 from pinned.trays import persistent, persistent_outbox, persistent_queue
 from statepatch import observable
@@ -845,6 +846,12 @@ class RunManager(Tray):
 
     def _make_run_ctx(self, item: Any) -> "RunManager.RunContext":
         record = item["record"]
+        root_meta = record.get("rootMeta")
+        if "rootMeta" not in record:
+            if record["messages"]:
+                root_meta = record["messages"][0].get("meta")
+            elif record.get("inputOutcomes"):
+                root_meta = record["inputOutcomes"][0]["meta"]
         messages = [
             {k: v for k, v in message.items() if k not in ("meta", "~admit")}
             for message in record["messages"]
@@ -857,7 +864,7 @@ class RunManager(Tray):
             _manager=self,
             _rollback_to=record["rollbackTo"] if "rollbackTo" in record else _ABSENT,
             _message_meta={m["id"]: m.get("meta") for m in record["messages"]},
-            _root_meta=record.get("rootMeta"),
+            _root_meta=root_meta,
             _input_outcomes=tuple(
                 (o["request"], o["response"], o["meta"])
                 for o in record.get("inputOutcomes", ())
@@ -877,6 +884,8 @@ class RunManager(Tray):
         self._sync_steering()
         try:
             await task
+        except EvictionRequested:
+            raise
         except asyncio.CancelledError:
             current = asyncio.current_task()
             if current is not None and current.cancelling():
@@ -886,6 +895,8 @@ class RunManager(Tray):
             # user cancel: _execute already converged on the settle discipline
         finally:
             self._active = None
+            if self._run_live is ctx:
+                self._run_live = None
 
     async def _execute(self, ctx: "RunManager.RunContext", fn: Any) -> None:
         try:
@@ -905,34 +916,25 @@ class RunManager(Tray):
                 )
             if isinstance(outcome, RunManager.Complete) and self.dispatching.value is not None:
                 raise RuntimeError("run returned Complete with an unconfirmed batch")
+        except EvictionRequested:
+            self._advance()
+            if self._stop_staged or ctx._tray_ctx.consumed:
+                self._cancel_run(ctx)
+                return
+            raise
         except asyncio.CancelledError:
-            if self._closing or self._retirement is not None:
-                raise  # teardown: the un-acked outbox redelivers next activation
-            self.outcome.value = {
-                "kind": "cancel",
-                "epoch": self.epoch.value,
-                "dispatchQueue": False,
-            }
-            self._settle(ctx)
-            fx = self._apply_intake()
-            rewind = self._staged_rewind
-            if rewind is not None and not rewind.applied_sent:
-                # park the in-drain rewind durably; a cancel starts no dispatch
-                self.nextDispatch.value = self._record_of(
-                    rewind.type,
-                    list(rewind.messages),
-                    rollback_to=rewind.rollback_to,
-                    root_meta=rewind.root_meta,
+            if (
+                self._closing
+                or self._retirement is not None
+                or (
+                    self._evicting()
+                    and not self._stop_staged
+                    and not ctx._tray_ctx.consumed
                 )
-                rewind.applied()
-                rewind.applied_sent = True
-            self.halted.value = "stop"
-            self.outcome.value = None
-            self._resolve_stop_waiters()
-            self._publish_status()
-            self._settle_applied(fx)
-            self._idle.set()
-            raise  # staged intake settled; a cancelled run starts no dispatch
+            ):
+                raise  # teardown: the un-acked outbox redelivers next activation
+            self._cancel_run(ctx)
+            raise
         except Exception as exc:
             if not isinstance(exc, StatewireReject):
                 logger.exception(
@@ -957,6 +959,31 @@ class RunManager(Tray):
         self._outcome = self.outcome.value
         self._advance()
 
+    def _cancel_run(self, ctx: "RunManager.RunContext") -> None:
+        self.outcome.value = {
+            "kind": "cancel",
+            "epoch": self.epoch.value,
+            "dispatchQueue": False,
+        }
+        self._settle(ctx)
+        fx = self._apply_intake()
+        rewind = self._staged_rewind
+        if rewind is not None and not rewind.applied_sent:
+            self.nextDispatch.value = self._record_of(
+                rewind.type,
+                list(rewind.messages),
+                rollback_to=rewind.rollback_to,
+                root_meta=rewind.root_meta,
+            )
+            rewind.applied()
+            rewind.applied_sent = True
+        self.halted.value = "stop"
+        self.outcome.value = None
+        self._resolve_stop_waiters()
+        self._publish_status()
+        self._settle_applied(fx)
+        self._idle.set()
+
     def _maybe_self_delete(self) -> None:
         """A settled machine with no durable obligations deletes its tray; the
         next send recreates it fresh."""

{harness_sdk_python-0.36.0.dist-info → harness_sdk_python-0.36.2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.5
 Name: harness-sdk-python
-Version: 0.36.0
+Version: 0.36.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
 Requires-Python: <4.0,>=3.12
-Requires-Dist: pinned<0.16,>=0.15.0
-Requires-Dist: statewire<0.17,>=0.16.0
+Requires-Dist: pinned<0.16,>=0.15.1
+Requires-Dist: statewire<0.18,>=0.17.2
 Provides-Extra: deepagents
 Requires-Dist: deepagents>=0.6.12; extra == 'deepagents'
 Requires-Dist: langchain-core>=0.3; extra == 'deepagents'

{harness_sdk_python-0.36.0.dist-info → harness_sdk_python-0.36.2.dist-info}/RECORD

@@ -4,8 +4,8 @@ harness_sdk/deepagents.py,sha256=3iWadYjG-Yv0fli4x8s5n3jOTs11JHS2cK6tj8foyvk,145
 harness_sdk/fenced_postgres.py,sha256=hbbrMR1WaqrHTNgd4wFrXfRrzQFGnyim-c8nZXgwWp4,13235
 harness_sdk/linear_thread.py,sha256=laDRXKLFsMlyWS4snqNAM6t38wO26gevLWXRNivhHSM,1810
 harness_sdk/protocol.py,sha256=4diqaWeYIVEKRjyKkqzeMMHJb7zrOQCvPkjQzkeIku8,140
-harness_sdk/run_manager.py,sha256=9yfxzelb5Ugs3qbsKiWDPfASljjlc0BbmoQv-8Bw72Y,81438
+harness_sdk/run_manager.py,sha256=gcd4nfXJ8ssD-v0yP2FF4WsUE-xWzgrChdG4RQV_UqA,82165
 harness_sdk/voice.py,sha256=bRloaKSS1d0mFvfIM4Bhp50ADBK_7GpdarGvmDBubjg,32370
-harness_sdk_python-0.36.0.dist-info/METADATA,sha256=rDfjbNchuYQEZzstgtiaspw-lFXIvBChAHPYSOH0s1E,2488
-harness_sdk_python-0.36.0.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-harness_sdk_python-0.36.0.dist-info/RECORD,,
+harness_sdk_python-0.36.2.dist-info/METADATA,sha256=aJPyd884khPDjcycx20eL8ZrLZSw1voyTkoIEajTCKs,2488
+harness_sdk_python-0.36.2.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+harness_sdk_python-0.36.2.dist-info/RECORD,,