langgraph-runtime-inmem 0.37.0.dev4__tar.gz → 0.37.0.dev5__tar.gz

{langgraph_runtime_inmem-0.37.0.dev4 → langgraph_runtime_inmem-0.37.0.dev5}/PKG-INFO

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

{langgraph_runtime_inmem-0.37.0.dev4 → langgraph_runtime_inmem-0.37.0.dev5}/langgraph_runtime_inmem/__init__.py

@@ -11,7 +11,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.37.0.dev4"
+__version__ = "0.37.0.dev5"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.37.0.dev4 → langgraph_runtime_inmem-0.37.0.dev5}/langgraph_runtime_inmem/ops.py

@@ -209,7 +209,8 @@ async def _save_thread_status(
 
 def _run_stream_mode_matches(event_mode: str, stream_mode: list[str] | None) -> bool:
     """Return True if a published run-stream event matches the join filter."""
-    if not stream_mode:
+    # A run's error is its outcome, not a selectable mode, so no filter drops it.
+    if not stream_mode or event_mode == "error":
         return True
     if event_mode in stream_mode:
         return True
@@ -3178,15 +3179,19 @@ class Runs(Authenticated):
                 data=action.encode(),
                 reason=reason,
             )
-            coros.append(stream_manager.put(run_id, thread_id, control_message))
+            # Cancels by status or assistant have no thread_id; the worker listens on the run's own.
+            coros.append(stream_manager.put(run_id, run["thread_id"], control_message))
 
-            queues = stream_manager.get_queues(run_id, thread_id)
+            queues = stream_manager.get_queues(run_id, run["thread_id"])
 
             if run["status"] in ("pending", "running"):
                 cancelable_runs.append(run)
                 if run["status"] == "pending":
                     pending_thread_ids.add(run["thread_id"])
-                if queues or action != "rollback":
+                if action == "rollback" and run["status"] == "running":
+                    # Its worker finalizes the rollback; staying "running" holds the thread's next run until then.
+                    pass
+                elif queues or action != "rollback":
                     run["status"] = "interrupted"
                     run["cancel_reason"] = reason
                     run["updated_at"] = datetime.now(tz=UTC)
@@ -3194,7 +3199,6 @@ class Runs(Authenticated):
                     await logger.ainfo(
                         "Eagerly deleting pending run with rollback action",
                         run_id=str(run_id),
-                        status=run["status"],
                     )
                     coros.append(Runs.delete(conn, run_id, thread_id=run["thread_id"]))
             else: