langgraph-api 0.0.36__tar.gz → 0.0.37__tar.gz

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.36
+Version: 0.0.37
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -10,6 +10,7 @@ Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: cloudpickle (>=3.0.0,<4.0.0)
 Requires-Dist: cryptography (>=43.0.3,<44.0.0)
 Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/langgraph_api/cli.py

@@ -130,6 +130,7 @@ def run_server(
     auth: AuthConfig | None = None,
     http: typing.Optional["HttpConfig"] = None,
     studio_url: str | None = None,
+    disable_persistence: bool = False,
     **kwargs: typing.Any,
 ):
     """Run the LangGraph API server."""
@@ -189,6 +190,7 @@ def run_server(
         LANGGRAPH_AUTH=json.dumps(auth) if auth else None,
         LANGGRAPH_HTTP=json.dumps(http) if http else None,
         LANGGRAPH_API_URL=local_url,
+        LANGGRAPH_DISABLE_FILE_PERSISTENCE=str(disable_persistence).lower(),
         # See https://developer.chrome.com/blog/private-network-access-update-2024-03
         ALLOW_PRIVATE_NETWORK="true",
     )

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/langgraph_api/serde.py

@@ -1,5 +1,4 @@
 import asyncio
-import pickle
 import uuid
 from base64 import b64encode
 from collections import deque
@@ -18,9 +17,13 @@ from re import Pattern
 from typing import Any, NamedTuple
 from zoneinfo import ZoneInfo
 
+import cloudpickle
 import orjson
+import structlog
 from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 
+logger = structlog.stdlib.get_logger(__name__)
+
 
 class Fragment(NamedTuple):
     buf: bytes
@@ -122,9 +125,15 @@ class Serializer(JsonPlusSerializer):
         try:
             return super().dumps_typed(obj)
         except TypeError:
-            return "pickle", pickle.dumps(obj)
+            return "pickle", cloudpickle.dumps(obj)
 
     def loads_typed(self, data: tuple[str, bytes]) -> Any:
         if data[0] == "pickle":
-            return pickle.loads(data[1])
+            try:
+                return cloudpickle.loads(data[1])
+            except Exception as e:
+                logger.warning(
+                    "Failed to unpickle object, replacing w None", exc_info=e
+                )
+                return None
         return super().loads_typed(data)

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/langgraph_storage/checkpoint.py

@@ -16,6 +16,9 @@ from langgraph_api.serde import Serializer
 logger = logging.getLogger(__name__)
 
 _EXCLUDED_KEYS = {"checkpoint_ns", "checkpoint_id", "run_id", "thread_id"}
+DISABLE_FILE_PERSISTENCE = (
+    os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
+)
 
 
 class InMemorySaver(MemorySaver):
@@ -56,7 +59,7 @@ class InMemorySaver(MemorySaver):
 
         super().__init__(
             serde=serde if serde is not None else Serializer(),
-            factory=factory,
+            factory=factory if not DISABLE_FILE_PERSISTENCE else None,
         )
 
     def put(

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/langgraph_storage/store.py

@@ -13,13 +13,22 @@ from langgraph.store.memory import InMemoryStore
 from langgraph_api.graph import resolve_embeddings
 
 _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)
-        self._data = PersistentDict(dict, filename=_STORE_FILE)
-        self._vectors = PersistentDict(lambda: defaultdict(dict), filename=_VECTOR_FILE)
+        if not DISABLE_FILE_PERSISTENCE:
+            self._data = PersistentDict(dict, filename=_STORE_FILE)
+            self._vectors = PersistentDict(
+                lambda: defaultdict(dict), filename=_VECTOR_FILE
+            )
+        else:
+            self._data = InMemoryStore._data
+            self._vectors = InMemoryStore._vectors
         self._load_data(self._data, which="data")
         self._load_data(self._vectors, which="vectors")
 

{langgraph_api-0.0.36 → langgraph_api-0.0.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.36"
+version = "0.0.37"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -36,6 +36,7 @@ structlog = ">=24.1.0,<26"
 pyjwt = "^2.9.0"
 cryptography = "^43.0.3"
 langgraph-sdk = "^0.1.58"
+cloudpickle = "^3.0.0"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.6.2"