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,