langgraph-api 0.0.33__py3-none-any.whl → 0.0.34__py3-none-any.whl

langgraph_api/serde.py

@@ -52,11 +52,17 @@ def default(obj):
     # https://github.com/ijl/orjson#serialize
     if isinstance(obj, Fragment):
         return orjson.Fragment(obj.buf)
-    if hasattr(obj, "model_dump") and callable(obj.model_dump):
+    if (
+        hasattr(obj, "model_dump")
+        and callable(obj.model_dump)
+        and not isinstance(obj, type)
+    ):
         return obj.model_dump()
-    elif hasattr(obj, "dict") and callable(obj.dict):
+    elif hasattr(obj, "dict") and callable(obj.dict) and not isinstance(obj, type):
         return obj.dict()
-    elif hasattr(obj, "_asdict") and callable(obj._asdict):
+    elif (
+        hasattr(obj, "_asdict") and callable(obj._asdict) and not isinstance(obj, type)
+    ):
         return obj._asdict()
     elif isinstance(obj, BaseException):
         return {"error": type(obj).__name__, "message": str(obj)}

{langgraph_api-0.0.33.dist-info → langgraph_api-0.0.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.33
+Version: 0.0.34
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -15,7 +15,7 @@ Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.4.0)
-Requires-Dist: langgraph-checkpoint (>=2.0.21,<3.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.23,<3.0)
 Requires-Dist: langgraph-sdk (>=0.1.58,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.4.0)
 Requires-Dist: orjson (>=3.9.7)

{langgraph_api-0.0.33.dist-info → langgraph_api-0.0.34.dist-info}/RECORD

@@ -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=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
+langgraph_api/serde.py,sha256=HfaIBNfdKg2W_mIpw_er5irG3GlIiiVqkqfbDgnYHpg,3671
 langgraph_api/server.py,sha256=CiNK327zTsEpoVGeJK1JOtZHvOBYRoz0CnBTZUmsC7c,4567
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
@@ -85,18 +85,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=V4t2GwYEJdPCHbhq_4Udhlv0TWKDzlMu_rlNPdTDc50,3589
+langgraph_storage/checkpoint.py,sha256=X6YHrCwIPVE0iEQy16GhleyyHUdXdCkC-c-Q1SW-rAM,3884
 langgraph_storage/database.py,sha256=I0AgFeJ-NSTT34vxKxQBUf1z2syFP0S8QpKCqTixrzY,5652
 langgraph_storage/inmem_stream.py,sha256=8bxkILIuFpr7P7RQ37SQAxrpRKvmbHdRB_nbfFiomlk,3263
-langgraph_storage/ops.py,sha256=0DhtDQllBnV495DPrwhbR_bQUg7ZrGBe0FxSZcicz5g,72469
+langgraph_storage/ops.py,sha256=bCgQd7VVezTsTZQe4QlWk-TZiD5YkUFU0_vi7Gxg4YE,72963
 langgraph_storage/queue.py,sha256=UDgsUTtUMfBSRDrQ8Onis-FJO4n7KTsX6sdpbY8Hs0A,5055
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
 langgraph_storage/store.py,sha256=33-J5-Xvobb9ArSa-GezP5KtfXgzWkHUHPyjRYmdw-E,2985
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=6okfuwPAKB2Hi1tup-EFza49pmN_iD2Yt5JktMALcq4,127984
-langgraph_api-0.0.33.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.33.dist-info/METADATA,sha256=PCu4vrh8uCMFyww-JUdBxsrQFN8NrAX9hckbr3SPem0,4027
-langgraph_api-0.0.33.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.33.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.33.dist-info/RECORD,,
+langgraph_api-0.0.34.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.34.dist-info/METADATA,sha256=w65HKkHiBPnTZRwJQyFjMTU9TE-kjQnOajtuUL2aBng,4027
+langgraph_api-0.0.34.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.34.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.34.dist-info/RECORD,,

langgraph_storage/checkpoint.py

@@ -105,7 +105,15 @@ class InMemorySaver(MemorySaver):
 MEMORY = InMemorySaver()
 
 
-def Checkpointer(*args, **kwargs):
+def Checkpointer(*args, unpack_hook=None, **kwargs):
+    if unpack_hook is not None:
+        saver = InMemorySaver(
+            serde=Serializer(__unpack_ext_hook__=unpack_hook), **kwargs
+        )
+        saver.writes = MEMORY.writes
+        saver.blobs = MEMORY.blobs
+        saver.storage = MEMORY.storage
+        return saver
     return MEMORY
 
 

langgraph_storage/ops.py

@@ -15,6 +15,7 @@ from typing import Any, Literal, cast
 from uuid import UUID, uuid4
 
 import structlog
+from langgraph.checkpoint.serde.jsonplus import _msgpack_ext_hook_to_json
 from langgraph.pregel.debug import CheckpointPayload
 from langgraph.pregel.types import StateSnapshot
 from langgraph_sdk import Auth
@@ -949,6 +950,11 @@ class Threads(Authenticated):
                 checkpointer.writes[
                     (str(new_thread_id), checkpoint_ns, checkpoint_id)
                 ] = mapped
+            # Copy the blobs
+            for k in list(checkpointer.blobs):
+                if str(k[0]) == str(thread_id):
+                    new_key = (str(new_thread_id), *k[1:])
+                    checkpointer.blobs[new_key] = checkpointer.blobs[k]
 
             async def row_generator() -> AsyncIterator[Thread]:
                 yield new_thread
@@ -967,7 +973,7 @@ class Threads(Authenticated):
             ctx: Auth.types.BaseAuthContext | None = None,
         ) -> StateSnapshot:
             """Get state for a thread."""
-            checkpointer = Checkpointer(conn)
+            checkpointer = Checkpointer(conn, unpack_hook=_msgpack_ext_hook_to_json)
             thread_id = _ensure_uuid(config["configurable"]["thread_id"])
             # Auth will be applied here so no need to use filters downstream
             thread_iter = await Threads.get(conn, thread_id, ctx=ctx)
@@ -1186,7 +1192,11 @@ class Threads(Authenticated):
             # If graph_id exists, get state history
             if graph_id := thread_metadata.get("graph_id"):
                 async with get_graph(
-                    graph_id, thread_config, checkpointer=Checkpointer(conn)
+                    graph_id,
+                    thread_config,
+                    checkpointer=Checkpointer(
+                        conn, unpack_hook=_msgpack_ext_hook_to_json
+                    ),
                 ) as graph:
                     # Convert before parameter if it's a string
                     before_param = (