From a01c1c2bfa7a2ac2f89278426278c96c302f7a35 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 13 Aug 2024 10:27:48 -0700 Subject: [PATCH] Passthrough checkpoint in config even when current sub-graph didn't choose to use it --- libs/langgraph/langgraph/pregel/algo.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/pregel/algo.py b/libs/langgraph/langgraph/pregel/algo.py index 2ddbcd873..3e8b7e48b 100644 --- a/libs/langgraph/langgraph/pregel/algo.py +++ b/libs/langgraph/langgraph/pregel/algo.py @@ -407,7 +407,12 @@ def prepare_next_tasks( PregelTaskWrites(name, writes, triggers), config, ), - CONFIG_KEY_CHECKPOINTER: checkpointer, + CONFIG_KEY_CHECKPOINTER: ( + checkpointer + or config["configurable"].get( + CONFIG_KEY_CHECKPOINTER + ) + ), CONFIG_KEY_RESUMING: is_resuming, "checkpoint_id": checkpoint["id"], "checkpoint_ns": checkpoint_ns,