langgraph-runtime-inmem 0.23.0__py3-none-any.whl → 0.23.2__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.2"
 __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/lifespan.py

@@ -29,7 +29,7 @@ async def lifespan(
     **kwargs: Any,
 ):
     import langgraph_api.config as config
-    from langgraph_api import __version__, feature_flags, graph, thread_ttl
+    from langgraph_api import __version__, feature_flags, graph
     from langgraph_api import (
         _checkpointer as api_checkpointer,
     )
@@ -87,7 +87,6 @@ async def lifespan(
                 tg.create_task(store_instance.start_ttl_sweeper())  # type: ignore
             else:
                 await logger.ainfo("Using custom store. Skipping store TTL sweeper.")
-            tg.create_task(thread_ttl.thread_ttl_sweep_loop())
 
             if feature_flags.USE_RUNTIME_CONTEXT_API:
                 from langgraph._internal._constants import CONFIG_KEY_RUNTIME

langgraph_runtime_inmem/ops.py

@@ -253,6 +253,7 @@ class Assistants(Authenticated):
         name: str,
         ctx: Auth.types.BaseAuthContext | None = None,
         description: str | None = None,
+        system: bool = False,
     ) -> AsyncIterator[Assistant]:
         """Insert an assistant."""
         from langgraph_api.graph import assert_graph_exists
@@ -3003,6 +3004,22 @@ class Crons:
         on_run_completed: Literal["delete", "keep"] | None = None,
         end_time: datetime | None = None,
         metadata: dict | None = None,
+        enabled: bool,
+        ctx: Auth.types.BaseAuthContext | None = None,
+    ) -> AsyncIterator[Cron]:
+        raise NotImplementedError
+
+    @staticmethod
+    async def update(
+        conn: InMemConnectionProto,
+        *,
+        cron_id: UUID,
+        schedule: str | None = None,
+        end_time: datetime | None = None,
+        enabled: bool | None = None,
+        on_run_completed: Literal["delete", "keep"] | None = None,
+        payload: dict | None = None,
+        metadata: dict | None = None,
         ctx: Auth.types.BaseAuthContext | None = None,
     ) -> AsyncIterator[Cron]:
         raise NotImplementedError
@@ -3038,6 +3055,7 @@ class Crons:
         *,
         assistant_id: UUID | None,
         thread_id: UUID | None,
+        enabled: bool,
         limit: int,
         offset: int,
         select: list[CronSelectField] | None = 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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.23.0
+Version: 0.23.2
 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.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+langgraph_runtime_inmem/__init__.py,sha256=IfmL9kXBjvAhwVwQVwlbKnJMJv_q3hKtHE_zOkCU17I,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
+langgraph_runtime_inmem/lifespan.py,sha256=zprYeZ05QfYJIa7h7KCeAJSDSpnvo-JWvwoQIXFZX2w,4860
+langgraph_runtime_inmem/metrics.py,sha256=_YiSkLnhQvHpMktk38SZo0abyL-5GihfVAtBo0-lFIc,403
+langgraph_runtime_inmem/ops.py,sha256=u6T1S90mqEg5CsMPu3jZrIxRJzArVvgyNVbYBvRiCQw,121370
+langgraph_runtime_inmem/queue.py,sha256=WM6ZJu25QPVjFXeJYW06GALLUgRsnRrA4YdypR0oG0U,9584
+langgraph_runtime_inmem/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
+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.2.dist-info/METADATA,sha256=Mra4MM_1TaYEatjLHb664VPCxvlQ-NjdZ-OtVe3RyDQ,570
+langgraph_runtime_inmem-0.23.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langgraph_runtime_inmem-0.23.2.dist-info/RECORD,,

langgraph_runtime_inmem-0.23.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-langgraph_runtime_inmem/__init__.py,sha256=oKfMmvvPRewTlo9oduB3tCC3l2b2MiD_9lka4PMm_5M,337
-langgraph_runtime_inmem/_persistence.py,sha256=exchMr_NQB_h7PHt0vq5QBh25cOGoW0jHAFo07b1BFI,1711
-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
-langgraph_runtime_inmem/lifespan.py,sha256=51w3ZKvxcosd7XKkTE2Tnxtr3tux4rJuChGEN0CuvCY,4935
-langgraph_runtime_inmem/metrics.py,sha256=_YiSkLnhQvHpMktk38SZo0abyL-5GihfVAtBo0-lFIc,403
-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,,