Bumped langgraph-checkpoint to 3.0, updated calls from dumps/loads to dumps_typed/loads_typed, fixed tests

Author: caseyclements

libs/langgraph-checkpoint-mongodb/langgraph/checkpoint/mongodb/utils.py

@@ -41,7 +41,7 @@ def loads_metadata(metadata: dict[str, Any]) -> CheckpointMetadata:
             output[key] = loads_metadata(value)
         return output
     else:
-        return serde.loads(metadata)
+        return serde.loads_typed(metadata)
 
 
 def dumps_metadata(
@@ -57,4 +57,4 @@ def dumps_metadata(
             output[key] = dumps_metadata(value)
         return output
     else:
-        return serde.dumps(metadata)
+        return serde.dumps_typed(metadata)

libs/langgraph-checkpoint-mongodb/pyproject.toml

@@ -9,8 +9,8 @@ description = "Library with a MongoDB implementation of LangGraph checkpoint sav
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "langgraph-checkpoint>=2.0.23,<3.0.0",
-    "langchain-mongodb>=0.6.1",
+    "langgraph-checkpoint>=3.0.0",
+    "langchain-mongodb<1.0.0",
     "pymongo>=4.12,<4.16",
 ]
 
@@ -21,7 +21,6 @@ dev = [
     "langchain-ollama>=0.2.2",
     "langchain-openai>=0.2.14",
     "langgraph>=0.3.23",
-    "langgraph-checkpoint>=2.0.9",
     "pytest-asyncio>=0.21.1",
     "pytest>=7.2.1",
     "pytest-mock>=3.11.1",

libs/langgraph-checkpoint-mongodb/tests/unit_tests/test_sync.py

@@ -154,7 +154,8 @@ def test_nested_filter() -> None:
         assert (
             isinstance(doc["metadata"], dict)
             and isinstance(doc["metadata"]["writes"], dict)
-            and isinstance(doc["metadata"]["writes"]["message"], bytes)
+            and doc["metadata"]["writes"]["message"][0] == "msgpack"
+            and isinstance(doc["metadata"]["writes"]["message"][1], bytes)
         )
 
         # Test values of checkpoint

libs/langgraph-checkpoint-mongodb/tests/unit_tests/test_time_travel.py

@@ -53,12 +53,13 @@ def checkpointer(request: Any) -> Generator[MongoDBSaver]:
     saver = MongoDBSaver(
         client=client,
         db_name=db_name,
-        collection_name=checkpoint_collection_name,
-        WRITES_COLLECTION_NAME=writes_collection_name,
+        checkpoint_collection_name=checkpoint_collection_name,
+        writes_collection_name=writes_collection_name,
         ttl=request.param,
     )
 
     # Can use this to compare
+    # from langgraph.checkpoint.memory import InMemorySaver
     # saver = InMemorySaver()
 
     yield saver
@@ -93,7 +94,7 @@ def test(checkpointer: MongoDBSaver) -> None:
     graph = workflow.compile(checkpointer=checkpointer)
 
     # Run the graph
-    graph.invoke(input=initial_state, config=config)  # type:ignore[arg-type]
+    graph.invoke(input=initial_state, config=config, stream_mode="checkpoints")  # type:ignore[arg-type]
 
     # Check to see whether the final state is approved
     final_state = graph.get_state(config=config)
@@ -108,10 +109,12 @@ def test(checkpointer: MongoDBSaver) -> None:
 
     target_checkpoint = None
     for checkpoint in checkpoints:
-        # Look for checkpoint after increment but before final processing
+        # Look for checkpoint after add_expense but before validate_expense
         if (
-            checkpoint.metadata and checkpoint.metadata.get("step") == 1
-        ):  # Before validate node
+            checkpoint.metadata
+            and checkpoint.metadata.get("step") == 1
+            and "validate_expense" in checkpoint.next
+        ):
             target_checkpoint = checkpoint
             break
 
@@ -126,15 +129,16 @@ def test(checkpointer: MongoDBSaver) -> None:
     assert target_checkpoint
     past_state = graph.get_state(target_checkpoint.config)
 
-    # Update the expense amount to 200 that validate amounts
+    # Update the expense amount to 200 that validates amounts
     updated_state = dict(**past_state.values)
-    # updated_state = {}
-    updated_state["amount"] = 200
-    updated_state["version"] = 2
+    updated_state["amount"] += 100
+    updated_state["version"] += 1
     updated_state["messages"] += ["Updated state"]
 
     updated_config = graph.update_state(
-        config=target_checkpoint.config, values=updated_state
+        config=target_checkpoint.config,
+        values=updated_state,
+        as_node="add_expense",
     )
 
     # Continue from the checkpoint