langgraph-runtime-inmem 0.17.1__tar.gz → 0.18.0__tar.gz

{langgraph_runtime_inmem-0.17.1 → langgraph_runtime_inmem-0.18.0}/PKG-INFO

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

{langgraph_runtime_inmem-0.17.1 → langgraph_runtime_inmem-0.18.0}/langgraph_runtime_inmem/__init__.py

@@ -9,7 +9,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.17.1"
+__version__ = "0.18.0"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.17.1 → langgraph_runtime_inmem-0.18.0}/langgraph_runtime_inmem/ops.py

@@ -1700,7 +1700,9 @@ class Threads(Authenticated):
 
             return []
 
-    class Stream:
+    class Stream(Authenticated):
+        resource = "threads"
+
         @staticmethod
         async def subscribe(
             conn: InMemConnectionProto | AsyncConnectionProto,
@@ -1736,8 +1738,11 @@ class Threads(Authenticated):
             *,
             last_event_id: str | None = None,
             stream_modes: list[ThreadStreamMode],
+            ctx: Auth.types.BaseAuthContext | None = None,
         ) -> AsyncIterator[tuple[bytes, bytes, bytes | None]]:
             """Stream the thread output."""
+            await Threads.Stream.check_thread_stream_auth(thread_id, ctx)
+
             from langgraph_api.utils.stream_codec import (
                 decode_stream_message,
             )
@@ -1889,6 +1894,24 @@ class Threads(Authenticated):
                 str(thread_id), Message(topic=topic, data=payload)
             )
 
+        @staticmethod
+        async def check_thread_stream_auth(
+            thread_id: UUID,
+            ctx: Auth.types.BaseAuthContext | None = None,
+        ) -> None:
+            async with connect() as conn:
+                filters = await Threads.Stream.handle_event(
+                    ctx,
+                    "read",
+                    Auth.types.ThreadsRead(thread_id=thread_id),
+                )
+                if filters:
+                    thread = await Threads._get_with_filters(
+                        cast(InMemConnectionProto, conn), thread_id, filters
+                    )
+                    if not thread:
+                        raise HTTPException(status_code=404, detail="Thread not found")
+
     @staticmethod
     async def count(
         conn: InMemConnectionProto,