@langchain/langgraph 0.0.15 → 0.0.17

package/dist/checkpoint/base.cjs

@@ -1,8 +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;

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,5 +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;

package/dist/pregel/index.cjs

@@ -406,7 +406,10 @@ class Pregel extends runnables_1.Runnable {
             else {
                 checkpoint = (0, base_js_2.copyCheckpoint)(checkpoint);
                 for (const k of this.streamChannelsList) {
-                    const version = checkpoint.channel_versions[k];
+                    const version = checkpoint.channel_versions[k] ?? 0;
+                    if (!checkpoint.versions_seen[constants_js_1.INTERRUPT]) {
+                        checkpoint.versions_seen[constants_js_1.INTERRUPT] = {};
+                    }
                     checkpoint.versions_seen[constants_js_1.INTERRUPT][k] = version;
                 }
             }
@@ -551,8 +554,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 +630,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 +641,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 +695,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";
@@ -402,7 +402,10 @@ export class Pregel extends Runnable {
             else {
                 checkpoint = copyCheckpoint(checkpoint);
                 for (const k of this.streamChannelsList) {
-                    const version = checkpoint.channel_versions[k];
+                    const version = checkpoint.channel_versions[k] ?? 0;
+                    if (!checkpoint.versions_seen[INTERRUPT]) {
+                        checkpoint.versions_seen[INTERRUPT] = {};
+                    }
                     checkpoint.versions_seen[INTERRUPT][k] = version;
                 }
             }
@@ -546,8 +549,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 +622,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 +633,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 +687,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/pregel/io.cjs

@@ -53,11 +53,10 @@ function* mapInput(inputChannels,
 chunk
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 ) {
-    if (chunk) {
+    if (chunk !== undefined && chunk !== null) {
         if (Array.isArray(inputChannels) &&
             typeof chunk === "object" &&
-            !Array.isArray(chunk) &&
-            !!chunk) {
+            !Array.isArray(chunk)) {
             for (const k in chunk) {
                 if (inputChannels.includes(k)) {
                     yield [k, chunk[k]];

package/dist/pregel/io.js

@@ -48,11 +48,10 @@ export function* mapInput(inputChannels,
 chunk
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 ) {
-    if (chunk) {
+    if (chunk !== undefined && chunk !== null) {
         if (Array.isArray(inputChannels) &&
             typeof chunk === "object" &&
-            !Array.isArray(chunk) &&
-            !!chunk) {
+            !Array.isArray(chunk)) {
             for (const k in chunk) {
                 if (inputChannels.includes(k)) {
                     yield [k, chunk[k]];

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.15",
+  "version": "0.0.17",
   "description": "LangGraph",
   "type": "module",
   "engines": {