langgraph-api 0.0.36__py3-none-any.whl → 0.0.37__py3-none-any.whl

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/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.dist-info → langgraph_api-0.0.37.dist-info}/METADATA

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

@@ -18,7 +18,7 @@ langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdT
 langgraph_api/auth/middleware.py,sha256=jU8aDSIZHdzCGdifejRF7ndHkSjBtqIHcBwFIuUdHEA,1875
 langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
-langgraph_api/cli.py,sha256=cSRKZzaxSbfDQO3-2YWAo2NgeE5IK0mabYWt4m5q-dU,12047
+langgraph_api/cli.py,sha256=76wJLx1ekaRmXR5DxGy1_RuA0lQq2jBUzHAAVUKV2Ho,12163
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
 langgraph_api/config.py,sha256=MwTyJUr2wwZdUbL0gIvol_cCGi24Py9DvBnSozE1ai8,9342
 langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
@@ -72,7 +72,7 @@ langgraph_api/patch.py,sha256=82xjuFqY7tgrUm-k1XWHI6k8S6QovSD0zhe--8_xW4o,1296
 langgraph_api/queue_entrypoint.py,sha256=4xICUxXarNV8DhnaqAMhVi3xCmyVKCL3J5NzHxPA9Xc,1835
 langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
 langgraph_api/schema.py,sha256=hNbg6ep2wiGBBtBJVNBgMYA8uC33AfaqhRXXVUY_At8,5361
-langgraph_api/serde.py,sha256=HfaIBNfdKg2W_mIpw_er5irG3GlIiiVqkqfbDgnYHpg,3671
+langgraph_api/serde.py,sha256=D5t_HeABMYKRAsoXWaWG0IsdaYM8yOXIez2wJUTIgT8,3963
 langgraph_api/server.py,sha256=bnXOOYztQmqR-QVpEFoRWB5Fzd33PuEIlwBK2R7W8NE,4849
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
@@ -86,18 +86,18 @@ langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 langgraph_license/middleware.py,sha256=_ODIYzQkymr6W9_Fp9wtf1kAQspnpsmr53xuzyF2GA0,612
 langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZoj0,203
 langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_storage/checkpoint.py,sha256=X6YHrCwIPVE0iEQy16GhleyyHUdXdCkC-c-Q1SW-rAM,3884
+langgraph_storage/checkpoint.py,sha256=Qq0y6vdh27qdF3h5nOLT5CcX9Rj2bcFqkVOMeCaGoK4,4036
 langgraph_storage/database.py,sha256=I0AgFeJ-NSTT34vxKxQBUf1z2syFP0S8QpKCqTixrzY,5652
 langgraph_storage/inmem_stream.py,sha256=8bxkILIuFpr7P7RQ37SQAxrpRKvmbHdRB_nbfFiomlk,3263
 langgraph_storage/ops.py,sha256=S0QHbnquEXvAM63NQMgqtyTjm-8nlq0pOvgJgFD38Z8,75312
 langgraph_storage/queue.py,sha256=UDgsUTtUMfBSRDrQ8Onis-FJO4n7KTsX6sdpbY8Hs0A,5055
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_storage/store.py,sha256=8QSM0gm6ZV40antWw0YHfnU71swZRsozjd85R-MSrjg,3100
+langgraph_storage/store.py,sha256=JB9jZ87GE19MVN9wgl3-esgR2eIkeipws9q6qsPWkgc,3399
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=-25y3NRQ88e_944UXo76Goa34HJhC7pj6I9tjYUwvuE,131492
-langgraph_api-0.0.36.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.36.dist-info/METADATA,sha256=T5nIH34PPCdvY8DcFJWKDgrrNutgpgwXGmv52fX_2AI,4027
-langgraph_api-0.0.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.36.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.36.dist-info/RECORD,,
+langgraph_api-0.0.37.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.37.dist-info/METADATA,sha256=zawmEdZt_TgVBU6iHpMFrfdREjAYy9PqXU6lyy1dBb8,4071
+langgraph_api-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.37.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.37.dist-info/RECORD,,

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_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")