From 66b9a7dee7e807e4ea32c0f3888ea55aceaa9831 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 15 Nov 2024 16:56:17 -0800 Subject: [PATCH] lib: When copying checkpoint, make it a child of the parent --- libs/langgraph/langgraph/pregel/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index d2d693bef..f49a2207a 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -961,7 +961,7 @@ class Pregel(PregelProtocol): next_checkpoint = create_checkpoint(checkpoint, None, step) # copy checkpoint next_config = checkpointer.put( - checkpoint_config, + saved.parent_config or saved.config if saved else checkpoint_config, next_checkpoint, { **checkpoint_metadata, @@ -1222,7 +1222,7 @@ class Pregel(PregelProtocol): next_checkpoint = create_checkpoint(checkpoint, None, step) # copy checkpoint next_config = await checkpointer.aput( - checkpoint_config, + saved.parent_config or saved.config if saved else checkpoint_config, next_checkpoint, { **checkpoint_metadata,