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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.38.0.dev2
+Version: 0.38.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.38.0.dev2 → langgraph_runtime_inmem-0.38.0.dev4}/langgraph_runtime_inmem/__init__.py

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

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

@@ -2296,9 +2296,16 @@ class Threads(Authenticated):
             last_event_id: str | None = None,
             stream_modes: list[ThreadStreamMode],
             ctx: Auth.types.BaseAuthContext | None = None,
+            auth_filters: Auth.types.FilterType | None = None,
         ) -> AsyncIterator[tuple[bytes, bytes, bytes | None, str | None]]:
-            """Stream the thread output."""
-            await Threads.Stream.check_thread_stream_auth(thread_id, ctx)
+            """Stream the thread output.
+
+            ``auth_filters`` comes from a ``check_thread_stream_auth`` the
+            caller already ran, so the read handler doesn't run twice. With
+            ``None``, the check runs here.
+            """
+            if auth_filters is None:
+                await Threads.Stream.check_thread_stream_auth(thread_id, ctx)
 
             from langgraph_api.utils.stream_codec import (  # noqa: PLC0415
                 decode_stream_message,
@@ -2498,7 +2505,11 @@ class Threads(Authenticated):
         async def check_thread_stream_auth(
             thread_id: UUID,
             ctx: Auth.types.BaseAuthContext | None = None,
-        ) -> None:
+        ) -> Auth.types.FilterType:
+            """Raise 403/404 for a thread the caller can't read.
+
+            Returns the auth filters for ``join_event_streaming``.
+            """
             async with connect() as conn:
                 filters = await Threads.Stream.handle_event(
                     ctx,
@@ -2511,6 +2522,7 @@ class Threads(Authenticated):
                     )
                     if not thread:
                         raise HTTPException(status_code=404, detail="Thread not found")
+            return filters or {}
 
     @staticmethod
     async def count(