langgraph-runtime-inmem 0.21.0__tar.gz → 0.22.0__tar.gz

{langgraph_runtime_inmem-0.21.0 → langgraph_runtime_inmem-0.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.21.0
+Version: 0.22.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.21.0 → langgraph_runtime_inmem-0.22.0}/langgraph_runtime_inmem/__init__.py

@@ -6,10 +6,11 @@ from langgraph_runtime_inmem import (
     ops,
     queue,
     retry,
+    routes,
     store,
 )
 
-__version__ = "0.21.0"
+__version__ = "0.22.0"
 __all__ = [
     "ops",
     "database",
@@ -19,5 +20,6 @@ __all__ = [
     "store",
     "queue",
     "metrics",
+    "routes",
     "__version__",
 ]

{langgraph_runtime_inmem-0.21.0 → langgraph_runtime_inmem-0.22.0}/langgraph_runtime_inmem/checkpoint.py

@@ -21,6 +21,17 @@ if typing.TYPE_CHECKING:
 logger = logging.getLogger(__name__)
 
 _EXCLUDED_KEYS = {"checkpoint_ns", "checkpoint_id", "run_id", "thread_id"}
+
+# Configurable keys that are transient (per-request) and should not be persisted in checkpoints
+_TRANSIENT_CONFIGURABLE_KEYS = frozenset(
+    {
+        "langgraph_request_id",
+        "langgraph_auth_user",
+        "langgraph_auth_user_id",
+        "langgraph_auth_permissions",
+    }
+)
+
 DISABLE_FILE_PERSISTENCE = (
     os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
 )
@@ -84,14 +95,26 @@ class InMemorySaver(MemorySaver):
         new_versions: dict[str, str | int | float],
     ) -> RunnableConfig:
         # TODO: Should this be done in OSS as well?
+        # Filter out transient fields that are request-scoped, not checkpoint-scoped
+        config_metadata = config.get("metadata", {})
         metadata = {
             **{
                 k: v
                 for k, v in config["configurable"].items()
-                if not k.startswith("__") and k not in _EXCLUDED_KEYS
+                if not k.startswith("__")
+                and k not in _EXCLUDED_KEYS
+                and k not in _TRANSIENT_CONFIGURABLE_KEYS
+            },
+            **{
+                k: v
+                for k, v in config_metadata.items()
+                if k not in _TRANSIENT_CONFIGURABLE_KEYS
+            },
+            **{
+                k: v
+                for k, v in metadata.items()
+                if k not in _TRANSIENT_CONFIGURABLE_KEYS
             },
-            **config.get("metadata", {}),
-            **metadata,
         }
         if not isinstance(checkpoint["id"], uuid.UUID):
             # Avoid type inconsistencies

{langgraph_runtime_inmem-0.21.0 → langgraph_runtime_inmem-0.22.0}/langgraph_runtime_inmem/ops.py

@@ -1437,6 +1437,7 @@ class Threads(Authenticated):
                     thread_config,
                     checkpointer=checkpointer,
                     store=(await get_store()),
+                    is_for_execution=False,
                 ) as graph:
                     result = await graph.aget_state(config, subgraphs=subgraphs)
                     if (
@@ -1532,6 +1533,7 @@ class Threads(Authenticated):
                     thread_config,
                     checkpointer=checkpointer,
                     store=(await get_store()),
+                    is_for_execution=False,
                 ) as graph:
                     update_config = config.copy()
                     update_config["configurable"] = {
@@ -1634,6 +1636,7 @@ class Threads(Authenticated):
                     thread_config,
                     checkpointer=Checkpointer(),
                     store=(await get_store()),
+                    is_for_execution=False,
                 ) as graph:
                     next_config = await graph.abulk_update_state(
                         config,
@@ -1734,6 +1737,7 @@ class Threads(Authenticated):
                         Checkpointer, conn, unpack_hook=_msgpack_ext_hook_to_json
                     ),
                     store=(await get_store()),
+                    is_for_execution=False,
                 ) as graph:
                     # Convert before parameter if it's a string
                     before_param = (

langgraph_runtime_inmem-0.22.0/langgraph_runtime_inmem/routes.py

@@ -0,0 +1,39 @@
+"""Internal routes for inmem runtime (testing/debugging only)."""
+
+import asyncio
+
+from langgraph_runtime_inmem.database import connect
+
+
+def get_internal_routes():
+    from langgraph_api.config import MIGRATIONS_PATH
+
+    if "__inmem" not in MIGRATIONS_PATH:
+        # not in a testing mode.
+        return []
+    from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
+
+    async def truncate(request: ApiRequest):
+        """Truncate all inmem data (for testing)."""
+        from langgraph_runtime.checkpoint import Checkpointer
+
+        await asyncio.to_thread(Checkpointer().clear)
+        async with connect() as conn:
+            await asyncio.to_thread(conn.clear)
+        return ApiResponse({"ok": True})
+
+    async def debug_get_raw_thread(request: ApiRequest):
+        """Return raw thread from store without decryption (for testing)."""
+        thread_id = request.path_params["thread_id"]
+        async with connect() as conn:
+            for thread in conn.store["threads"]:
+                if str(thread["thread_id"]) == thread_id:
+                    return ApiResponse(thread)
+        return ApiResponse({"error": "not found"}, status_code=404)
+
+    return [
+        ApiRoute("/internal/truncate", truncate, methods=["POST"]),
+        ApiRoute(
+            "/internal/debug/thread/{thread_id}", debug_get_raw_thread, methods=["GET"]
+        ),
+    ]