Update put to be consistent with put_writes wrt ttl

caseyclements · caseyclements · commit e21428a64286 · 2025-09-23T13:58:03.000-04:00
diff --git a/libs/langgraph-checkpoint-mongodb/langgraph/checkpoint/mongodb/saver.py b/libs/langgraph-checkpoint-mongodb/langgraph/checkpoint/mongodb/saver.py
@@ -390,7 +390,7 @@ def put(
             "checkpoint_id": checkpoint_id,
         }
         if self.ttl:
-            upsert_query["created_at"] = datetime.now()
+            doc["created_at"] = datetime.now()
 
         self.checkpoint_collection.update_one(upsert_query, {"$set": doc}, upsert=True)
         return {

Original file line number	Diff line number	Diff line change
`@@ -390,7 +390,7 @@ def put(`
`390`	`390`	`"checkpoint_id": checkpoint_id,`
`391`	`391`	`}`
`392`	`392`	`if self.ttl:`
`393`		`- upsert_query["created_at"] = datetime.now()`
	`393`	`+ doc["created_at"] = datetime.now()`
`394`	`394`
`395`	`395`	`self.checkpoint_collection.update_one(upsert_query, {"$set": doc}, upsert=True)`
`396`	`396`	`return {`