@langchain/langgraph 0.0.14 → 0.0.16

package/dist/checkpoint/base.cjs

@@ -1,7 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
+exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = exports.getVersionSeen = exports.getChannelVersion = void 0;
+const base_js_1 = require("../serde/base.cjs");
 const id_js_1 = require("./id.cjs");
+function getChannelVersion(checkpoint, channel) {
+    return checkpoint.channel_versions[channel] ?? 0;
+}
+exports.getChannelVersion = getChannelVersion;
+function getVersionSeen(checkpoint, node, channel) {
+    return checkpoint.versions_seen[node]?.[channel] ?? 0;
+}
+exports.getVersionSeen = getVersionSeen;
 function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
         return obj;
@@ -43,7 +52,7 @@ class BaseCheckpointSaver {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: JSON
+            value: base_js_1.DefaultSerializer
         });
         this.serde = serde || this.serde;
     }

package/dist/checkpoint/base.d.ts

@@ -15,7 +15,7 @@ export interface CheckpointMetadata {
      * ... for the nth checkpoint afterwards. */
     writes?: Record<string, unknown>;
 }
-export interface Checkpoint {
+export interface Checkpoint<N extends string = string, C extends string = string> {
     /**
      * Version number
      */
@@ -31,21 +31,23 @@ export interface Checkpoint {
     /**
      * @default {}
      */
-    channel_values: Record<string, unknown>;
+    channel_values: Record<C, unknown>;
     /**
      * @default {}
      */
-    channel_versions: Record<string, number>;
+    channel_versions: Record<C, number>;
     /**
      * @default {}
      */
-    versions_seen: Record<string, Record<string, number>>;
+    versions_seen: Record<N, Record<C, number>>;
 }
 export interface ReadonlyCheckpoint extends Readonly<Checkpoint> {
     readonly channel_values: Readonly<Record<string, unknown>>;
     readonly channel_versions: Readonly<Record<string, number>>;
     readonly versions_seen: Readonly<Record<string, Readonly<Record<string, number>>>>;
 }
+export declare function getChannelVersion(checkpoint: ReadonlyCheckpoint, channel: string): number;
+export declare function getVersionSeen(checkpoint: ReadonlyCheckpoint, node: string, channel: string): number;
 export declare function deepCopy<T>(obj: T): T;
 export declare function emptyCheckpoint(): Checkpoint;
 export declare function copyCheckpoint(checkpoint: ReadonlyCheckpoint): Checkpoint;

package/dist/checkpoint/base.js

@@ -1,4 +1,11 @@
+import { DefaultSerializer } from "../serde/base.js";
 import { uuid6 } from "./id.js";
+export function getChannelVersion(checkpoint, channel) {
+    return checkpoint.channel_versions[channel] ?? 0;
+}
+export function getVersionSeen(checkpoint, node, channel) {
+    return checkpoint.versions_seen[node]?.[channel] ?? 0;
+}
 export function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
         return obj;
@@ -37,7 +44,7 @@ export class BaseCheckpointSaver {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: JSON
+            value: DefaultSerializer
         });
         this.serde = serde || this.serde;
     }

package/dist/checkpoint/memory.cjs

@@ -22,8 +22,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             if (checkpoint) {
                 return {
                     config,
-                    checkpoint: this.serde.parse(checkpoint[0]),
-                    metadata: this.serde.parse(checkpoint[1]),
+                    checkpoint: (await this.serde.parse(checkpoint[0])),
+                    metadata: (await this.serde.parse(checkpoint[1])),
                 };
             }
         }
@@ -33,8 +33,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const checkpoint = checkpoints[maxThreadTs];
                 return {
                     config: { configurable: { thread_id, checkpoint_id: maxThreadTs } },
-                    checkpoint: this.serde.parse(checkpoint[0]),
-                    metadata: this.serde.parse(checkpoint[1]),
+                    checkpoint: (await this.serde.parse(checkpoint[0])),
+                    metadata: (await this.serde.parse(checkpoint[1])),
                 };
             }
         }
@@ -50,8 +50,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             .slice(0, limit)) {
             yield {
                 config: { configurable: { thread_id, checkpoint_id } },
-                checkpoint: this.serde.parse(checkpoint[0]),
-                metadata: this.serde.parse(checkpoint[1]),
+                checkpoint: (await this.serde.parse(checkpoint[0])),
+                metadata: (await this.serde.parse(checkpoint[1])),
             };
         }
     }

package/dist/checkpoint/memory.js

@@ -19,8 +19,8 @@ export class MemorySaver extends BaseCheckpointSaver {
             if (checkpoint) {
                 return {
                     config,
-                    checkpoint: this.serde.parse(checkpoint[0]),
-                    metadata: this.serde.parse(checkpoint[1]),
+                    checkpoint: (await this.serde.parse(checkpoint[0])),
+                    metadata: (await this.serde.parse(checkpoint[1])),
                 };
             }
         }
@@ -30,8 +30,8 @@ export class MemorySaver extends BaseCheckpointSaver {
                 const checkpoint = checkpoints[maxThreadTs];
                 return {
                     config: { configurable: { thread_id, checkpoint_id: maxThreadTs } },
-                    checkpoint: this.serde.parse(checkpoint[0]),
-                    metadata: this.serde.parse(checkpoint[1]),
+                    checkpoint: (await this.serde.parse(checkpoint[0])),
+                    metadata: (await this.serde.parse(checkpoint[1])),
                 };
             }
         }
@@ -47,8 +47,8 @@ export class MemorySaver extends BaseCheckpointSaver {
             .slice(0, limit)) {
             yield {
                 config: { configurable: { thread_id, checkpoint_id } },
-                checkpoint: this.serde.parse(checkpoint[0]),
-                metadata: this.serde.parse(checkpoint[1]),
+                checkpoint: (await this.serde.parse(checkpoint[0])),
+                metadata: (await this.serde.parse(checkpoint[1])),
             };
         }
     }

package/dist/checkpoint/sqlite.cjs

@@ -61,8 +61,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                 if (row) {
                     return {
                         config,
-                        checkpoint: this.serde.parse(row.checkpoint),
-                        metadata: this.serde.parse(row.metadata),
+                        checkpoint: (await this.serde.parse(row.checkpoint)),
+                        metadata: (await this.serde.parse(row.metadata)),
                         parentConfig: row.parent_id
                             ? {
                                 configurable: {
@@ -91,8 +91,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                             checkpoint_id: row.checkpoint_id,
                         },
                     },
-                    checkpoint: this.serde.parse(row.checkpoint),
-                    metadata: this.serde.parse(row.metadata),
+                    checkpoint: (await this.serde.parse(row.checkpoint)),
+                    metadata: (await this.serde.parse(row.metadata)),
                     parentConfig: row.parent_id
                         ? {
                             configurable: {
@@ -125,8 +125,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                                 checkpoint_id: row.checkpoint_id,
                             },
                         },
-                        checkpoint: this.serde.parse(row.checkpoint),
-                        metadata: this.serde.parse(row.metadata),
+                        checkpoint: (await this.serde.parse(row.checkpoint)),
+                        metadata: (await this.serde.parse(row.metadata)),
                         parentConfig: row.parent_id
                             ? {
                                 configurable: {

package/dist/checkpoint/sqlite.js

@@ -55,8 +55,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                 if (row) {
                     return {
                         config,
-                        checkpoint: this.serde.parse(row.checkpoint),
-                        metadata: this.serde.parse(row.metadata),
+                        checkpoint: (await this.serde.parse(row.checkpoint)),
+                        metadata: (await this.serde.parse(row.metadata)),
                         parentConfig: row.parent_id
                             ? {
                                 configurable: {
@@ -85,8 +85,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                             checkpoint_id: row.checkpoint_id,
                         },
                     },
-                    checkpoint: this.serde.parse(row.checkpoint),
-                    metadata: this.serde.parse(row.metadata),
+                    checkpoint: (await this.serde.parse(row.checkpoint)),
+                    metadata: (await this.serde.parse(row.metadata)),
                     parentConfig: row.parent_id
                         ? {
                             configurable: {
@@ -119,8 +119,8 @@ CREATE TABLE IF NOT EXISTS checkpoints (
                                 checkpoint_id: row.checkpoint_id,
                             },
                         },
-                        checkpoint: this.serde.parse(row.checkpoint),
-                        metadata: this.serde.parse(row.metadata),
+                        checkpoint: (await this.serde.parse(row.checkpoint)),
+                        metadata: (await this.serde.parse(row.metadata)),
                         parentConfig: row.parent_id
                             ? {
                                 configurable: {

package/dist/pregel/index.cjs

@@ -551,8 +551,8 @@ async function executeTasks(tasks, stepTimeout) {
     }
 }
 function _shouldInterrupt(checkpoint, interruptNodes, snapshotChannels, tasks) {
-    const seen = checkpoint.versions_seen[constants_js_1.INTERRUPT];
-    const anySnapshotChannelUpdated = snapshotChannels.some((chan) => checkpoint.channel_versions[chan] > seen?.[chan]);
+    const anySnapshotChannelUpdated = snapshotChannels.some((chan) => (0, base_js_2.getChannelVersion)(checkpoint, chan) >
+        (0, base_js_2.getVersionSeen)(checkpoint, constants_js_1.INTERRUPT, chan));
     const anyTaskNodeInInterruptNodes = tasks.some((task) => interruptNodes === "*"
         ? !task.config?.tags?.includes(constants_js_1.TAG_HIDDEN)
         : interruptNodes.includes(task.name));
@@ -627,11 +627,6 @@ function _prepareNextTasks(checkpoint, processes, channels, forExecution) {
     // Check if any processes should be run in next step
     // If so, prepare the values to be passed to them
     for (const [name, proc] of Object.entries(processes)) {
-        let seen = newCheckpoint.versions_seen[name];
-        if (!seen) {
-            newCheckpoint.versions_seen[name] = {};
-            seen = newCheckpoint.versions_seen[name];
-        }
         // If any of the channels read by this process were updated
         if (proc.triggers
             .filter((chan) => {
@@ -643,7 +638,8 @@ function _prepareNextTasks(checkpoint, processes, channels, forExecution) {
                 return false;
             }
         })
-            .some((chan) => newCheckpoint.channel_versions[chan] > (seen[chan] ?? 0))) {
+            .some((chan) => (0, base_js_2.getChannelVersion)(newCheckpoint, chan) >
+            (0, base_js_2.getVersionSeen)(newCheckpoint, name, chan))) {
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             let val;
             // If all trigger channels subscribed by this process are not empty
@@ -696,11 +692,14 @@ function _prepareNextTasks(checkpoint, processes, channels, forExecution) {
             }
             if (forExecution) {
                 // Update seen versions
+                if (!newCheckpoint.versions_seen[name]) {
+                    newCheckpoint.versions_seen[name] = {};
+                }
                 proc.triggers.forEach((chan) => {
                     const version = newCheckpoint.channel_versions[chan];
                     if (version !== undefined) {
                         // side effect: updates newCheckpoint
-                        seen[chan] = version;
+                        newCheckpoint.versions_seen[name][chan] = version;
                     }
                 });
                 const node = proc.getNode();

package/dist/pregel/index.js

@@ -2,7 +2,7 @@
 import { Runnable, RunnableSequence, _coerceToRunnable, ensureConfig, patchConfig, } from "@langchain/core/runnables";
 import { IterableReadableStream } from "@langchain/core/utils/stream";
 import { createCheckpoint, emptyChannels, } from "../channels/base.js";
-import { copyCheckpoint, emptyCheckpoint, } from "../checkpoint/base.js";
+import { copyCheckpoint, emptyCheckpoint, getChannelVersion, getVersionSeen, } from "../checkpoint/base.js";
 import { PregelNode } from "./read.js";
 import { validateGraph, validateKeys } from "./validate.js";
 import { mapInput, mapOutputUpdates, mapOutputValues, readChannel, readChannels, single, } from "./io.js";
@@ -546,8 +546,8 @@ async function executeTasks(tasks, stepTimeout) {
     }
 }
 export function _shouldInterrupt(checkpoint, interruptNodes, snapshotChannels, tasks) {
-    const seen = checkpoint.versions_seen[INTERRUPT];
-    const anySnapshotChannelUpdated = snapshotChannels.some((chan) => checkpoint.channel_versions[chan] > seen?.[chan]);
+    const anySnapshotChannelUpdated = snapshotChannels.some((chan) => getChannelVersion(checkpoint, chan) >
+        getVersionSeen(checkpoint, INTERRUPT, chan));
     const anyTaskNodeInInterruptNodes = tasks.some((task) => interruptNodes === "*"
         ? !task.config?.tags?.includes(TAG_HIDDEN)
         : interruptNodes.includes(task.name));
@@ -619,11 +619,6 @@ export function _prepareNextTasks(checkpoint, processes, channels, forExecution)
     // Check if any processes should be run in next step
     // If so, prepare the values to be passed to them
     for (const [name, proc] of Object.entries(processes)) {
-        let seen = newCheckpoint.versions_seen[name];
-        if (!seen) {
-            newCheckpoint.versions_seen[name] = {};
-            seen = newCheckpoint.versions_seen[name];
-        }
         // If any of the channels read by this process were updated
         if (proc.triggers
             .filter((chan) => {
@@ -635,7 +630,8 @@ export function _prepareNextTasks(checkpoint, processes, channels, forExecution)
                 return false;
             }
         })
-            .some((chan) => newCheckpoint.channel_versions[chan] > (seen[chan] ?? 0))) {
+            .some((chan) => getChannelVersion(newCheckpoint, chan) >
+            getVersionSeen(newCheckpoint, name, chan))) {
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             let val;
             // If all trigger channels subscribed by this process are not empty
@@ -688,11 +684,14 @@ export function _prepareNextTasks(checkpoint, processes, channels, forExecution)
             }
             if (forExecution) {
                 // Update seen versions
+                if (!newCheckpoint.versions_seen[name]) {
+                    newCheckpoint.versions_seen[name] = {};
+                }
                 proc.triggers.forEach((chan) => {
                     const version = newCheckpoint.channel_versions[chan];
                     if (version !== undefined) {
                         // side effect: updates newCheckpoint
-                        seen[chan] = version;
+                        newCheckpoint.versions_seen[name][chan] = version;
                     }
                 });
                 const node = proc.getNode();

package/dist/serde/base.cjs

@@ -1,2 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.DefaultSerializer = void 0;
+const load_1 = require("@langchain/core/load");
+exports.DefaultSerializer = {
+    stringify: JSON.stringify,
+    parse: load_1.load,
+};

package/dist/serde/base.d.ts

@@ -1,4 +1,12 @@
+import { load } from "@langchain/core/load";
 export interface SerializerProtocol<D> {
     stringify(obj: D): string;
-    parse(data: string): D;
+    parse(data: string): Promise<D>;
 }
+export declare const DefaultSerializer: {
+    stringify: {
+        (value: any, replacer?: ((this: any, key: string, value: any) => any) | undefined, space?: string | number | undefined): string;
+        (value: any, replacer?: (string | number)[] | null | undefined, space?: string | number | undefined): string;
+    };
+    parse: typeof load;
+};

package/dist/serde/base.js

@@ -1 +1,5 @@
-export {};
+import { load } from "@langchain/core/load";
+export const DefaultSerializer = {
+    stringify: JSON.stringify,
+    parse: load,
+};

package/dist/tests/pregel.test.js

@@ -209,6 +209,33 @@ describe("_shouldInterrupt", () => {
             },
         ])).toBe(true);
     });
+    it("should return true if any snapshot channel has been updated since last interrupt and any channel written to is in interrupt nodes list", () => {
+        // set up test
+        const checkpoint = {
+            v: 1,
+            id: uuid6(-1),
+            ts: "2024-04-19T17:19:07.952Z",
+            channel_values: {
+                channel1: "channel1value",
+            },
+            channel_versions: {
+                channel1: 2, // current channel version is greater than last version seen
+            },
+            versions_seen: {},
+        };
+        const interruptNodes = ["node1"];
+        const snapshotChannels = ["channel1"];
+        // call method / assertions
+        expect(_shouldInterrupt(checkpoint, interruptNodes, snapshotChannels, [
+            {
+                name: "node1",
+                input: undefined,
+                proc: new RunnablePassthrough(),
+                writes: [],
+                config: undefined,
+            },
+        ])).toBe(true);
+    });
     it("should return false if all snapshot channels have not been updated", () => {
         // set up test
         const checkpoint = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.0.14",
+  "version": "0.0.16",
   "description": "LangGraph",
   "type": "module",
   "engines": {