Avoid copying checkpoint when fetching at beginning of loop

langchain-ai · Sep 12, 2024 · 43ec65c · 43ec65c
1 parent d669b5c
commit 43ec65c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py
@@ -659,7 +659,7 @@ def __enter__(self) -> Self:
                 **saved.config.get("configurable", {}),
             },
         }
-        self.checkpoint = copy_checkpoint(saved.checkpoint)
+        self.checkpoint = saved.checkpoint
         self.checkpoint_metadata = saved.metadata
         self.checkpoint_pending_writes = (
             [(str(tid), k, v) for tid, k, v in saved.pending_writes]
@@ -777,7 +777,7 @@ async def __aenter__(self) -> Self:
                 **saved.config.get("configurable", {}),
             },
         }
-        self.checkpoint = copy_checkpoint(saved.checkpoint)
+        self.checkpoint = saved.checkpoint
         self.checkpoint_metadata = saved.metadata
         self.checkpoint_pending_writes = (
             [(str(tid), k, v) for tid, k, v in saved.pending_writes]