@langchain/langgraph 0.2.31 → 0.2.32

package/dist/graph/graph.d.ts

@@ -65,7 +65,7 @@ export declare class Graph<N extends string = typeof END, RunInput = any, RunOut
     }): CompiledGraph<N>;
     validate(interrupt?: string[]): void;
 }
-export declare class CompiledGraph<N extends string, State = any, Update = any, ConfigurableFieldType extends Record<string, any> = Record<string, any>> extends Pregel<Record<N | typeof START, PregelNode<State, Update>>, Record<N | typeof START | typeof END | string, BaseChannel>, ConfigurableFieldType & Record<string, any>, Update, State> {
+export declare class CompiledGraph<N extends string, State = any, Update = any, ConfigurableFieldType extends Record<string, any> = Record<string, any>, InputType = any, OutputType = any> extends Pregel<Record<N | typeof START, PregelNode<State, Update>>, Record<N | typeof START | typeof END | string, BaseChannel>, ConfigurableFieldType & Record<string, any>, InputType, OutputType> {
     NodeType: N;
     RunInput: State;
     RunOutput: Update;

package/dist/graph/state.d.ts

@@ -128,7 +128,7 @@ export declare class StateGraph<SD extends StateDefinition | unknown, S = SD ext
  * Should not be instantiated directly, only using the StateGraph `.compile()`
  * instance method.
  */
-export declare class CompiledStateGraph<S, U, N extends string = typeof START, I extends StateDefinition = StateDefinition, O extends StateDefinition = StateDefinition, C extends StateDefinition = StateDefinition> extends CompiledGraph<N, S, U, StateType<C>> {
+export declare class CompiledStateGraph<S, U, N extends string = typeof START, I extends StateDefinition = StateDefinition, O extends StateDefinition = StateDefinition, C extends StateDefinition = StateDefinition> extends CompiledGraph<N, S, U, StateType<C>, UpdateType<I>, StateType<O>> {
     builder: StateGraph<unknown, S, U, N, I, O, C>;
     attachNode(key: typeof START, node?: never): void;
     attachNode(key: N, node: StateGraphNodeSpec<S, U>): void;

package/dist/pregel/index.cjs

@@ -345,6 +345,19 @@ class Pregel extends runnables_1.Runnable {
                 });
             }
         }
+        // apply pending writes
+        const nullWrites = (saved.pendingWrites ?? [])
+            .filter((w) => w[0] === constants_js_1.NULL_TASK_ID)
+            .map((w) => w.slice(1));
+        if (nullWrites.length > 0) {
+            (0, algo_js_1._applyWrites)(saved.checkpoint, channels, [
+                {
+                    name: constants_js_1.INPUT,
+                    writes: nullWrites,
+                    triggers: [],
+                },
+            ]);
+        }
         // assemble the state snapshot
         return {
             values: (0, io_js_1.readChannels)(channels, this.streamChannelsAsIs),
@@ -505,7 +518,7 @@ class Pregel extends runnables_1.Runnable {
                 // apply null writes
                 const nullWrites = (saved.pendingWrites || [])
                     .filter((w) => w[0] === constants_js_1.NULL_TASK_ID)
-                    .flatMap((w) => w.slice(1));
+                    .map((w) => w.slice(1));
                 if (nullWrites.length > 0) {
                     (0, algo_js_1._applyWrites)(saved.checkpoint, channels, [
                         {
@@ -547,6 +560,45 @@ class Pregel extends runnables_1.Runnable {
             }, {});
             return (0, index_js_1.patchCheckpointMap)(nextConfig, saved ? saved.metadata : undefined);
         }
+        // apply pending writes, if not on specific checkpoint
+        if (config.configurable?.checkpoint_id === undefined &&
+            saved?.pendingWrites !== undefined &&
+            saved.pendingWrites.length > 0) {
+            // tasks for this checkpoint
+            const nextTasks = (0, algo_js_1._prepareNextTasks)(checkpoint, saved.pendingWrites, this.nodes, channels, managed, saved.config, true, {
+                store: this.store,
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                checkpointer: this.checkpointer,
+                step: (saved.metadata?.step ?? -1) + 1,
+            });
+            // apply null writes
+            const nullWrites = (saved.pendingWrites ?? [])
+                .filter((w) => w[0] === constants_js_1.NULL_TASK_ID)
+                .map((w) => w.slice(1));
+            if (nullWrites.length > 0) {
+                (0, algo_js_1._applyWrites)(saved.checkpoint, channels, [
+                    {
+                        name: constants_js_1.INPUT,
+                        writes: nullWrites,
+                        triggers: [],
+                    },
+                ]);
+            }
+            // apply writes
+            for (const [tid, k, v] of saved.pendingWrites) {
+                if ([constants_js_1.ERROR, constants_js_1.INTERRUPT, langgraph_checkpoint_1.SCHEDULED].includes(k) ||
+                    nextTasks[tid] === undefined) {
+                    continue;
+                }
+                nextTasks[tid].writes.push([k, v]);
+            }
+            const tasks = Object.values(nextTasks).filter((task) => {
+                return task.writes.length > 0;
+            });
+            if (tasks.length > 0) {
+                (0, algo_js_1._applyWrites)(checkpoint, channels, tasks);
+            }
+        }
         const nonNullVersion = Object.values(checkpoint.versions_seen)
             .map((seenVersions) => {
             return Object.values(seenVersions);

package/dist/pregel/index.js

@@ -341,6 +341,19 @@ export class Pregel extends Runnable {
                 });
             }
         }
+        // apply pending writes
+        const nullWrites = (saved.pendingWrites ?? [])
+            .filter((w) => w[0] === NULL_TASK_ID)
+            .map((w) => w.slice(1));
+        if (nullWrites.length > 0) {
+            _applyWrites(saved.checkpoint, channels, [
+                {
+                    name: INPUT,
+                    writes: nullWrites,
+                    triggers: [],
+                },
+            ]);
+        }
         // assemble the state snapshot
         return {
             values: readChannels(channels, this.streamChannelsAsIs),
@@ -501,7 +514,7 @@ export class Pregel extends Runnable {
                 // apply null writes
                 const nullWrites = (saved.pendingWrites || [])
                     .filter((w) => w[0] === NULL_TASK_ID)
-                    .flatMap((w) => w.slice(1));
+                    .map((w) => w.slice(1));
                 if (nullWrites.length > 0) {
                     _applyWrites(saved.checkpoint, channels, [
                         {
@@ -543,6 +556,45 @@ export class Pregel extends Runnable {
             }, {});
             return patchCheckpointMap(nextConfig, saved ? saved.metadata : undefined);
         }
+        // apply pending writes, if not on specific checkpoint
+        if (config.configurable?.checkpoint_id === undefined &&
+            saved?.pendingWrites !== undefined &&
+            saved.pendingWrites.length > 0) {
+            // tasks for this checkpoint
+            const nextTasks = _prepareNextTasks(checkpoint, saved.pendingWrites, this.nodes, channels, managed, saved.config, true, {
+                store: this.store,
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                checkpointer: this.checkpointer,
+                step: (saved.metadata?.step ?? -1) + 1,
+            });
+            // apply null writes
+            const nullWrites = (saved.pendingWrites ?? [])
+                .filter((w) => w[0] === NULL_TASK_ID)
+                .map((w) => w.slice(1));
+            if (nullWrites.length > 0) {
+                _applyWrites(saved.checkpoint, channels, [
+                    {
+                        name: INPUT,
+                        writes: nullWrites,
+                        triggers: [],
+                    },
+                ]);
+            }
+            // apply writes
+            for (const [tid, k, v] of saved.pendingWrites) {
+                if ([ERROR, INTERRUPT, SCHEDULED].includes(k) ||
+                    nextTasks[tid] === undefined) {
+                    continue;
+                }
+                nextTasks[tid].writes.push([k, v]);
+            }
+            const tasks = Object.values(nextTasks).filter((task) => {
+                return task.writes.length > 0;
+            });
+            if (tasks.length > 0) {
+                _applyWrites(checkpoint, channels, tasks);
+            }
+        }
         const nonNullVersion = Object.values(checkpoint.versions_seen)
             .map((seenVersions) => {
             return Object.values(seenVersions);

package/dist/pregel/io.cjs

@@ -72,17 +72,16 @@ function* mapCommand(cmd, pendingWrites) {
                 throw new Error(`In Command.send, expected Send or string, got ${typeof send}`);
             }
         }
-        // TODO: handle goto str for state graph
     }
     if (cmd.resume) {
         if (typeof cmd.resume === "object" &&
-            !!cmd.resume &&
             Object.keys(cmd.resume).length &&
             Object.keys(cmd.resume).every(uuid_1.validate)) {
             for (const [tid, resume] of Object.entries(cmd.resume)) {
-                // Find existing resume values for this task ID
-                const existing = (pendingWrites.find(([id, type]) => id === tid && type === constants_js_1.RESUME)?.[2] ?? []);
-                // Ensure we have an array and append the resume value
+                const existing = pendingWrites
+                    .filter((w) => w[0] === tid && w[1] === constants_js_1.RESUME)
+                    .map((w) => w[2])
+                    .slice(0, 1) ?? [];
                 existing.push(resume);
                 yield [tid, constants_js_1.RESUME, existing];
             }

package/dist/pregel/io.d.ts

@@ -7,7 +7,7 @@ export declare function readChannels<C extends PropertyKey>(channels: Record<C,
 /**
  * Map input chunk to a sequence of pending writes in the form (channel, value).
  */
-export declare function mapCommand(cmd: Command, pendingWrites: CheckpointPendingWrite<string>[]): Generator<[string, string, unknown]>;
+export declare function mapCommand(cmd: Command, pendingWrites: CheckpointPendingWrite[]): Generator<[string, string, unknown]>;
 /**
  * Map input chunk to a sequence of pending writes in the form [channel, value].
  */

package/dist/pregel/io.js

@@ -67,17 +67,16 @@ export function* mapCommand(cmd, pendingWrites) {
                 throw new Error(`In Command.send, expected Send or string, got ${typeof send}`);
             }
         }
-        // TODO: handle goto str for state graph
     }
     if (cmd.resume) {
         if (typeof cmd.resume === "object" &&
-            !!cmd.resume &&
             Object.keys(cmd.resume).length &&
             Object.keys(cmd.resume).every(validate)) {
             for (const [tid, resume] of Object.entries(cmd.resume)) {
-                // Find existing resume values for this task ID
-                const existing = (pendingWrites.find(([id, type]) => id === tid && type === RESUME)?.[2] ?? []);
-                // Ensure we have an array and append the resume value
+                const existing = pendingWrites
+                    .filter((w) => w[0] === tid && w[1] === RESUME)
+                    .map((w) => w[2])
+                    .slice(0, 1) ?? [];
                 existing.push(resume);
                 yield [tid, RESUME, existing];
             }

package/dist/pregel/loop.cjs

@@ -409,14 +409,9 @@ class PregelLoop {
         }
         // save writes
         for (const [c, v] of writesCopy) {
-            if (c in langgraph_checkpoint_1.WRITES_IDX_MAP) {
-                const idx = this.checkpointPendingWrites.findIndex((w) => w[0] === taskId && w[1] === c);
-                if (idx !== -1) {
-                    this.checkpointPendingWrites[idx] = [taskId, c, v];
-                }
-                else {
-                    this.checkpointPendingWrites.push([taskId, c, v]);
-                }
+            const idx = this.checkpointPendingWrites.findIndex((w) => w[0] === taskId && w[1] === c);
+            if (c in langgraph_checkpoint_1.WRITES_IDX_MAP && idx !== -1) {
+                this.checkpointPendingWrites[idx] = [taskId, c, v];
             }
             else {
                 this.checkpointPendingWrites.push([taskId, c, v]);
@@ -586,7 +581,7 @@ class PregelLoop {
     /**
      * Resuming from previous checkpoint requires
      * - finding a previous checkpoint
-     * - receiving None input (outer graph) or RESUMING flag (subgraph)
+     * - receiving null input (outer graph) or RESUMING flag (subgraph)
      */
     async _first(inputKeys) {
         const isResuming = Object.keys(this.checkpoint.channel_versions).length !== 0 &&
@@ -610,6 +605,19 @@ class PregelLoop {
                 this.putWrites(tid, ws);
             }
         }
+        // apply null writes
+        const nullWrites = (this.checkpointPendingWrites ?? [])
+            .filter((w) => w[0] === constants_js_1.NULL_TASK_ID)
+            .map((w) => w.slice(1));
+        if (nullWrites.length > 0) {
+            (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, [
+                {
+                    name: constants_js_1.INPUT,
+                    writes: nullWrites,
+                    triggers: [],
+                },
+            ], this.checkpointerGetNextVersion);
+        }
         if (isResuming) {
             for (const channelName of Object.keys(this.channels)) {
                 if (this.checkpoint.channel_versions[channelName] !== undefined) {

package/dist/pregel/loop.d.ts

@@ -116,7 +116,7 @@ export declare class PregelLoop {
     /**
      * Resuming from previous checkpoint requires
      * - finding a previous checkpoint
-     * - receiving None input (outer graph) or RESUMING flag (subgraph)
+     * - receiving null input (outer graph) or RESUMING flag (subgraph)
      */
     protected _first(inputKeys: string | string[]): Promise<void>;
     protected _emit(values: [StreamMode, unknown][]): void;

package/dist/pregel/loop.js

@@ -1,7 +1,7 @@
 import { IterableReadableStream } from "@langchain/core/utils/stream";
 import { copyCheckpoint, emptyCheckpoint, AsyncBatchedStore, WRITES_IDX_MAP, } from "@langchain/langgraph-checkpoint";
 import { createCheckpoint, emptyChannels, } from "../channels/base.js";
-import { _isCommand, CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_READ, CONFIG_KEY_RESUMING, CONFIG_KEY_STREAM, ERROR, INPUT, INTERRUPT, RESUME, TAG_HIDDEN, } from "../constants.js";
+import { _isCommand, CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_READ, CONFIG_KEY_RESUMING, CONFIG_KEY_STREAM, ERROR, INPUT, INTERRUPT, NULL_TASK_ID, RESUME, TAG_HIDDEN, } from "../constants.js";
 import { _applyWrites, _prepareNextTasks, increment, shouldInterrupt, } from "./algo.js";
 import { gatherIterator, gatherIteratorSync, prefixGenerator, } from "../utils.js";
 import { mapCommand, mapInput, mapOutputUpdates, mapOutputValues, readChannels, } from "./io.js";
@@ -405,14 +405,9 @@ export class PregelLoop {
         }
         // save writes
         for (const [c, v] of writesCopy) {
-            if (c in WRITES_IDX_MAP) {
-                const idx = this.checkpointPendingWrites.findIndex((w) => w[0] === taskId && w[1] === c);
-                if (idx !== -1) {
-                    this.checkpointPendingWrites[idx] = [taskId, c, v];
-                }
-                else {
-                    this.checkpointPendingWrites.push([taskId, c, v]);
-                }
+            const idx = this.checkpointPendingWrites.findIndex((w) => w[0] === taskId && w[1] === c);
+            if (c in WRITES_IDX_MAP && idx !== -1) {
+                this.checkpointPendingWrites[idx] = [taskId, c, v];
             }
             else {
                 this.checkpointPendingWrites.push([taskId, c, v]);
@@ -582,7 +577,7 @@ export class PregelLoop {
     /**
      * Resuming from previous checkpoint requires
      * - finding a previous checkpoint
-     * - receiving None input (outer graph) or RESUMING flag (subgraph)
+     * - receiving null input (outer graph) or RESUMING flag (subgraph)
      */
     async _first(inputKeys) {
         const isResuming = Object.keys(this.checkpoint.channel_versions).length !== 0 &&
@@ -606,6 +601,19 @@ export class PregelLoop {
                 this.putWrites(tid, ws);
             }
         }
+        // apply null writes
+        const nullWrites = (this.checkpointPendingWrites ?? [])
+            .filter((w) => w[0] === NULL_TASK_ID)
+            .map((w) => w.slice(1));
+        if (nullWrites.length > 0) {
+            _applyWrites(this.checkpoint, this.channels, [
+                {
+                    name: INPUT,
+                    writes: nullWrites,
+                    triggers: [],
+                },
+            ], this.checkpointerGetNextVersion);
+        }
         if (isResuming) {
             for (const channelName of Object.keys(this.channels)) {
                 if (this.checkpoint.channel_versions[channelName] !== undefined) {

package/package.json

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