langgraph-runtime-inmem 0.17.1__py3-none-any.whl → 0.18.0__py3-none-any.whl

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/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,

{langgraph_runtime_inmem-0.17.1.dist-info → langgraph_runtime_inmem-0.18.0.dist-info}/METADATA

@@ -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.dist-info → langgraph_runtime_inmem-0.18.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
-langgraph_runtime_inmem/__init__.py,sha256=xWciGBAIjNSLumyTQszfz9Bq-MYAzM9BGJ1nYR8feao,311
+langgraph_runtime_inmem/__init__.py,sha256=0OJlzkajmSaVNDTxR81HA_IpgMleuq6KhszFWEohtGM,311
 langgraph_runtime_inmem/checkpoint.py,sha256=nc1G8DqVdIu-ibjKTqXfbPfMbAsKjPObKqegrSzo6Po,4432
 langgraph_runtime_inmem/database.py,sha256=g2XYa5KN-T8MbDeFH9sfUApDG62Wp4BACumVnDtxYhI,6403
 langgraph_runtime_inmem/inmem_stream.py,sha256=PFLWbsxU8RqbT5mYJgNk6v5q6TWJRIY1hkZWhJF8nkI,9094
 langgraph_runtime_inmem/lifespan.py,sha256=fCoYcN_h0cxmj6-muC-f0csPdSpyepZuGRD1yBrq4XM,4755
 langgraph_runtime_inmem/metrics.py,sha256=_YiSkLnhQvHpMktk38SZo0abyL-5GihfVAtBo0-lFIc,403
-langgraph_runtime_inmem/ops.py,sha256=T-q2OxrK_Uw7NL9ac-B8VeZg_G-fcpBGNDtvcCBzLrs,108207
+langgraph_runtime_inmem/ops.py,sha256=umfHzcvGX8hdzAxYryeX5doCyA9HnfwEeEPupj7Ckyg,109111
 langgraph_runtime_inmem/queue.py,sha256=yV781CDjlsKNvkJ3puHyiHNnJpIIdB1G_gTY9dg6mys,9994
 langgraph_runtime_inmem/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
 langgraph_runtime_inmem/store.py,sha256=rTfL1JJvd-j4xjTrL8qDcynaWF6gUJ9-GDVwH0NBD_I,3506
-langgraph_runtime_inmem-0.17.1.dist-info/METADATA,sha256=pn-tX9xWNJbxajNmU5bD-RTwer6Mg4wCLEIlNKXWe8E,570
-langgraph_runtime_inmem-0.17.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_runtime_inmem-0.17.1.dist-info/RECORD,,
+langgraph_runtime_inmem-0.18.0.dist-info/METADATA,sha256=oigJEnL7JdWwUoUK6pos7aCx9PZaDbIXyzd2RaTqyR4,570
+langgraph_runtime_inmem-0.18.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_runtime_inmem-0.18.0.dist-info/RECORD,,