@langchain/langgraph 0.2.1 → 0.2.2

package/dist/graph/graph.cjs

@@ -50,7 +50,6 @@ class Branch {
         }
         let destinations;
         if (this.ends) {
-            // destinations = [r if isinstance(r, Send) else self.ends[r] for r in result]
             destinations = result.map((r) => ((0, constants_js_1._isSend)(r) ? r : this.ends[r]));
         }
         else {

package/dist/graph/graph.js

@@ -47,7 +47,6 @@ export class Branch {
         }
         let destinations;
         if (this.ends) {
-            // destinations = [r if isinstance(r, Send) else self.ends[r] for r in result]
             destinations = result.map((r) => (_isSend(r) ? r : this.ends[r]));
         }
         else {

package/dist/pregel/index.cjs

@@ -361,7 +361,10 @@ class Pregel extends runnables_1.Runnable {
                 task),
             },
         }));
-        // apply to checkpoint and save
+        if (saved !== undefined) {
+            await this.checkpointer.putWrites(checkpointConfig, task.writes, task.id);
+        }
+        // apply to checkpoint
         // TODO: Why does keyof StrRecord allow number and symbol?
         (0, algo_js_1._applyWrites)(checkpoint, channels, [task], this.checkpointer.getNextVersion.bind(this.checkpointer));
         const newVersions = (0, utils_js_1.getNewChannelVersions)(checkpointPreviousVersions, checkpoint.channel_versions);
@@ -455,10 +458,6 @@ class Pregel extends runnables_1.Runnable {
         // assign defaults
         const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer,] = this._defaults(inputConfig);
         let loop;
-        let backgroundError;
-        const onBackgroundError = (e) => {
-            backgroundError = e;
-        };
         try {
             loop = await loop_js_1.PregelLoop.initialize({
                 input,
@@ -466,17 +465,15 @@ class Pregel extends runnables_1.Runnable {
                 checkpointer,
                 nodes: this.nodes,
                 channelSpecs: this.channels,
-                onBackgroundError,
                 outputKeys,
                 streamKeys: this.streamChannelsAsIs,
             });
-            while (backgroundError === undefined &&
-                (await loop.tick({
-                    inputKeys: this.inputChannels,
-                    interruptAfter,
-                    interruptBefore,
-                    manager: runManager,
-                }))) {
+            while (await loop.tick({
+                inputKeys: this.inputChannels,
+                interruptAfter,
+                interruptBefore,
+                manager: runManager,
+            })) {
                 if (debug) {
                     (0, debug_js_1.printStepCheckpoint)(loop.checkpointMetadata.step, loop.channels, this.streamChannelsList);
                 }
@@ -542,10 +539,6 @@ class Pregel extends runnables_1.Runnable {
                     (0, debug_js_1.printStepWrites)(loop.step, loop.tasks.map((task) => task.writes).flat(), this.streamChannelsList);
                 }
             }
-            // Checkpointing failures
-            if (backgroundError !== undefined) {
-                throw backgroundError;
-            }
             while (loop.stream.length > 0) {
                 const nextItem = loop.stream.shift();
                 if (nextItem === undefined) {
@@ -567,6 +560,7 @@ class Pregel extends runnables_1.Runnable {
                     `limit by setting the "recursionLimit" config key.`,
                 ].join(" "));
             }
+            await Promise.all(loop?.checkpointerPromises ?? []);
             await runManager?.handleChainEnd((0, io_js_1.readChannels)(loop.channels, outputKeys));
         }
         catch (e) {
@@ -574,7 +568,7 @@ class Pregel extends runnables_1.Runnable {
             throw e;
         }
         finally {
-            await loop?.backgroundTasksPromise;
+            await Promise.all(loop?.checkpointerPromises ?? []);
         }
     }
     /**

package/dist/pregel/index.js

@@ -357,7 +357,10 @@ export class Pregel extends Runnable {
                 task),
             },
         }));
-        // apply to checkpoint and save
+        if (saved !== undefined) {
+            await this.checkpointer.putWrites(checkpointConfig, task.writes, task.id);
+        }
+        // apply to checkpoint
         // TODO: Why does keyof StrRecord allow number and symbol?
         _applyWrites(checkpoint, channels, [task], this.checkpointer.getNextVersion.bind(this.checkpointer));
         const newVersions = getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions);
@@ -451,10 +454,6 @@ export class Pregel extends Runnable {
         // assign defaults
         const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer,] = this._defaults(inputConfig);
         let loop;
-        let backgroundError;
-        const onBackgroundError = (e) => {
-            backgroundError = e;
-        };
         try {
             loop = await PregelLoop.initialize({
                 input,
@@ -462,17 +461,15 @@ export class Pregel extends Runnable {
                 checkpointer,
                 nodes: this.nodes,
                 channelSpecs: this.channels,
-                onBackgroundError,
                 outputKeys,
                 streamKeys: this.streamChannelsAsIs,
             });
-            while (backgroundError === undefined &&
-                (await loop.tick({
-                    inputKeys: this.inputChannels,
-                    interruptAfter,
-                    interruptBefore,
-                    manager: runManager,
-                }))) {
+            while (await loop.tick({
+                inputKeys: this.inputChannels,
+                interruptAfter,
+                interruptBefore,
+                manager: runManager,
+            })) {
                 if (debug) {
                     printStepCheckpoint(loop.checkpointMetadata.step, loop.channels, this.streamChannelsList);
                 }
@@ -538,10 +535,6 @@ export class Pregel extends Runnable {
                     printStepWrites(loop.step, loop.tasks.map((task) => task.writes).flat(), this.streamChannelsList);
                 }
             }
-            // Checkpointing failures
-            if (backgroundError !== undefined) {
-                throw backgroundError;
-            }
             while (loop.stream.length > 0) {
                 const nextItem = loop.stream.shift();
                 if (nextItem === undefined) {
@@ -563,6 +556,7 @@ export class Pregel extends Runnable {
                     `limit by setting the "recursionLimit" config key.`,
                 ].join(" "));
             }
+            await Promise.all(loop?.checkpointerPromises ?? []);
             await runManager?.handleChainEnd(readChannels(loop.channels, outputKeys));
         }
         catch (e) {
@@ -570,7 +564,7 @@ export class Pregel extends Runnable {
             throw e;
         }
         finally {
-            await loop?.backgroundTasksPromise;
+            await Promise.all(loop?.checkpointerPromises ?? []);
         }
     }
     /**

package/dist/pregel/loop.cjs

@@ -18,9 +18,6 @@ const INPUT_DONE = Symbol.for("INPUT_DONE");
 const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
 const DEFAULT_LOOP_LIMIT = 25;
 class PregelLoop {
-    get backgroundTasksPromise() {
-        return this._putCheckpointPromise;
-    }
     constructor(params) {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         Object.defineProperty(this, "input", {
@@ -113,6 +110,12 @@ class PregelLoop {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "skipDoneTasks", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -133,23 +136,23 @@ class PregelLoop {
             writable: true,
             value: new double_ended_queue_1.default()
         });
-        Object.defineProperty(this, "isNested", {
+        Object.defineProperty(this, "checkpointerPromises", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: []
         });
-        Object.defineProperty(this, "_putCheckpointPromise", {
+        Object.defineProperty(this, "isNested", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: Promise.resolve()
+            value: void 0
         });
-        Object.defineProperty(this, "onBackgroundError", {
+        Object.defineProperty(this, "_checkpointerChainedPromise", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: Promise.resolve()
         });
         this.input = params.input;
         this.config = params.config;
@@ -174,7 +177,7 @@ class PregelLoop {
         this.outputKeys = params.outputKeys;
         this.streamKeys = params.streamKeys;
         this.nodes = params.nodes;
-        this.onBackgroundError = params.onBackgroundError;
+        this.skipDoneTasks = this.config.configurable?.checkpoint_id === undefined;
     }
     static async initialize(params) {
         const saved = (await params.checkpointer?.getTuple(params.config)) ?? {
@@ -217,19 +220,13 @@ class PregelLoop {
             outputKeys: params.outputKeys ?? [],
             streamKeys: params.streamKeys ?? [],
             nodes: params.nodes,
-            onBackgroundError: params.onBackgroundError,
         });
     }
-    async _checkpointerPutAfterPrevious(input) {
-        try {
-            await this._putCheckpointPromise;
-        }
-        finally {
-            this._putCheckpointPromise =
-                this.checkpointer
-                    ?.put(input.config, input.checkpoint, input.metadata, input.newVersions)
-                    .catch(this.onBackgroundError) ?? Promise.resolve();
-        }
+    _checkpointerPutAfterPrevious(input) {
+        this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(() => {
+            return this.checkpointer?.put(input.config, input.checkpoint, input.metadata, input.newVersions);
+        });
+        this.checkpointerPromises.push(this._checkpointerChainedPromise);
     }
     /**
      * Put writes for a task, to be read by the next tick.
@@ -241,17 +238,16 @@ class PregelLoop {
             return [taskId, key, value];
         });
         this.checkpointPendingWrites.push(...pendingWrites);
-        if (this.checkpointer !== undefined) {
-            void this.checkpointer
-                .putWrites({
-                ...this.checkpointConfig,
-                configurable: {
-                    ...this.checkpointConfig.configurable,
-                    checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                    checkpoint_id: this.checkpoint.id,
-                },
-            }, writes, taskId)
-                .catch(this.onBackgroundError);
+        const putWritePromise = this.checkpointer?.putWrites({
+            ...this.checkpointConfig,
+            configurable: {
+                ...this.checkpointConfig.configurable,
+                checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
+                checkpoint_id: this.checkpoint.id,
+            },
+        }, writes, taskId);
+        if (putWritePromise !== undefined) {
+            this.checkpointerPromises.push(putWritePromise);
         }
         const task = this.tasks.find((task) => task.id === taskId);
         if (task !== undefined) {
@@ -320,7 +316,7 @@ class PregelLoop {
             return false;
         }
         // if there are pending writes from a previous loop, apply them
-        if (this.checkpointPendingWrites.length > 0) {
+        if (this.checkpointPendingWrites.length > 0 && this.skipDoneTasks) {
             for (const [tid, k, v] of this.checkpointPendingWrites) {
                 if (k === constants_js_1.ERROR || k === constants_js_1.INTERRUPT) {
                     continue;
@@ -392,7 +388,7 @@ class PregelLoop {
             // save input checkpoint
             await this._putCheckpoint({
                 source: "input",
-                writes: this.input ?? null,
+                writes: Object.fromEntries(inputWrites),
             });
         }
         // done with input

package/dist/pregel/loop.d.ts

@@ -9,7 +9,6 @@ export type PregelLoopInitializeParams = {
     input?: any;
     config: RunnableConfig;
     checkpointer?: BaseCheckpointSaver;
-    onBackgroundError: (e: Error) => void;
     outputKeys: string | string[];
     streamKeys: string | string[];
     nodes: Record<string, PregelNode>;
@@ -29,7 +28,6 @@ type PregelLoopParams = {
     stop: number;
     outputKeys: string | string[];
     streamKeys: string | string[];
-    onBackgroundError: (e: Error) => void;
     nodes: Record<string, PregelNode>;
 };
 export declare class PregelLoop {
@@ -48,13 +46,13 @@ export declare class PregelLoop {
     protected outputKeys: string | string[];
     protected streamKeys: string | string[];
     protected nodes: Record<string, PregelNode>;
+    protected skipDoneTasks: boolean;
     status: "pending" | "done" | "interrupt_before" | "interrupt_after" | "out_of_steps";
     tasks: PregelExecutableTask<any, any>[];
     stream: Deque<[StreamMode, any]>;
+    checkpointerPromises: Promise<unknown>[];
     protected isNested: boolean;
-    protected _putCheckpointPromise: Promise<unknown>;
-    onBackgroundError: (e: Error) => void;
-    get backgroundTasksPromise(): Promise<unknown>;
+    protected _checkpointerChainedPromise: Promise<unknown>;
     constructor(params: PregelLoopParams);
     static initialize(params: PregelLoopInitializeParams): Promise<PregelLoop>;
     protected _checkpointerPutAfterPrevious(input: {
@@ -62,7 +60,7 @@ export declare class PregelLoop {
         checkpoint: Checkpoint;
         metadata: CheckpointMetadata;
         newVersions: Record<string, string | number>;
-    }): Promise<void>;
+    }): void;
     /**
      * Put writes for a task, to be read by the next tick.
      * @param taskId

package/dist/pregel/loop.js

@@ -12,9 +12,6 @@ const INPUT_DONE = Symbol.for("INPUT_DONE");
 const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
 const DEFAULT_LOOP_LIMIT = 25;
 export class PregelLoop {
-    get backgroundTasksPromise() {
-        return this._putCheckpointPromise;
-    }
     constructor(params) {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         Object.defineProperty(this, "input", {
@@ -107,6 +104,12 @@ export class PregelLoop {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "skipDoneTasks", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -127,23 +130,23 @@ export class PregelLoop {
             writable: true,
             value: new Deque()
         });
-        Object.defineProperty(this, "isNested", {
+        Object.defineProperty(this, "checkpointerPromises", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: []
         });
-        Object.defineProperty(this, "_putCheckpointPromise", {
+        Object.defineProperty(this, "isNested", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: Promise.resolve()
+            value: void 0
         });
-        Object.defineProperty(this, "onBackgroundError", {
+        Object.defineProperty(this, "_checkpointerChainedPromise", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: Promise.resolve()
         });
         this.input = params.input;
         this.config = params.config;
@@ -168,7 +171,7 @@ export class PregelLoop {
         this.outputKeys = params.outputKeys;
         this.streamKeys = params.streamKeys;
         this.nodes = params.nodes;
-        this.onBackgroundError = params.onBackgroundError;
+        this.skipDoneTasks = this.config.configurable?.checkpoint_id === undefined;
     }
     static async initialize(params) {
         const saved = (await params.checkpointer?.getTuple(params.config)) ?? {
@@ -211,19 +214,13 @@ export class PregelLoop {
             outputKeys: params.outputKeys ?? [],
             streamKeys: params.streamKeys ?? [],
             nodes: params.nodes,
-            onBackgroundError: params.onBackgroundError,
         });
     }
-    async _checkpointerPutAfterPrevious(input) {
-        try {
-            await this._putCheckpointPromise;
-        }
-        finally {
-            this._putCheckpointPromise =
-                this.checkpointer
-                    ?.put(input.config, input.checkpoint, input.metadata, input.newVersions)
-                    .catch(this.onBackgroundError) ?? Promise.resolve();
-        }
+    _checkpointerPutAfterPrevious(input) {
+        this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(() => {
+            return this.checkpointer?.put(input.config, input.checkpoint, input.metadata, input.newVersions);
+        });
+        this.checkpointerPromises.push(this._checkpointerChainedPromise);
     }
     /**
      * Put writes for a task, to be read by the next tick.
@@ -235,17 +232,16 @@ export class PregelLoop {
             return [taskId, key, value];
         });
         this.checkpointPendingWrites.push(...pendingWrites);
-        if (this.checkpointer !== undefined) {
-            void this.checkpointer
-                .putWrites({
-                ...this.checkpointConfig,
-                configurable: {
-                    ...this.checkpointConfig.configurable,
-                    checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
-                    checkpoint_id: this.checkpoint.id,
-                },
-            }, writes, taskId)
-                .catch(this.onBackgroundError);
+        const putWritePromise = this.checkpointer?.putWrites({
+            ...this.checkpointConfig,
+            configurable: {
+                ...this.checkpointConfig.configurable,
+                checkpoint_ns: this.config.configurable?.checkpoint_ns ?? "",
+                checkpoint_id: this.checkpoint.id,
+            },
+        }, writes, taskId);
+        if (putWritePromise !== undefined) {
+            this.checkpointerPromises.push(putWritePromise);
         }
         const task = this.tasks.find((task) => task.id === taskId);
         if (task !== undefined) {
@@ -314,7 +310,7 @@ export class PregelLoop {
             return false;
         }
         // if there are pending writes from a previous loop, apply them
-        if (this.checkpointPendingWrites.length > 0) {
+        if (this.checkpointPendingWrites.length > 0 && this.skipDoneTasks) {
             for (const [tid, k, v] of this.checkpointPendingWrites) {
                 if (k === ERROR || k === INTERRUPT) {
                     continue;
@@ -386,7 +382,7 @@ export class PregelLoop {
             // save input checkpoint
             await this._putCheckpoint({
                 source: "input",
-                writes: this.input ?? null,
+                writes: Object.fromEntries(inputWrites),
             });
         }
         // done with input

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -31,7 +31,7 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/langgraph-checkpoint": "~0.0.4",
+    "@langchain/langgraph-checkpoint": "~0.0.5",
     "double-ended-queue": "^2.1.0-0",
     "uuid": "^10.0.0",
     "zod": "^3.23.8"