langgraph-api 0.0.33__tar.gz → 0.0.34__tar.gz

{langgraph_api-0.0.33 → langgraph_api-0.0.34}/PKG-INFO

@@ -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 → langgraph_api-0.0.34}/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 → langgraph_api-0.0.34}/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_api-0.0.33 → langgraph_api-0.0.34}/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 = (

{langgraph_api-0.0.33 → langgraph_api-0.0.34}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.33"
+version = "0.0.34"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -24,7 +24,7 @@ sse-starlette = ">=2.1.0,<2.2.0"
 starlette = ">=0.38.6"
 watchfiles = ">=0.13"
 langgraph = ">=0.2.56,<0.4.0"
-langgraph-checkpoint = ">=2.0.21,<3.0"
+langgraph-checkpoint = ">=2.0.23,<3.0"
 orjson = ">=3.9.7"
 uvicorn = ">=0.26.0"
 langsmith = ">=0.1.63,<0.4.0"