Passthrough checkpoint in config even when current sub-graph didn't choose to use it

2026-09-09 11:17:53 +02:00 · 2024-08-13 10:27:48 -07:00
parent e2a72bf511
commit a01c1c2bfa
1 changed files with 6 additions and 1 deletions
@@ -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,