langgraph-runtime-inmem 0.23.0__py3-none-any.whl → 0.23.1__py3-none-any.whl

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/_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/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/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):

{langgraph_runtime_inmem-0.23.0.dist-info → langgraph_runtime_inmem-0.23.1.dist-info}/METADATA

@@ -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.dist-info → langgraph_runtime_inmem-0.23.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-langgraph_runtime_inmem/__init__.py,sha256=oKfMmvvPRewTlo9oduB3tCC3l2b2MiD_9lka4PMm_5M,337
-langgraph_runtime_inmem/_persistence.py,sha256=exchMr_NQB_h7PHt0vq5QBh25cOGoW0jHAFo07b1BFI,1711
+langgraph_runtime_inmem/__init__.py,sha256=D9oGOcx4U1hp9tKw_-AFKVGi4nI5GYSdCpnBhzFVnR0,337
+langgraph_runtime_inmem/_persistence.py,sha256=0h_HLXc079TG0rXoiQjICPZLhhBpTsmZ2PG8UbB5YY4,1879
 langgraph_runtime_inmem/checkpoint.py,sha256=VD5c6CktsToo_f4qPe1WP_csdonQoOb7h5lHv4U0ZAE,8372
 langgraph_runtime_inmem/database.py,sha256=iP7W1SI4kUkqcHtkg3aMmP-YLgZfvMHANwN-P4Pb1pY,6607
 langgraph_runtime_inmem/inmem_stream.py,sha256=PFLWbsxU8RqbT5mYJgNk6v5q6TWJRIY1hkZWhJF8nkI,9094
@@ -8,8 +8,8 @@ langgraph_runtime_inmem/metrics.py,sha256=_YiSkLnhQvHpMktk38SZo0abyL-5GihfVAtBo0
 langgraph_runtime_inmem/ops.py,sha256=OZ1VicFoh9eOxD15LbZsmCZ1JTaWkMUjdY0nF0t9e-k,120806
 langgraph_runtime_inmem/queue.py,sha256=WM6ZJu25QPVjFXeJYW06GALLUgRsnRrA4YdypR0oG0U,9584
 langgraph_runtime_inmem/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_runtime_inmem/routes.py,sha256=VVNxgJ8FWI3kDBoIgQUWN1gY5ivo7L954Agxzv72TAY,1377
-langgraph_runtime_inmem/store.py,sha256=a3YKsLnFv4bu3zPvagIFv0xmtrIp_pmGvj1CnD3PHL0,3682
-langgraph_runtime_inmem-0.23.0.dist-info/METADATA,sha256=9x1G8RrISXklr1MikxsE4kJFZTswjmATqNgA0EsPQYY,570
-langgraph_runtime_inmem-0.23.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langgraph_runtime_inmem-0.23.0.dist-info/RECORD,,
+langgraph_runtime_inmem/routes.py,sha256=25W-8fQCmBD3MHTv91GIGgBl8tfl-MovdXDd34yVkzM,1541
+langgraph_runtime_inmem/store.py,sha256=EehjI-LWemnMYqDeMJp75hvZK8oInZL-JhnfnEwtlPk,3751
+langgraph_runtime_inmem-0.23.1.dist-info/METADATA,sha256=waXdjJtsILPbF6sOf5enVzC-un7WNJDm0UCUoNJxDtk,570
+langgraph_runtime_inmem-0.23.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langgraph_runtime_inmem-0.23.1.dist-info/RECORD,,