langgraph-runtime-inmem 0.32.0.dev15__tar.gz → 0.32.0.dev16__tar.gz

{langgraph_runtime_inmem-0.32.0.dev15 → langgraph_runtime_inmem-0.32.0.dev16}/PKG-INFO

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

{langgraph_runtime_inmem-0.32.0.dev15 → langgraph_runtime_inmem-0.32.0.dev16}/langgraph_runtime_inmem/__init__.py

@@ -10,7 +10,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.32.0.dev15"
+__version__ = "0.32.0.dev16"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.32.0.dev15 → langgraph_runtime_inmem-0.32.0.dev16}/langgraph_runtime_inmem/ops.py

@@ -75,6 +75,23 @@ LANGGRAPH_PY_MINOR = tuple(map(int, __version__.split(".")[:2]))
 USE_NEW_INTERRUPTS = LANGGRAPH_PY_MINOR >= (0, 6)
 
 
+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:
+        return True
+    if event_mode in stream_mode:
+        return True
+    if (
+        "messages" in stream_mode or "messages-tuple" in stream_mode
+    ) and event_mode.startswith("messages"):
+        return True
+    if "|" in event_mode:
+        base_mode, _, _ = event_mode.partition("|")
+        if base_mode in stream_mode:
+            return True
+    return False
+
+
 def _ensure_uuid(id_: str | uuid.UUID | None) -> uuid.UUID:
     if isinstance(id_, str):
         return uuid.UUID(id_)
@@ -3036,17 +3053,7 @@ class Runs(Authenticated):
                         if mode == "control":
                             if payload == b"done":
                                 return
-                        elif (
-                            not stream_mode
-                            or mode in stream_mode
-                            or (
-                                (
-                                    "messages" in stream_mode
-                                    or "messages-tuple" in stream_mode
-                                )
-                                and mode.startswith("messages")
-                            )
-                        ):
+                        elif _run_stream_mode_matches(mode, stream_mode):
                             yield mode.encode(), payload, id
                             logger.debug(
                                 "Replayed run event",
@@ -3068,17 +3075,7 @@ class Runs(Authenticated):
                             if mode == "control":
                                 if payload == b"done":
                                     break
-                            elif (
-                                not stream_mode
-                                or mode in stream_mode
-                                or (
-                                    (
-                                        "messages" in stream_mode
-                                        or "messages-tuple" in stream_mode
-                                    )
-                                    and mode.startswith("messages")
-                                )
-                            ):
+                            elif _run_stream_mode_matches(mode, stream_mode):
                                 # We only return a stream ID if the run is resumable
                                 stream_id = (
                                     id