Merge pull request #1329 from langchain-ai/nc/13aug/config-checkpoint-passthrough

Passthrough checkpoint in config even when current sub-graph didn't choose to use it
This commit is contained in:
Nuno Campos
2024-08-13 16:49:12 -07:00
committed by GitHub
+6 -1
View File
@@ -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,