langgraph-runtime-inmem 0.36.0.dev2__tar.gz → 0.36.0.dev4__tar.gz

{langgraph_runtime_inmem-0.36.0.dev2 → langgraph_runtime_inmem-0.36.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.36.0.dev2
+Version: 0.36.0.dev4
 Summary: Inmem implementation for the LangGraph API server.
 Author-email: Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0

{langgraph_runtime_inmem-0.36.0.dev2 → langgraph_runtime_inmem-0.36.0.dev4}/langgraph_runtime_inmem/__init__.py

@@ -10,7 +10,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.36.0.dev2"
+__version__ = "0.36.0.dev4"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.36.0.dev2 → langgraph_runtime_inmem-0.36.0.dev4}/langgraph_runtime_inmem/inmem_stream.py

@@ -26,6 +26,8 @@ class Message:
     topic: bytes
     data: bytes
     id: bytes | None = None
+    # Control messages only: server-generated explanation of an interrupt/rollback
+    reason: str | None = None
 
 
 class ContextQueue(asyncio.Queue):

{langgraph_runtime_inmem-0.36.0.dev2 → langgraph_runtime_inmem-0.36.0.dev4}/langgraph_runtime_inmem/ops.py

@@ -620,6 +620,7 @@ class Assistants(Authenticated):
                 assistant_id=assistant_id,
                 action="interrupt",
                 ctx=ctx,
+                reason=f"assistant {assistant_id} was deleted",
             )
 
             # 4. Delete assistant
@@ -2858,6 +2859,7 @@ class Runs(Authenticated):
         status: Literal["pending", "running", "all"] | None = None,
         assistant_id: UUID | None = None,
         ctx: Auth.types.BaseAuthContext | None = None,
+        reason: str | None = None,
     ) -> None:
         """
         Cancel runs in memory. Must provide either:
@@ -2971,6 +2973,7 @@ class Runs(Authenticated):
             control_message = Message(
                 topic=f"run:{run_id}:control".encode(),
                 data=action.encode(),
+                reason=reason,
             )
             coros.append(stream_manager.put(run_id, thread_id, control_message))
 
@@ -3187,7 +3190,8 @@ class Runs(Authenticated):
                                 # We only return a stream ID if the run is resumable
                                 stream_id = (
                                     id
-                                    if run.get("kwargs", {}).get("resumable")
+                                    if run is not None
+                                    and run.get("kwargs", {}).get("resumable")
                                     else None
                                 )
                                 yield mode.encode(), payload, stream_id
@@ -3224,7 +3228,13 @@ class Runs(Authenticated):
                 raise e.http_exception from None
             except:
                 if cancel_on_disconnect:
-                    create_task(cancel_run(thread_id, run_id))
+                    create_task(
+                        cancel_run(
+                            thread_id,
+                            run_id,
+                            reason="client disconnected from the run stream (on_disconnect=cancel)",
+                        )
+                    )
                 raise
             finally:
                 stream_manager = get_stream_manager()
@@ -3279,25 +3289,27 @@ async def listen_for_cancellation(
 
     stream_manager = get_stream_manager()
 
+    def _handle_control_message(message: Message) -> bool:
+        """Set `done` from a control message. Returns True once the run is done."""
+        reason = message.reason
+        if message.data == b"rollback":
+            done.set(UserRollback(reason) if reason else UserRollback())
+        elif message.data == b"interrupt":
+            done.set(UserInterrupt(reason) if reason else UserInterrupt())
+        elif message.data == b"done":
+            done.set()
+            return True
+        return False
+
     if control_key := stream_manager.get_control_key(run_id, thread_id):
-        payload = control_key.data
-        if payload == b"rollback":
-            done.set(UserRollback())
-        elif payload == b"interrupt":
-            done.set(UserInterrupt())
+        _handle_control_message(control_key)
 
     while not done.is_set():
         try:
             # This task gets cancelled when Runs.enter exits anyway,
             # so we can have a pretty lengthy timeout here
             message = await asyncio.wait_for(queue.get(), timeout=240)
-            payload = message.data
-            if payload == b"rollback":
-                done.set(UserRollback())
-            elif payload == b"interrupt":
-                done.set(UserInterrupt())
-            elif payload == b"done":
-                done.set()
+            if _handle_control_message(message):
                 break
         except TimeoutError:
             break
@@ -3896,10 +3908,13 @@ class Crons(Authenticated):
 
 
 async def cancel_run(
-    thread_id: UUID, run_id: UUID, ctx: Auth.types.BaseAuthContext | None = None
+    thread_id: UUID,
+    run_id: UUID,
+    ctx: Auth.types.BaseAuthContext | None = None,
+    reason: str | None = None,
 ) -> None:
     async with connect() as conn:
-        await Runs.cancel(conn, [run_id], thread_id=thread_id, ctx=ctx)
+        await Runs.cancel(conn, [run_id], thread_id=thread_id, ctx=ctx, reason=reason)
 
 
 async def _get_checkpointer(