langgraph-checkpoint 2.0.2__tar.gz → 2.0.3__tar.gz

{langgraph_checkpoint-2.0.2 → langgraph_checkpoint-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-checkpoint
-Version: 2.0.2
+Version: 2.0.3
 Summary: Library with base interfaces for LangGraph checkpoint savers.
 Home-page: https://www.github.com/langchain-ai/langgraph
 License: MIT

{langgraph_checkpoint-2.0.2 → langgraph_checkpoint-2.0.3}/langgraph/checkpoint/memory/__init__.py

@@ -364,8 +364,12 @@ class MemorySaver(
         checkpoint_ns = config["configurable"]["checkpoint_ns"]
         checkpoint_id = config["configurable"]["checkpoint_id"]
         outer_key = (thread_id, checkpoint_ns, checkpoint_id)
+        outer_writes_ = self.writes.get(outer_key)
         for idx, (c, v) in enumerate(writes):
             inner_key = (task_id, WRITES_IDX_MAP.get(c, idx))
+            if inner_key[1] >= 0 and outer_writes_ and inner_key in outer_writes_:
+                continue
+
             self.writes[outer_key][inner_key] = (task_id, c, self.serde.dumps_typed(v))
 
     async def aget_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:

{langgraph_checkpoint-2.0.2 → langgraph_checkpoint-2.0.3}/langgraph/checkpoint/serde/jsonplus.py

@@ -25,7 +25,7 @@ from langchain_core.load.serializable import Serializable
 from zoneinfo import ZoneInfo
 
 from langgraph.checkpoint.serde.base import SerializerProtocol
-from langgraph.checkpoint.serde.types import SendProtocol
+from langgraph.checkpoint.serde.types import CommandProtocol, SendProtocol
 from langgraph.store.base import Item
 
 LC_REVIVER = Reviver()
@@ -122,6 +122,11 @@ class JsonPlusSerializer(SerializerProtocol):
             return self._encode_constructor_args(
                 obj.__class__, kwargs={"node": obj.node, "arg": obj.arg}
             )
+        elif isinstance(obj, CommandProtocol):
+            return self._encode_constructor_args(
+                obj.__class__,
+                kwargs={k: getattr(obj, k) for k in obj.__all_slots__},
+            )
         elif isinstance(obj, (bytes, bytearray)):
             return self._encode_constructor_args(
                 obj.__class__, method="fromhex", args=(obj.hex(),)
@@ -402,6 +407,17 @@ def _msgpack_default(obj: Any) -> Union[str, msgpack.ExtType]:
                 (obj.__class__.__module__, obj.__class__.__name__, (obj.node, obj.arg)),
             ),
         )
+    elif isinstance(obj, CommandProtocol):
+        return msgpack.ExtType(
+            EXT_CONSTRUCTOR_KW_ARGS,
+            _msgpack_enc(
+                (
+                    obj.__class__.__module__,
+                    obj.__class__.__name__,
+                    {k: getattr(obj, k) for k in obj.__all_slots__},
+                ),
+            ),
+        )
     elif dataclasses.is_dataclass(obj):
         # doesn't use dataclasses.asdict to avoid deepcopy and recursion
         return msgpack.ExtType(

{langgraph_checkpoint-2.0.2 → langgraph_checkpoint-2.0.3}/langgraph/checkpoint/serde/types.py

@@ -4,6 +4,7 @@ from typing import (
     Protocol,
     Sequence,
     TypeVar,
+    Union,
     runtime_checkable,
 )
 
@@ -48,3 +49,11 @@ class SendProtocol(Protocol):
     def __repr__(self) -> str: ...
 
     def __eq__(self, value: object) -> bool: ...
+
+
+@runtime_checkable
+class CommandProtocol(Protocol):
+    # Mirrors langgraph.types.Command
+    update: Optional[dict[str, Any]]
+    send: Union[Any, Sequence[Any]]
+    __all_slots__: set[str]

{langgraph_checkpoint-2.0.2 → langgraph_checkpoint-2.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint"
-version = "2.0.2"
+version = "2.0.3"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 authors = []
 license = "MIT"