langgraph-runtime-inmem 0.23.0__tar.gz → 0.23.1__tar.gz

{langgraph_runtime_inmem-0.23.0 → langgraph_runtime_inmem-0.23.1}/PKG-INFO

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

{langgraph_runtime_inmem-0.23.0 → langgraph_runtime_inmem-0.23.1}/langgraph_runtime_inmem/__init__.py

@@ -10,7 +10,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.23.0"
+__version__ = "0.23.1"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.23.0 → langgraph_runtime_inmem-0.23.1}/langgraph_runtime_inmem/_persistence.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import functools
 import logging
+import os
 import threading
 import weakref
 
@@ -14,10 +15,15 @@ logger = logging.getLogger(__name__)
 _stores: dict[str, weakref.ref[PersistentDict]] = {}
 _flush_thread: tuple[threading.Event, threading.Thread] | None = None
 _flush_interval: int = 10
+DISABLE_FILE_PERSISTENCE = (
+    os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
+)
 
 
 def register_persistent_dict(d: PersistentDict) -> None:
     """Register a PersistentDict for periodic flushing."""
+    if DISABLE_FILE_PERSISTENCE:
+        return
     global _flush_thread
     _stores[d.filename] = weakref.ref(d)
     if _flush_thread is None:

{langgraph_runtime_inmem-0.23.0 → langgraph_runtime_inmem-0.23.1}/langgraph_runtime_inmem/routes.py

@@ -8,6 +8,13 @@ from langgraph_runtime_inmem.database import connect
 def get_internal_routes():
     from langgraph_api.config import MIGRATIONS_PATH
 
+    try:
+        from langgraph_api.middleware import http_logger
+
+        http_logger.PATHS_IGNORE.add("/internal/truncate")
+    except ImportError:
+        pass
+
     if "__inmem" not in MIGRATIONS_PATH:
         # not in a testing mode.
         return []

{langgraph_runtime_inmem-0.23.0 → langgraph_runtime_inmem-0.23.1}/langgraph_runtime_inmem/store.py

@@ -10,29 +10,27 @@ from langgraph.store.base import BaseStore, Op, Result
 from langgraph.store.base.batch import AsyncBatchedBaseStore
 from langgraph.store.memory import InMemoryStore
 
-from langgraph_runtime_inmem._persistence import register_persistent_dict
+from langgraph_runtime_inmem import _persistence
 
 _STORE_CONFIG = None
-DISABLE_FILE_PERSISTENCE = (
-    os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
-)
 
 
 class DiskBackedInMemStore(InMemoryStore):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
-        if not DISABLE_FILE_PERSISTENCE:
+        if not _persistence.DISABLE_FILE_PERSISTENCE:
             self._data = PersistentDict(dict, filename=_STORE_FILE)
             self._vectors = PersistentDict(
                 lambda: defaultdict(dict), filename=_VECTOR_FILE
             )
-            register_persistent_dict(self._data)
-            register_persistent_dict(self._vectors)
+            _persistence.register_persistent_dict(self._data)
+            _persistence.register_persistent_dict(self._vectors)
+            self._load_data(self._data, which="data")
+            self._load_data(self._vectors, which="vectors")
         else:
-            self._data = InMemoryStore._data
-            self._vectors = InMemoryStore._vectors
-        self._load_data(self._data, which="data")
-        self._load_data(self._vectors, which="vectors")
+            self._data = defaultdict(dict)
+            # [ns][key][path]
+            self._vectors = defaultdict(lambda: defaultdict(dict))
 
     def _load_data(self, container: PersistentDict, which: str) -> None:
         if not container.filename:
@@ -58,8 +56,10 @@ class DiskBackedInMemStore(InMemoryStore):
         return asyncio.create_task(asyncio.sleep(0))
 
     def close(self) -> None:
-        self._data.close()
-        self._vectors.close()
+        if isinstance(self._data, PersistentDict):
+            self._data.close()
+        if isinstance(self._vectors, PersistentDict):
+            self._vectors.close()
 
 
 class BatchedStore(AsyncBatchedBaseStore):