langgraph-runtime-inmem 0.6.11__py3-none-any.whl → 0.6.13__py3-none-any.whl

langgraph_runtime_inmem/__init__.py

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

langgraph_runtime_inmem/ops.py

@@ -1001,6 +1001,7 @@ class Threads(Authenticated):
         thread_id: UUID,
         run_id: UUID,
         run_status: RunStatus | Literal["rollback"],
+        graph_id: str,
         checkpoint: CheckpointPayload | None = None,
         exception: BaseException | None = None,
     ) -> None:
@@ -1080,6 +1081,7 @@ class Threads(Authenticated):
                 final_thread_status = "busy"
             else:
                 final_thread_status = base_thread_status
+        thread["metadata"]["graph_id"] = graph_id
         thread.update(
             {
                 "updated_at": now,
@@ -1273,7 +1275,15 @@ class Threads(Authenticated):
             metadata = thread.get("metadata", {})
             thread_config = thread.get("config", {})
 
-            if graph_id := metadata.get("graph_id"):
+            # Fallback to graph_id from run if not in thread metadata
+            graph_id = metadata.get("graph_id")
+            if not graph_id:
+                for run in conn.store["runs"]:
+                    if run["thread_id"] == thread_id:
+                        graph_id = run["kwargs"]["config"]["configurable"]["graph_id"]
+                        break
+
+            if graph_id:
                 # format latest checkpoint for response
                 checkpointer.latest_iter = checkpoint
                 async with get_graph(
@@ -1351,7 +1361,15 @@ class Threads(Authenticated):
                     detail=f"Thread {thread_id} has in-flight runs: {pending_runs}",
                 )
 
-            if graph_id := metadata.get("graph_id"):
+            # Fallback to graph_id from run if not in thread metadata
+            graph_id = metadata.get("graph_id")
+            if not graph_id:
+                for run in conn.store["runs"]:
+                    if run["thread_id"] == thread_id:
+                        graph_id = run["kwargs"]["config"]["configurable"]["graph_id"]
+                        break
+
+            if graph_id:
                 config["configurable"].setdefault("graph_id", graph_id)
 
                 checkpointer.latest_iter = checkpoint
@@ -1389,7 +1407,11 @@ class Threads(Authenticated):
                         checkpoint_id=next_config["configurable"]["checkpoint_id"],
                     )
             else:
-                raise HTTPException(status_code=400, detail="Thread has no graph ID.")
+                raise HTTPException(
+                    status_code=400,
+                    detail=f"Thread '{thread['thread_id']}' has no assigned graph ID. This usually occurs when no runs have been made on this particular thread."
+                    " This operation requires a graph ID. Please ensure a run has been made for the thread or manually update the thread metadata (by setting the 'graph_id' field) before running this operation.",
+                )
 
         @staticmethod
         async def bulk(
@@ -1471,7 +1493,11 @@ class Threads(Authenticated):
                         checkpoint=next_config["configurable"],
                     )
             else:
-                raise HTTPException(status_code=400, detail="Thread has no graph ID")
+                raise HTTPException(
+                    status_code=400,
+                    detail=f"Thread '{thread['thread_id']}' has no assigned graph ID. This usually occurs when no runs have been made on this particular thread."
+                    " This operation requires a graph ID. Please ensure a run has been made for the thread or manually update the thread metadata (by setting the 'graph_id' field) before running this operation.",
+                )
 
         @staticmethod
         async def list(
@@ -1678,9 +1704,9 @@ class Runs(Authenticated):
             await stream_manager.remove_control_queue(run_id, queue)
 
     @staticmethod
-    async def sweep(conn: InMemConnectionProto) -> list[UUID]:
+    async def sweep() -> None:
         """Sweep runs that are no longer running"""
-        return []
+        pass
 
     @staticmethod
     def _merge_jsonb(*objects: dict) -> dict:
@@ -1714,6 +1740,7 @@ class Runs(Authenticated):
         ctx: Auth.types.BaseAuthContext | None = None,
     ) -> AsyncIterator[Run]:
         """Create a run."""
+        from langgraph_api.config import FF_RICH_THREADS
         from langgraph_api.schema import Run, Thread
 
         assistant_id = _ensure_uuid(assistant_id)
@@ -1759,34 +1786,50 @@ class Runs(Authenticated):
             # Create new thread
             if thread_id is None:
                 thread_id = uuid4()
-            thread = Thread(
-                thread_id=thread_id,
-                status="busy",
-                metadata={
-                    "graph_id": assistant["graph_id"],
-                    "assistant_id": str(assistant_id),
-                    **(config.get("metadata") or {}),
-                    **metadata,
-                },
-                config=Runs._merge_jsonb(
-                    assistant["config"],
-                    config,
-                    {
-                        "configurable": Runs._merge_jsonb(
-                            Runs._get_configurable(assistant["config"]),
-                            Runs._get_configurable(config),
-                        )
+            if FF_RICH_THREADS:
+                thread = Thread(
+                    thread_id=thread_id,
+                    status="busy",
+                    metadata={
+                        "graph_id": assistant["graph_id"],
+                        "assistant_id": str(assistant_id),
+                        **(config.get("metadata") or {}),
+                        **metadata,
                     },
-                ),
-                created_at=datetime.now(UTC),
-                updated_at=datetime.now(UTC),
-                values=b"",
-            )
+                    config=Runs._merge_jsonb(
+                        assistant["config"],
+                        config,
+                        {
+                            "configurable": Runs._merge_jsonb(
+                                Runs._get_configurable(assistant["config"]),
+                                Runs._get_configurable(config),
+                            )
+                        },
+                    ),
+                    created_at=datetime.now(UTC),
+                    updated_at=datetime.now(UTC),
+                    values=b"",
+                )
+            else:
+                thread = Thread(
+                    thread_id=thread_id,
+                    status="idle",
+                    metadata={
+                        "graph_id": assistant["graph_id"],
+                        "assistant_id": str(assistant_id),
+                        **(config.get("metadata") or {}),
+                        **metadata,
+                    },
+                    config={},
+                    created_at=datetime.now(UTC),
+                    updated_at=datetime.now(UTC),
+                    values=b"",
+                )
             await logger.ainfo("Creating thread", thread_id=thread_id)
             conn.store["threads"].append(thread)
         elif existing_thread:
             # Update existing thread
-            if existing_thread["status"] != "busy":
+            if FF_RICH_THREADS and existing_thread["status"] != "busy":
                 existing_thread["status"] = "busy"
                 existing_thread["metadata"] = Runs._merge_jsonb(
                     existing_thread["metadata"],
@@ -1875,7 +1918,9 @@ class Runs(Authenticated):
                             "metadata": merged_metadata,
                         },
                     ),
-                    "context": configurable,
+                    "context": Runs._merge_jsonb(
+                        assistant.get("context", {}), kwargs.get("context", {})
+                    ),
                 },
             ),
             multitask_strategy=multitask_strategy,

langgraph_runtime_inmem/queue.py

@@ -128,8 +128,7 @@ async def queue():
                         # sweep runs if needed
                         if do_sweep:
                             last_sweep_secs = loop.time()
-                            run_ids = await ops.Runs.sweep(conn)
-                            logger.info("Swept runs", run_ids=run_ids)
+                            await ops.Runs.sweep()
             except Exception as exc:
                 # keep trying to run the scheduler indefinitely
                 logger.exception("Background worker scheduler failed", exc_info=exc)

{langgraph_runtime_inmem-0.6.11.dist-info → langgraph_runtime_inmem-0.6.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.6.11
+Version: 0.6.13
 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.11.dist-info → langgraph_runtime_inmem-0.6.13.dist-info}/RECORD

@@ -1,13 +1,13 @@
-langgraph_runtime_inmem/__init__.py,sha256=FaiDP9SdI1qgH5N7_RMO7yqFhq5967V9hXeJsDFc9PA,311
+langgraph_runtime_inmem/__init__.py,sha256=mPj20Y7jJEr9XATlg7oD3Q0F4BJdwWnfN7JMijIiRHc,311
 langgraph_runtime_inmem/checkpoint.py,sha256=nc1G8DqVdIu-ibjKTqXfbPfMbAsKjPObKqegrSzo6Po,4432
 langgraph_runtime_inmem/database.py,sha256=G_6L2khpRDSpS2Vs_SujzHayODcwG5V2IhFP7LLBXgw,6349
 langgraph_runtime_inmem/inmem_stream.py,sha256=UWk1srLF44HZPPbRdArGGhsy0MY0UOJKSIxBSO7Hosc,5138
 langgraph_runtime_inmem/lifespan.py,sha256=t0w2MX2dGxe8yNtSX97Z-d2pFpllSLS4s1rh2GJDw5M,3557
 langgraph_runtime_inmem/metrics.py,sha256=HhO0RC2bMDTDyGBNvnd2ooLebLA8P1u5oq978Kp_nAA,392
-langgraph_runtime_inmem/ops.py,sha256=pf5wtLPTJWwmjr9zsGXHdN7slXrYR2PQvCx71KujFqE,91593
-langgraph_runtime_inmem/queue.py,sha256=nqfgz7j_Jkh5Ek5-RsHB2Uvwbxguu9IUPkGXIxvFPns,10037
+langgraph_runtime_inmem/ops.py,sha256=ffxYACydXPmWKt89AY22M1j8s75JbFMtzXI1SKC3R50,93949
+langgraph_runtime_inmem/queue.py,sha256=33qfFKPhQicZ1qiibllYb-bTFzUNSN2c4bffPACP5es,9952
 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.11.dist-info/METADATA,sha256=U4PqBeXV9YamUB2A8etc50Z8mZ2WxYJ0swoo3WSy6nk,566
-langgraph_runtime_inmem-0.6.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_runtime_inmem-0.6.11.dist-info/RECORD,,
+langgraph_runtime_inmem-0.6.13.dist-info/METADATA,sha256=GUxT1ClUkx_1b0nR-tiuetNRDfSoB5iejMXS2ieX9HA,566
+langgraph_runtime_inmem-0.6.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_runtime_inmem-0.6.13.dist-info/RECORD,,