langgraph-runtime-inmem 0.14.1__tar.gz → 0.16.0__tar.gz

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/.gitignore

@@ -65,3 +65,4 @@ pnpm-debug.log*
 .editorconfig
 *.db
 executor_py/tests/bin/engine-test-server
+.notes/

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.14.1
+Version: 0.16.0
 Summary: Inmem implementation for the LangGraph API server.
 Author-email: Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 Requires-Python: >=3.11.0
 Requires-Dist: blockbuster<2.0.0,>=1.5.24
-Requires-Dist: langgraph-checkpoint>=2.0.25
-Requires-Dist: langgraph>=0.2
+Requires-Dist: langgraph-checkpoint<4,>=3
+Requires-Dist: langgraph<2,>=0.4.10
 Requires-Dist: sse-starlette>=2
 Requires-Dist: starlette>=0.37
 Requires-Dist: structlog>23

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/langgraph_runtime_inmem/__init__.py

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

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/langgraph_runtime_inmem/database.py

@@ -142,7 +142,9 @@ class InMemConnectionProto:
 
 
 @asynccontextmanager
-async def connect(*, __test__: bool = False) -> AsyncIterator["AsyncConnectionProto"]:
+async def connect(
+    *, supports_core_api: bool = False, __test__: bool = False
+) -> AsyncIterator["AsyncConnectionProto"]:
     yield InMemConnectionProto()
 
 

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/langgraph_runtime_inmem/lifespan.py

@@ -14,6 +14,13 @@ from langgraph_runtime_inmem.database import start_pool, stop_pool
 logger = structlog.stdlib.get_logger(__name__)
 
 
+_LAST_LIFESPAN_ERROR: BaseException | None = None
+
+
+def get_last_error() -> BaseException | None:
+    return _LAST_LIFESPAN_ERROR
+
+
 @asynccontextmanager
 async def lifespan(
     app: Starlette | None = None,
@@ -42,9 +49,27 @@ async def lifespan(
     except RuntimeError:
         await logger.aerror("Failed to set loop")
 
+    global _LAST_LIFESPAN_ERROR
+    _LAST_LIFESPAN_ERROR = None
+
     await start_http_client()
     await start_pool()
     await start_ui_bundler()
+
+    async def _log_graph_load_failure(err: graph.GraphLoadError) -> None:
+        cause = err.__cause__ or err.cause
+        log_fields = err.log_fields()
+        log_fields["action"] = "fix_user_graph"
+        await logger.aerror(
+            f"Graph '{err.spec.id}' failed to load: {err.cause_message}",
+            **log_fields,
+        )
+        await logger.adebug(
+            "Full graph load failure traceback (internal)",
+            **{k: v for k, v in log_fields.items() if k != "user_traceback"},
+            exc_info=cause,
+        )
+
     try:
         async with SimpleTaskGroup(
             cancel=True,
@@ -77,11 +102,21 @@ async def lifespan(
             var_child_runnable_config.set(langgraph_config)
 
             # Keep after the setter above so users can access the store from within the factory function
-            await graph.collect_graphs_from_env(True)
+            try:
+                await graph.collect_graphs_from_env(True)
+            except graph.GraphLoadError as exc:
+                _LAST_LIFESPAN_ERROR = exc
+                await _log_graph_load_failure(exc)
+                raise
             if config.N_JOBS_PER_WORKER > 0:
                 tg.create_task(queue_with_signal())
 
             yield
+    except graph.GraphLoadError as exc:
+        _LAST_LIFESPAN_ERROR = exc
+        raise
+    except asyncio.CancelledError:
+        pass
     finally:
         await api_store.exit_store()
         await stop_ui_bundler()
@@ -98,3 +133,6 @@ async def queue_with_signal():
     except Exception as exc:
         logger.exception("Queue failed. Signaling shutdown", exc_info=exc)
         signal.raise_signal(signal.SIGINT)
+
+
+lifespan.get_last_error = get_last_error  # type: ignore[attr-defined]

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/langgraph_runtime_inmem/metrics.py

@@ -1,7 +1,7 @@
 from langgraph_runtime_inmem.queue import get_num_workers
 
 
-def get_metrics() -> dict[str, int]:
+def get_metrics() -> dict[str, dict[str, int]]:
     from langgraph_api import config
 
     workers_max = config.N_JOBS_PER_WORKER

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/langgraph_runtime_inmem/ops.py

@@ -556,6 +556,8 @@ class Assistants(Authenticated):
                 "metadata": version_data["metadata"],
                 "version": version_data["version"],
                 "updated_at": datetime.now(UTC),
+                "name": version_data["name"],
+                "description": version_data["description"],
             }
         )
 
@@ -1230,13 +1232,23 @@ class Threads(Authenticated):
         """Create a copy of an existing thread."""
         thread_id = _ensure_uuid(thread_id)
         new_thread_id = uuid4()
-        filters = await Threads.handle_event(
+        read_filters = await Threads.handle_event(
             ctx,
             "read",
             Auth.types.ThreadsRead(
+                thread_id=thread_id,
+            ),
+        )
+        # Assert that the user has permissions to create a new thread.
+        # (We don't actually need the filters.)
+        await Threads.handle_event(
+            ctx,
+            "create",
+            Auth.types.ThreadsCreate(
                 thread_id=new_thread_id,
             ),
         )
+
         async with conn.pipeline():
             # Find the original thread in our store
             original_thread = next(
@@ -1245,8 +1257,8 @@ class Threads(Authenticated):
 
             if not original_thread:
                 return _empty_generator()
-            if filters and not _check_filter_match(
-                original_thread["metadata"], filters
+            if read_filters and not _check_filter_match(
+                original_thread["metadata"], read_filters
             ):
                 return _empty_generator()
 
@@ -1399,6 +1411,7 @@ class Threads(Authenticated):
             """Add state to a thread."""
             from langgraph_api.graph import get_graph
             from langgraph_api.schema import ThreadUpdateResponse
+            from langgraph_api.state import state_snapshot_to_thread_state
             from langgraph_api.store import get_store
             from langgraph_api.utils import fetchone
 
@@ -1487,7 +1500,7 @@ class Threads(Authenticated):
                     from langgraph_api.serde import json_dumpb
 
                     event_data = {
-                        "state": state,
+                        "state": state_snapshot_to_thread_state(state),
                         "thread_id": str(thread_id),
                     }
                     await Threads.Stream.publish(
@@ -2107,7 +2120,6 @@ 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)
@@ -2154,49 +2166,35 @@ class Runs(Authenticated):
             # Create new thread
             if thread_id is None:
                 thread_id = uuid4()
-            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,
-                    },
-                    config=Runs._merge_jsonb(
-                        assistant["config"],
-                        config,
-                        {
-                            "configurable": Runs._merge_jsonb(
-                                Runs._get_configurable(assistant["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,
+
+            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"]),
+                        )
                     },
-                    config={},
-                    created_at=datetime.now(UTC),
-                    updated_at=datetime.now(UTC),
-                    values=b"",
-                )
+                ),
+                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 FF_RICH_THREADS and existing_thread["status"] != "busy":
+            if existing_thread["status"] != "busy":
                 existing_thread["status"] = "busy"
                 existing_thread["metadata"] = Runs._merge_jsonb(
                     existing_thread["metadata"],

{langgraph_runtime_inmem-0.14.1 → langgraph_runtime_inmem-0.16.0}/pyproject.toml

@@ -14,11 +14,11 @@ readme = "README.md"
 requires-python = ">=3.11.0"
 dependencies = [
     "blockbuster>=1.5.24,<2.0.0",
-    "langgraph>=0.2",
+    "langgraph>=0.4.10,<2",
     "structlog>23",
     "sse-starlette>=2",
     "starlette>=0.37",
-    "langgraph-checkpoint>=2.0.25",
+    "langgraph-checkpoint>=3,<4",
 ]
 
 [dependency-groups]
@@ -62,4 +62,4 @@ exclude = [
 [tool.ruff]
 lint.select = ["E", "F", "UP", "B", "I"]
 lint.ignore = ["E501", "B008"]
-target-version = "py311"
+target-version = "py311"