langgraph-runtime-inmem 0.6.4__py3-none-any.whl → 0.6.6__py3-none-any.whl

langgraph_runtime_inmem/__init__.py

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

langgraph_runtime_inmem/database.py

@@ -182,6 +182,8 @@ async def start_pool() -> None:
         for a in GLOBAL_STORE["assistants"]:
             if a["metadata"].get("created_by") == "system":
                 GLOBAL_STORE["assistants"].remove(a)
+            if "context" not in a:
+                a["context"] = {}
     for k in ["crons"]:
         if not GLOBAL_STORE.get(k):
             GLOBAL_STORE[k] = {}

langgraph_runtime_inmem/ops.py

@@ -378,7 +378,7 @@ class Assistants(Authenticated):
             "version": new_version,
             "graph_id": graph_id if graph_id is not None else assistant["graph_id"],
             "config": config if config is not None else assistant["config"],
-            "context": context if context is not None else assistant["context"],
+            "context": context if context is not None else assistant.get("context", {}),
             "metadata": metadata if metadata is not None else assistant["metadata"],
             "created_at": now,
             "name": name if name is not None else assistant["name"],
@@ -1958,10 +1958,16 @@ class Runs(Authenticated):
         # wait for the run to complete
         # Rely on this join's auth
         async for mode, chunk, _ in Runs.Stream.join(
-            run_id, thread_id=thread_id, ctx=ctx, ignore_404=True
+            run_id,
+            thread_id=thread_id,
+            ctx=ctx,
+            ignore_404=True,
+            stream_mode=["values", "updates", "error"],
         ):
             if mode == b"values":
                 last_chunk = chunk
+            elif mode == b"updates" and b"__interrupt__" in chunk:
+                last_chunk = chunk
             elif mode == b"error":
                 last_chunk = orjson.dumps({"__error__": orjson.Fragment(chunk)})
         # if we received a final chunk, return it
@@ -2217,16 +2223,20 @@ class Runs(Authenticated):
             thread_id: UUID,
             ignore_404: bool = False,
             cancel_on_disconnect: bool = False,
-            stream_mode: StreamMode | asyncio.Queue | None = None,
+            stream_channel: asyncio.Queue | None = None,
+            stream_mode: list[StreamMode] | StreamMode,
             last_event_id: str | None = None,
             ctx: Auth.types.BaseAuthContext | None = None,
         ) -> AsyncIterator[tuple[bytes, bytes, bytes | None]]:
             """Stream the run output."""
             from langgraph_api.asyncio import create_task
 
+            if stream_mode and not isinstance(stream_mode, list):
+                stream_mode = [stream_mode]
+
             queue = (
-                stream_mode
-                if isinstance(stream_mode, asyncio.Queue)
+                stream_channel
+                if stream_channel
                 else await Runs.Stream.subscribe(run_id, stream_mode=stream_mode)
             )
 
@@ -2247,6 +2257,7 @@ class Runs(Authenticated):
                                     status_code=404, detail="Thread not found"
                                 )
                             )
+                    run = await Runs.get(conn, run_id, thread_id=thread_id, ctx=ctx)
                     channel_prefix = f"run:{run_id}:stream:"
                     len_prefix = len(channel_prefix.encode())
 
@@ -2258,14 +2269,18 @@ class Runs(Authenticated):
                             if data == b"done":
                                 return
                         else:
-                            yield topic[len_prefix:], data, id
-                            logger.debug(
-                                "Replayed run event",
-                                run_id=str(run_id),
-                                message_id=id,
-                                stream_mode=topic[len_prefix:],
-                                data=data,
-                            )
+                            mode = topic[len_prefix:]
+                            if mode == b"updates" and "updates" not in stream_mode:
+                                continue
+                            else:
+                                yield mode, data, id
+                                logger.debug(
+                                    "Replayed run event",
+                                    run_id=str(run_id),
+                                    message_id=id,
+                                    stream_mode=mode,
+                                    data=data,
+                                )
 
                     while True:
                         try:
@@ -2278,14 +2293,18 @@ class Runs(Authenticated):
                                     break
                             else:
                                 # Extract mode from topic
-                                yield topic[len_prefix:], data, id
-                                logger.debug(
-                                    "Streamed run event",
-                                    run_id=str(run_id),
-                                    stream_mode=topic[len_prefix:],
-                                    message_id=id,
-                                    data=data,
-                                )
+                                mode = topic[len_prefix:]
+                                if mode == b"updates" and "updates" not in stream_mode:
+                                    continue
+                                else:
+                                    yield mode, data, id
+                                    logger.debug(
+                                        "Streamed run event",
+                                        run_id=str(run_id),
+                                        stream_mode=mode,
+                                        message_id=id,
+                                        data=data,
+                                    )
                         except TimeoutError:
                             # Check if the run is still pending
                             run_iter = await Runs.get(

{langgraph_runtime_inmem-0.6.4.dist-info → langgraph_runtime_inmem-0.6.6.dist-info}/METADATA

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

{langgraph_runtime_inmem-0.6.4.dist-info → langgraph_runtime_inmem-0.6.6.dist-info}/RECORD

@@ -1,13 +1,13 @@
-langgraph_runtime_inmem/__init__.py,sha256=8jTuoC-OAbGipvvHN2ZvFRwD7HOFnZxWa616pD7txnY,310
+langgraph_runtime_inmem/__init__.py,sha256=KDmH2YP4H44U-xjgCT_hgTMK9eR-VgmkmI6wGA-0k4g,310
 langgraph_runtime_inmem/checkpoint.py,sha256=nc1G8DqVdIu-ibjKTqXfbPfMbAsKjPObKqegrSzo6Po,4432
-langgraph_runtime_inmem/database.py,sha256=bOPc9cgHysMnJ_RbRpU0yOLxK4ERL9et7artgTBEecY,6280
+langgraph_runtime_inmem/database.py,sha256=G_6L2khpRDSpS2Vs_SujzHayODcwG5V2IhFP7LLBXgw,6349
 langgraph_runtime_inmem/inmem_stream.py,sha256=65z_2mBNJ0-yJsXWnlYwRc71039_y6Sa0MN8fL_U3Ko,4581
 langgraph_runtime_inmem/lifespan.py,sha256=t0w2MX2dGxe8yNtSX97Z-d2pFpllSLS4s1rh2GJDw5M,3557
 langgraph_runtime_inmem/metrics.py,sha256=HhO0RC2bMDTDyGBNvnd2ooLebLA8P1u5oq978Kp_nAA,392
-langgraph_runtime_inmem/ops.py,sha256=6ZI-PUr1wfJoSoJ4LOn-dsAuIOTPkEODwvK1g4O1OZc,88538
+langgraph_runtime_inmem/ops.py,sha256=QHkoJiiKoKuMTNKlkpApxNsDhiR2CtlcQDR-zraTuew,89412
 langgraph_runtime_inmem/queue.py,sha256=nqfgz7j_Jkh5Ek5-RsHB2Uvwbxguu9IUPkGXIxvFPns,10037
 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.6.4.dist-info/METADATA,sha256=JyDtIIEb7c6nSx6xBUNUTBx5ZuYhGWT0hlzfog4Q5PU,565
-langgraph_runtime_inmem-0.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_runtime_inmem-0.6.4.dist-info/RECORD,,
+langgraph_runtime_inmem-0.6.6.dist-info/METADATA,sha256=hkFJC5S9J-g-c1eGRwLWUh1nsMCE4omusGOeMIKar8M,565
+langgraph_runtime_inmem-0.6.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_runtime_inmem-0.6.6.dist-info/RECORD,,