@langchain/langgraph 0.3.11 → 0.4.0

package/dist/pregel/loop.cjs

@@ -124,12 +124,6 @@ class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "managed", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "checkpoint", {
             enumerable: true,
             configurable: true,
@@ -184,7 +178,7 @@ class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "checkpointDuring", {
+        Object.defineProperty(this, "durability", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -220,12 +214,6 @@ class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "prevCheckpoint", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -326,7 +314,6 @@ class PregelLoop {
         this.checkpointMetadata = params.checkpointMetadata;
         this.checkpointPreviousVersions = params.checkpointPreviousVersions;
         this.channels = params.channels;
-        this.managed = params.managed;
         this.checkpointPendingWrites = params.checkpointPendingWrites;
         this.step = params.step;
         this.stop = params.stop;
@@ -345,7 +332,7 @@ class PregelLoop {
         this.prevCheckpointConfig = params.prevCheckpointConfig;
         this.interruptAfter = params.interruptAfter;
         this.interruptBefore = params.interruptBefore;
-        this.checkpointDuring = params.checkpointDuring;
+        this.durability = params.durability;
         this.debug = params.debug;
         this.triggerToNodes = params.triggerToNodes;
     }
@@ -390,12 +377,7 @@ class PregelLoop {
         const saved = (await params.checkpointer?.getTuple(checkpointConfig)) ?? {
             config,
             checkpoint: (0, langgraph_checkpoint_1.emptyCheckpoint)(),
-            metadata: {
-                source: "input",
-                step: -2,
-                writes: null,
-                parents: {},
-            },
+            metadata: { source: "input", step: -2, parents: {} },
             pendingWrites: [],
         };
         checkpointConfig = {
@@ -420,7 +402,7 @@ class PregelLoop {
             : undefined;
         if (store) {
             // Start the store. This is a batch store, so it will run continuously
-            store.start();
+            await store.start();
         }
         return new PregelLoop({
             input: params.input,
@@ -432,7 +414,6 @@ class PregelLoop {
             prevCheckpointConfig,
             checkpointNamespace,
             channels,
-            managed: params.managed,
             isNested,
             manager: params.manager,
             skipDoneTasks,
@@ -448,7 +429,7 @@ class PregelLoop {
             cache: params.cache,
             interruptAfter: params.interruptAfter,
             interruptBefore: params.interruptBefore,
-            checkpointDuring: params.checkpointDuring,
+            durability: params.durability,
             debug: params.debug,
             triggerToNodes: params.triggerToNodes,
         });
@@ -459,13 +440,6 @@ class PregelLoop {
         });
         this.checkpointerPromises.push(this._checkpointerChainedPromise);
     }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async updateManagedValues(key, values) {
-        const mv = this.managed.get(key);
-        if (mv && "update" in mv && typeof mv.update === "function") {
-            await mv.update(values);
-        }
-    }
     /**
      * Put writes for a task, to be read by the next tick.
      * @param taskId
@@ -485,16 +459,12 @@ class PregelLoop {
         for (const [c, v] of writesCopy) {
             this.checkpointPendingWrites.push([taskId, c, v]);
         }
-        const putWritePromise = this.checkpointer?.putWrites({
-            ...this.checkpointConfig,
-            configurable: {
-                ...this.checkpointConfig.configurable,
-                checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                checkpoint_id: this.checkpoint.id,
-            },
-        }, writesCopy, taskId);
-        if (putWritePromise !== undefined) {
-            this.checkpointerPromises.push(putWritePromise);
+        const config = (0, index_js_1.patchConfigurable)(this.checkpointConfig, {
+            [constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
+            [constants_js_1.CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,
+        });
+        if (this.durability !== "exit" && this.checkpointer != null) {
+            this.checkpointerPromises.push(this.checkpointer.putWrites(config, writesCopy, taskId));
         }
         if (this.tasks) {
             this._outputWrites(taskId, writesCopy);
@@ -585,7 +555,7 @@ class PregelLoop {
      */
     async tick(params) {
         if (this.store && !this.store.isRunning) {
-            this.store?.start();
+            await this.store?.start();
         }
         const { inputKeys = [] } = params;
         if (this.status !== "pending") {
@@ -602,19 +572,13 @@ class PregelLoop {
             // finish superstep
             const writes = Object.values(this.tasks).flatMap((t) => t.writes);
             // All tasks have finished
-            const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
-            for (const [key, values] of Object.entries(managedValueWrites)) {
-                await this.updateManagedValues(key, values);
-            }
+            (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
             // produce values output
             const valuesOutput = await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, writes, this.channels), "values"));
             this._emit(valuesOutput);
             // clear pending writes
             this.checkpointPendingWrites = [];
-            await this._putCheckpoint({
-                source: "loop",
-                writes: (0, io_js_1.mapOutputUpdates)(this.outputKeys, Object.values(this.tasks).map((task) => [task, task.writes])).next().value ?? null,
-            });
+            await this._putCheckpoint({ source: "loop" });
             // after execution, check if we should interrupt
             if ((0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptAfter, Object.values(this.tasks))) {
                 this.status = "interrupt_after";
@@ -632,7 +596,7 @@ class PregelLoop {
             this.status = "out_of_steps";
             return false;
         }
-        const nextTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, this.config, true, {
+        const nextTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             isResuming: this.isResuming,
@@ -682,7 +646,7 @@ class PregelLoop {
     }
     async finishAndHandleError(error) {
         // persist current checkpoint and writes
-        if (!this.checkpointDuring &&
+        if (this.durability === "exit" &&
             // if it's a top graph
             (!this.isNested ||
                 // or a nested graph with error or interrupt
@@ -701,10 +665,7 @@ class PregelLoop {
             if (this.tasks !== undefined &&
                 this.checkpointPendingWrites.length > 0 &&
                 Object.values(this.tasks).some((task) => task.writes.length > 0)) {
-                const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
-                for (const [key, values] of Object.entries(managedValueWrites)) {
-                    await this.updateManagedValues(key, values);
-                }
+                (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
                 this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
             }
             // Emit INTERRUPT event
@@ -723,7 +684,7 @@ class PregelLoop {
             this.toInterrupt.push(task);
             return;
         }
-        const pushed = (0, algo_js_1._prepareSingleTask)([constants_js_1.PUSH, task.path ?? [], writeIdx, task.id, call], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, task.config ?? {}, true, {
+        const pushed = (0, algo_js_1._prepareSingleTask)([constants_js_1.PUSH, task.path ?? [], writeIdx, task.id, call], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, task.config ?? {}, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             manager: this.manager,
@@ -825,14 +786,14 @@ class PregelLoop {
             // we need to create a new checkpoint for Command(update=...) or Command(goto=...)
             // in case the result of Command(goto=...) is an interrupt.
             // If not done, the checkpoint containing the interrupt will be lost.
-            await this._putCheckpoint({ source: "input", writes: {} });
+            await this._putCheckpoint({ source: "input" });
             this.input = INPUT_DONE;
         }
         else {
             // map inputs to channel updates
             const inputWrites = await (0, utils_js_1.gatherIterator)((0, io_js_1.mapInput)(inputKeys, this.input));
             if (inputWrites.length > 0) {
-                const discardTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, this.config, true, { step: this.step });
+                const discardTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, { step: this.step });
                 (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(discardTasks).concat([
                     {
                         name: constants_js_1.INPUT,
@@ -841,10 +802,7 @@ class PregelLoop {
                     },
                 ]), this.checkpointerGetNextVersion, this.triggerToNodes);
                 // save input checkpoint
-                await this._putCheckpoint({
-                    source: "input",
-                    writes: Object.fromEntries(inputWrites),
-                });
+                await this._putCheckpoint({ source: "input" });
                 this.input = INPUT_DONE;
             }
             else if (!(constants_js_1.CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) {
@@ -895,7 +853,7 @@ class PregelLoop {
     }
     _putCheckpoint(inputMetadata) {
         const exiting = this.checkpointMetadata === inputMetadata;
-        const doCheckpoint = this.checkpointer != null && (this.checkpointDuring || exiting);
+        const doCheckpoint = this.checkpointer != null && (this.durability !== "exit" || exiting);
         const storeCheckpoint = (checkpoint) => {
             // store the previous checkpoint config for debug events
             this.prevCheckpointConfig = this.checkpointConfig?.configurable
@@ -906,13 +864,9 @@ class PregelLoop {
             // this is achieved by writing child checkpoints as progress is made
             // (so that error recovery / resuming from interrupt don't lose work)
             // but doing so always with an id equal to that of the parent checkpoint
-            this.checkpointConfig = {
-                ...this.checkpointConfig,
-                configurable: {
-                    ...this.checkpointConfig.configurable,
-                    checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                },
-            };
+            this.checkpointConfig = (0, index_js_1.patchConfigurable)(this.checkpointConfig, {
+                [constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
+            });
             const channelVersions = { ...this.checkpoint.channel_versions };
             const newVersions = (0, index_js_1.getNewChannelVersions)(this.checkpointPreviousVersions, channelVersions);
             this.checkpointPreviousVersions = channelVersions;
@@ -933,16 +887,6 @@ class PregelLoop {
                 },
             };
         };
-        // We need to retroactively store the previous checkpoint
-        // if it turns out that pending sends are scheduled.
-        // TODO: remove when `pending_sends` is removed from checkpoints
-        if (!exiting &&
-            !this.checkpointDuring &&
-            this.checkpointer != null &&
-            this.prevCheckpoint != null &&
-            this.checkpoint.pending_sends.length > 0) {
-            storeCheckpoint(this.prevCheckpoint);
-        }
         if (!exiting) {
             this.checkpointMetadata = {
                 ...inputMetadata,
@@ -950,11 +894,6 @@ class PregelLoop {
                 parents: this.config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP] ?? {},
             };
         }
-        // Store previous checkpoint in case pending_sends are scheduled
-        // for next checkpoint.
-        if (!this.checkpointDuring) {
-            this.prevCheckpoint = this.checkpoint;
-        }
         // create new checkpoint
         this.checkpoint = (0, base_js_1.createCheckpoint)(this.checkpoint, doCheckpoint ? this.channels : undefined, this.step, exiting ? { id: this.checkpoint.id } : undefined);
         // Bail if no checkpointer

package/dist/pregel/loop.d.ts

@@ -2,10 +2,9 @@ import type { RunnableConfig } from "@langchain/core/runnables";
 import type { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All, BaseStore, AsyncBatchedStore, BaseCache, CacheFullKey, CacheNamespace } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
-import { Call, PregelExecutableTask, StreamMode } from "./types.js";
+import type { Call, Durability, PregelExecutableTask, StreamMode } from "./types.js";
 import { Command } from "../constants.js";
 import { PregelNode } from "./read.js";
-import { ManagedValueMapping } from "../managed/base.js";
 import { LangGraphRunnableConfig } from "./runnable_types.js";
 import { IterableReadableWritableStream } from "./stream.js";
 export type PregelLoopInitializeParams = {
@@ -16,13 +15,12 @@ export type PregelLoopInitializeParams = {
     streamKeys: string | string[];
     nodes: Record<string, PregelNode>;
     channelSpecs: Record<string, BaseChannel>;
-    managed: ManagedValueMapping;
     stream: IterableReadableWritableStream;
     store?: BaseStore;
     cache?: BaseCache<PendingWrite<string>[]>;
     interruptAfter: string[] | All;
     interruptBefore: string[] | All;
-    checkpointDuring: boolean;
+    durability: Durability;
     manager?: CallbackManagerForChainRun;
     debug: boolean;
     triggerToNodes: Record<string, string[]>;
@@ -37,7 +35,6 @@ type PregelLoopParams = {
     checkpointPendingWrites: CheckpointPendingWrite[];
     checkpointConfig: RunnableConfig;
     channels: Record<string, BaseChannel>;
-    managed: ManagedValueMapping;
     step: number;
     stop: number;
     outputKeys: string | string[];
@@ -53,7 +50,7 @@ type PregelLoopParams = {
     prevCheckpointConfig: RunnableConfig | undefined;
     interruptAfter: string[] | All;
     interruptBefore: string[] | All;
-    checkpointDuring: boolean;
+    durability: Durability;
     debug: boolean;
     triggerToNodes: Record<string, string[]>;
 };
@@ -79,9 +76,8 @@ export declare class PregelLoop {
     output: any;
     config: LangGraphRunnableConfig;
     protected checkpointer?: BaseCheckpointSaver;
-    protected checkpointerGetNextVersion: (current: number | undefined, channel: BaseChannel) => number;
+    protected checkpointerGetNextVersion: (current: number | undefined) => number;
     channels: Record<string, BaseChannel>;
-    managed: ManagedValueMapping;
     protected checkpoint: Checkpoint;
     protected checkpointIdSaved: string | undefined;
     protected checkpointConfig: RunnableConfig;
@@ -91,13 +87,12 @@ export declare class PregelLoop {
     protected checkpointPreviousVersions: Record<string, string | number>;
     step: number;
     protected stop: number;
-    protected checkpointDuring: boolean;
+    protected durability: Durability;
     protected outputKeys: string | string[];
     protected streamKeys: string | string[];
     protected nodes: Record<string, PregelNode>;
     protected skipDoneTasks: boolean;
     protected prevCheckpointConfig: RunnableConfig | undefined;
-    protected prevCheckpoint: Checkpoint | undefined;
     status: "pending" | "done" | "interrupt_before" | "interrupt_after" | "out_of_steps";
     tasks: Record<string, PregelExecutableTask<any, any>>;
     stream: IterableReadableWritableStream;
@@ -121,7 +116,6 @@ export declare class PregelLoop {
         metadata: CheckpointMetadata;
         newVersions: Record<string, string | number>;
     }): void;
-    protected updateManagedValues(key: string, values: any[]): Promise<void>;
     /**
      * Put writes for a task, to be read by the next tick.
      * @param taskId

package/dist/pregel/loop.js

@@ -121,12 +121,6 @@ export class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "managed", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "checkpoint", {
             enumerable: true,
             configurable: true,
@@ -181,7 +175,7 @@ export class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "checkpointDuring", {
+        Object.defineProperty(this, "durability", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -217,12 +211,6 @@ export class PregelLoop {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "prevCheckpoint", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -323,7 +311,6 @@ export class PregelLoop {
         this.checkpointMetadata = params.checkpointMetadata;
         this.checkpointPreviousVersions = params.checkpointPreviousVersions;
         this.channels = params.channels;
-        this.managed = params.managed;
         this.checkpointPendingWrites = params.checkpointPendingWrites;
         this.step = params.step;
         this.stop = params.stop;
@@ -342,7 +329,7 @@ export class PregelLoop {
         this.prevCheckpointConfig = params.prevCheckpointConfig;
         this.interruptAfter = params.interruptAfter;
         this.interruptBefore = params.interruptBefore;
-        this.checkpointDuring = params.checkpointDuring;
+        this.durability = params.durability;
         this.debug = params.debug;
         this.triggerToNodes = params.triggerToNodes;
     }
@@ -387,12 +374,7 @@ export class PregelLoop {
         const saved = (await params.checkpointer?.getTuple(checkpointConfig)) ?? {
             config,
             checkpoint: emptyCheckpoint(),
-            metadata: {
-                source: "input",
-                step: -2,
-                writes: null,
-                parents: {},
-            },
+            metadata: { source: "input", step: -2, parents: {} },
             pendingWrites: [],
         };
         checkpointConfig = {
@@ -417,7 +399,7 @@ export class PregelLoop {
             : undefined;
         if (store) {
             // Start the store. This is a batch store, so it will run continuously
-            store.start();
+            await store.start();
         }
         return new PregelLoop({
             input: params.input,
@@ -429,7 +411,6 @@ export class PregelLoop {
             prevCheckpointConfig,
             checkpointNamespace,
             channels,
-            managed: params.managed,
             isNested,
             manager: params.manager,
             skipDoneTasks,
@@ -445,7 +426,7 @@ export class PregelLoop {
             cache: params.cache,
             interruptAfter: params.interruptAfter,
             interruptBefore: params.interruptBefore,
-            checkpointDuring: params.checkpointDuring,
+            durability: params.durability,
             debug: params.debug,
             triggerToNodes: params.triggerToNodes,
         });
@@ -456,13 +437,6 @@ export class PregelLoop {
         });
         this.checkpointerPromises.push(this._checkpointerChainedPromise);
     }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async updateManagedValues(key, values) {
-        const mv = this.managed.get(key);
-        if (mv && "update" in mv && typeof mv.update === "function") {
-            await mv.update(values);
-        }
-    }
     /**
      * Put writes for a task, to be read by the next tick.
      * @param taskId
@@ -482,16 +456,12 @@ export class PregelLoop {
         for (const [c, v] of writesCopy) {
             this.checkpointPendingWrites.push([taskId, c, v]);
         }
-        const putWritePromise = this.checkpointer?.putWrites({
-            ...this.checkpointConfig,
-            configurable: {
-                ...this.checkpointConfig.configurable,
-                checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                checkpoint_id: this.checkpoint.id,
-            },
-        }, writesCopy, taskId);
-        if (putWritePromise !== undefined) {
-            this.checkpointerPromises.push(putWritePromise);
+        const config = patchConfigurable(this.checkpointConfig, {
+            [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
+            [CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,
+        });
+        if (this.durability !== "exit" && this.checkpointer != null) {
+            this.checkpointerPromises.push(this.checkpointer.putWrites(config, writesCopy, taskId));
         }
         if (this.tasks) {
             this._outputWrites(taskId, writesCopy);
@@ -582,7 +552,7 @@ export class PregelLoop {
      */
     async tick(params) {
         if (this.store && !this.store.isRunning) {
-            this.store?.start();
+            await this.store?.start();
         }
         const { inputKeys = [] } = params;
         if (this.status !== "pending") {
@@ -599,19 +569,13 @@ export class PregelLoop {
             // finish superstep
             const writes = Object.values(this.tasks).flatMap((t) => t.writes);
             // All tasks have finished
-            const managedValueWrites = _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
-            for (const [key, values] of Object.entries(managedValueWrites)) {
-                await this.updateManagedValues(key, values);
-            }
+            _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
             // produce values output
             const valuesOutput = await gatherIterator(prefixGenerator(mapOutputValues(this.outputKeys, writes, this.channels), "values"));
             this._emit(valuesOutput);
             // clear pending writes
             this.checkpointPendingWrites = [];
-            await this._putCheckpoint({
-                source: "loop",
-                writes: mapOutputUpdates(this.outputKeys, Object.values(this.tasks).map((task) => [task, task.writes])).next().value ?? null,
-            });
+            await this._putCheckpoint({ source: "loop" });
             // after execution, check if we should interrupt
             if (shouldInterrupt(this.checkpoint, this.interruptAfter, Object.values(this.tasks))) {
                 this.status = "interrupt_after";
@@ -629,7 +593,7 @@ export class PregelLoop {
             this.status = "out_of_steps";
             return false;
         }
-        const nextTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, this.config, true, {
+        const nextTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             isResuming: this.isResuming,
@@ -679,7 +643,7 @@ export class PregelLoop {
     }
     async finishAndHandleError(error) {
         // persist current checkpoint and writes
-        if (!this.checkpointDuring &&
+        if (this.durability === "exit" &&
             // if it's a top graph
             (!this.isNested ||
                 // or a nested graph with error or interrupt
@@ -698,10 +662,7 @@ export class PregelLoop {
             if (this.tasks !== undefined &&
                 this.checkpointPendingWrites.length > 0 &&
                 Object.values(this.tasks).some((task) => task.writes.length > 0)) {
-                const managedValueWrites = _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
-                for (const [key, values] of Object.entries(managedValueWrites)) {
-                    await this.updateManagedValues(key, values);
-                }
+                _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
                 this._emit(gatherIteratorSync(prefixGenerator(mapOutputValues(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
             }
             // Emit INTERRUPT event
@@ -720,7 +681,7 @@ export class PregelLoop {
             this.toInterrupt.push(task);
             return;
         }
-        const pushed = _prepareSingleTask([PUSH, task.path ?? [], writeIdx, task.id, call], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, task.config ?? {}, true, {
+        const pushed = _prepareSingleTask([PUSH, task.path ?? [], writeIdx, task.id, call], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, task.config ?? {}, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             manager: this.manager,
@@ -822,14 +783,14 @@ export class PregelLoop {
             // we need to create a new checkpoint for Command(update=...) or Command(goto=...)
             // in case the result of Command(goto=...) is an interrupt.
             // If not done, the checkpoint containing the interrupt will be lost.
-            await this._putCheckpoint({ source: "input", writes: {} });
+            await this._putCheckpoint({ source: "input" });
             this.input = INPUT_DONE;
         }
         else {
             // map inputs to channel updates
             const inputWrites = await gatherIterator(mapInput(inputKeys, this.input));
             if (inputWrites.length > 0) {
-                const discardTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.managed, this.config, true, { step: this.step });
+                const discardTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, { step: this.step });
                 _applyWrites(this.checkpoint, this.channels, Object.values(discardTasks).concat([
                     {
                         name: INPUT,
@@ -838,10 +799,7 @@ export class PregelLoop {
                     },
                 ]), this.checkpointerGetNextVersion, this.triggerToNodes);
                 // save input checkpoint
-                await this._putCheckpoint({
-                    source: "input",
-                    writes: Object.fromEntries(inputWrites),
-                });
+                await this._putCheckpoint({ source: "input" });
                 this.input = INPUT_DONE;
             }
             else if (!(CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) {
@@ -892,7 +850,7 @@ export class PregelLoop {
     }
     _putCheckpoint(inputMetadata) {
         const exiting = this.checkpointMetadata === inputMetadata;
-        const doCheckpoint = this.checkpointer != null && (this.checkpointDuring || exiting);
+        const doCheckpoint = this.checkpointer != null && (this.durability !== "exit" || exiting);
         const storeCheckpoint = (checkpoint) => {
             // store the previous checkpoint config for debug events
             this.prevCheckpointConfig = this.checkpointConfig?.configurable
@@ -903,13 +861,9 @@ export class PregelLoop {
             // this is achieved by writing child checkpoints as progress is made
             // (so that error recovery / resuming from interrupt don't lose work)
             // but doing so always with an id equal to that of the parent checkpoint
-            this.checkpointConfig = {
-                ...this.checkpointConfig,
-                configurable: {
-                    ...this.checkpointConfig.configurable,
-                    checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                },
-            };
+            this.checkpointConfig = patchConfigurable(this.checkpointConfig, {
+                [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
+            });
             const channelVersions = { ...this.checkpoint.channel_versions };
             const newVersions = getNewChannelVersions(this.checkpointPreviousVersions, channelVersions);
             this.checkpointPreviousVersions = channelVersions;
@@ -930,16 +884,6 @@ export class PregelLoop {
                 },
             };
         };
-        // We need to retroactively store the previous checkpoint
-        // if it turns out that pending sends are scheduled.
-        // TODO: remove when `pending_sends` is removed from checkpoints
-        if (!exiting &&
-            !this.checkpointDuring &&
-            this.checkpointer != null &&
-            this.prevCheckpoint != null &&
-            this.checkpoint.pending_sends.length > 0) {
-            storeCheckpoint(this.prevCheckpoint);
-        }
         if (!exiting) {
             this.checkpointMetadata = {
                 ...inputMetadata,
@@ -947,11 +891,6 @@ export class PregelLoop {
                 parents: this.config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP] ?? {},
             };
         }
-        // Store previous checkpoint in case pending_sends are scheduled
-        // for next checkpoint.
-        if (!this.checkpointDuring) {
-            this.prevCheckpoint = this.checkpoint;
-        }
         // create new checkpoint
         this.checkpoint = createCheckpoint(this.checkpoint, doCheckpoint ? this.channels : undefined, this.step, exiting ? { id: this.checkpoint.id } : undefined);
         // Bail if no checkpointer