@langchain/langgraph 0.1.0-rc.0 → 0.1.0

package/dist/pregel/index.d.ts

@@ -1,6 +1,7 @@
 import { Runnable, RunnableConfig, RunnableFunc } from "@langchain/core/runnables";
+import { IterableReadableStream } from "@langchain/core/utils/stream";
+import { BaseCheckpointSaver, CheckpointListOptions } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
-import { BaseCheckpointSaver, CheckpointListOptions } from "../checkpoint/base.js";
 import { PregelNode } from "./read.js";
 import { ChannelWrite } from "./write.js";
 import { All, PregelInterface, PregelParams, StateSnapshot, StreamMode } from "./types.js";
@@ -53,8 +54,19 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     validate(): this;
     get streamChannelsList(): Array<keyof Cc>;
     get streamChannelsAsIs(): keyof Cc | Array<keyof Cc>;
+    /**
+     * Get the current state of the graph.
+     */
     getState(config: RunnableConfig): Promise<StateSnapshot>;
+    /**
+     * Get the history of the state of the graph.
+     */
     getStateHistory(config: RunnableConfig, options?: CheckpointListOptions): AsyncIterableIterator<StateSnapshot>;
+    /**
+     * Update the state of the graph with the given values, as if they came from
+     * node `as_node`. If `as_node` is not provided, it will be set to the last node
+     * that updated the state, if not ambiguous.
+     */
     updateState(config: RunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn): Promise<RunnableConfig>;
     _defaults(config: PregelOptions<Nn, Cc>): [
         boolean,
@@ -71,11 +83,37 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
         // interrupt after
         BaseCheckpointSaver | undefined
     ];
+    /**
+     * Stream graph steps for a single input.
+     * @param input The input to the graph.
+     * @param options The configuration to use for the run.
+     * @param options.streamMode The mode to stream output. Defaults to value set on initialization.
+     *   Options are "values", "updates", and "debug". Default is "values".
+     *     values: Emit the current values of the state for each step.
+     *     updates: Emit only the updates to the state for each step.
+     *         Output is a dict with the node name as key and the updated values as value.
+     *     debug: Emit debug events for each step.
+     * @param options.outputKeys The keys to stream. Defaults to all non-context channels.
+     * @param options.interruptBefore Nodes to interrupt before.
+     * @param options.interruptAfter Nodes to interrupt after.
+     * @param options.debug Whether to print debug information during execution.
+     */
+    stream(input: PregelInputType, options?: Partial<PregelOptions<Nn, Cc>>): Promise<IterableReadableStream<PregelOutputType>>;
     _streamIterator(input: PregelInputType, options?: Partial<PregelOptions<Nn, Cc>>): AsyncGenerator<PregelOutputType>;
     /**
      * Run the graph with a single input and config.
-     * @param input
-     * @param options
+     * @param input The input to the graph.
+     * @param options The configuration to use for the run.
+     * @param options.streamMode The mode to stream output. Defaults to value set on initialization.
+     *   Options are "values", "updates", and "debug". Default is "values".
+     *     values: Emit the current values of the state for each step.
+     *     updates: Emit only the updates to the state for each step.
+     *         Output is a dict with the node name as key and the updated values as value.
+     *     debug: Emit debug events for each step.
+     * @param options.outputKeys The keys to stream. Defaults to all non-context channels.
+     * @param options.interruptBefore Nodes to interrupt before.
+     * @param options.interruptAfter Nodes to interrupt after.
+     * @param options.debug Whether to print debug information during execution.
      */
     invoke(input: PregelInputType, options?: Partial<PregelOptions<Nn, Cc>>): Promise<PregelOutputType>;
 }

package/dist/pregel/index.js

@@ -1,16 +1,15 @@
 /* eslint-disable no-param-reassign */
 import { Runnable, RunnableSequence, _coerceToRunnable, ensureConfig, getCallbackManagerForConfig, patchConfig, } from "@langchain/core/runnables";
+import { copyCheckpoint, emptyCheckpoint, uuid5, } from "@langchain/langgraph-checkpoint";
 import { createCheckpoint, emptyChannels, } from "../channels/base.js";
-import { copyCheckpoint, emptyCheckpoint, } from "../checkpoint/base.js";
 import { PregelNode } from "./read.js";
 import { validateGraph, validateKeys } from "./validate.js";
 import { mapOutputUpdates, readChannels } from "./io.js";
-import { mapDebugTaskResults, printStepCheckpoint, printStepTasks, printStepWrites, } from "./debug.js";
+import { mapDebugTaskResults, printStepCheckpoint, printStepTasks, printStepWrites, tasksWithWrites, } from "./debug.js";
 import { ChannelWrite, PASSTHROUGH } from "./write.js";
-import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, INTERRUPT, } from "../constants.js";
+import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, ERROR, INTERRUPT, } from "../constants.js";
 import { GraphRecursionError, GraphValueError, InvalidUpdateError, } from "../errors.js";
 import { executeTasks, _prepareNextTasks, _localRead, _applyWrites, } from "./algo.js";
-import { uuid5 } from "../checkpoint/id.js";
 import { prefixGenerator } from "../utils.js";
 import { _coerceToDict, getNewChannelVersions } from "./utils.js";
 import { PregelLoop } from "./loop.js";
@@ -207,6 +206,9 @@ export class Pregel extends Runnable {
             return Object.keys(this.channels);
         }
     }
+    /**
+     * Get the current state of the graph.
+     */
     async getState(config) {
         if (!this.checkpointer) {
             throw new GraphValueError("No checkpointer set");
@@ -214,26 +216,31 @@ export class Pregel extends Runnable {
         const saved = await this.checkpointer.getTuple(config);
         const checkpoint = saved ? saved.checkpoint : emptyCheckpoint();
         const channels = emptyChannels(this.channels, checkpoint);
-        const [, nextTasks] = _prepareNextTasks(checkpoint, this.nodes, channels, saved !== undefined ? saved.config : config, false, { step: -1 });
+        const nextTasks = _prepareNextTasks(checkpoint, this.nodes, channels, saved !== undefined ? saved.config : config, false, { step: saved ? (saved.metadata?.step ?? -1) + 1 : -1 });
         return {
             values: readChannels(channels, this.streamChannelsAsIs),
             next: nextTasks.map((task) => task.name),
+            tasks: tasksWithWrites(nextTasks, saved?.pendingWrites ?? []),
             metadata: saved?.metadata,
             config: saved ? saved.config : config,
             createdAt: saved?.checkpoint.ts,
             parentConfig: saved?.parentConfig,
         };
     }
+    /**
+     * Get the history of the state of the graph.
+     */
     async *getStateHistory(config, options) {
         if (!this.checkpointer) {
             throw new GraphValueError("No checkpointer set");
         }
         for await (const saved of this.checkpointer.list(config, options)) {
             const channels = emptyChannels(this.channels, saved.checkpoint);
-            const [, nextTasks] = _prepareNextTasks(saved.checkpoint, this.nodes, channels, saved.config, false, { step: -1 });
+            const nextTasks = _prepareNextTasks(saved.checkpoint, this.nodes, channels, saved.config, false, { step: -1 });
             yield {
                 values: readChannels(channels, this.streamChannelsAsIs),
                 next: nextTasks.map((task) => task.name),
+                tasks: tasksWithWrites(nextTasks, saved.pendingWrites ?? []),
                 metadata: saved.metadata,
                 config: saved.config,
                 createdAt: saved.checkpoint.ts,
@@ -241,6 +248,11 @@ export class Pregel extends Runnable {
             };
         }
     }
+    /**
+     * Update the state of the graph with the given values, as if they came from
+     * node `as_node`. If `as_node` is not provided, it will be set to the last node
+     * that updated the state, if not ambiguous.
+     */
     async updateState(config, values, asNode) {
         if (!this.checkpointer) {
             throw new GraphValueError("No checkpointer set");
@@ -263,8 +275,8 @@ export class Pregel extends Runnable {
             },
         };
         // Find last node that updated the state, if not provided
-        if (values === undefined && asNode === undefined) {
-            return await this.checkpointer.put(checkpointConfig, createCheckpoint(checkpoint, {}, step), {
+        if (values == null && asNode === undefined) {
+            return await this.checkpointer.put(checkpointConfig, createCheckpoint(checkpoint, undefined, step), {
                 source: "update",
                 step,
                 writes: {},
@@ -276,7 +288,7 @@ export class Pregel extends Runnable {
         })
             .flat()
             .find((v) => !!v);
-        if (asNode === undefined && !nonNullVersion) {
+        if (asNode === undefined && nonNullVersion === undefined) {
             if (typeof this.inputChannels === "string" &&
                 this.nodes[this.inputChannels] !== undefined) {
                 asNode = this.inputChannels;
@@ -401,6 +413,24 @@ export class Pregel extends Runnable {
             defaultCheckpointer,
         ];
     }
+    /**
+     * Stream graph steps for a single input.
+     * @param input The input to the graph.
+     * @param options The configuration to use for the run.
+     * @param options.streamMode The mode to stream output. Defaults to value set on initialization.
+     *   Options are "values", "updates", and "debug". Default is "values".
+     *     values: Emit the current values of the state for each step.
+     *     updates: Emit only the updates to the state for each step.
+     *         Output is a dict with the node name as key and the updated values as value.
+     *     debug: Emit debug events for each step.
+     * @param options.outputKeys The keys to stream. Defaults to all non-context channels.
+     * @param options.interruptBefore Nodes to interrupt before.
+     * @param options.interruptAfter Nodes to interrupt after.
+     * @param options.debug Whether to print debug information during execution.
+     */
+    async stream(input, options) {
+        return super.stream(input, options);
+    }
     async *_streamIterator(input, options) {
         const inputConfig = ensureConfig(options);
         if (inputConfig.recursionLimit === undefined ||
@@ -459,21 +489,47 @@ export class Pregel extends Runnable {
                 // execute tasks, and wait for one to fail or all to finish.
                 // each task is independent from all other concurrent tasks
                 // yield updates/debug output as each task finishes
-                const tasks = loop.tasks.map((pregelTask) => () => {
-                    return pregelTask.proc.invoke(pregelTask.input, pregelTask.config);
-                });
-                await executeTasks(tasks, this.stepTimeout, config.signal);
-                for (const task of loop.tasks) {
-                    loop.putWrites(task.id, task.writes);
-                }
-                if (streamMode.includes("updates")) {
-                    // TODO: Refactor
-                    for await (const task of loop.tasks) {
-                        yield* prefixGenerator(mapOutputUpdates(outputKeys, [task]), streamMode.length > 1 ? "updates" : undefined);
+                const tasks = Object.fromEntries(loop.tasks
+                    .filter((task) => task.writes.length === 0)
+                    .map((pregelTask) => {
+                    return [
+                        pregelTask.id,
+                        async () => {
+                            let error;
+                            let result;
+                            try {
+                                result = await pregelTask.proc.invoke(pregelTask.input, pregelTask.config);
+                                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                            }
+                            catch (e) {
+                                error = e;
+                                error.pregelTaskId = pregelTask.id;
+                            }
+                            return {
+                                task: pregelTask,
+                                result,
+                                error,
+                            };
+                        },
+                    ];
+                }));
+                try {
+                    for await (const task of executeTasks(tasks, this.stepTimeout, config.signal)) {
+                        loop.putWrites(task.id, task.writes);
+                        if (streamMode.includes("updates")) {
+                            yield* prefixGenerator(mapOutputUpdates(outputKeys, [task]), streamMode.length > 1 ? "updates" : undefined);
+                        }
+                        if (streamMode.includes("debug")) {
+                            yield* prefixGenerator(mapDebugTaskResults(loop.step, [task], this.streamChannelsList), streamMode.length > 1 ? "debug" : undefined);
+                        }
                     }
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
                 }
-                if (streamMode.includes("debug")) {
-                    yield* prefixGenerator(mapDebugTaskResults(loop.step, loop.tasks, this.streamChannelsList), streamMode.length > 1 ? "debug" : undefined);
+                catch (e) {
+                    if (e.pregelTaskId) {
+                        loop.putWrites(e.pregelTaskId, [[ERROR, { message: e.message }]]);
+                    }
+                    throw e;
                 }
                 if (debug) {
                     printStepWrites(loop.step, loop.tasks.map((task) => task.writes).flat(), this.streamChannelsList);
@@ -515,8 +571,18 @@ export class Pregel extends Runnable {
     }
     /**
      * Run the graph with a single input and config.
-     * @param input
-     * @param options
+     * @param input The input to the graph.
+     * @param options The configuration to use for the run.
+     * @param options.streamMode The mode to stream output. Defaults to value set on initialization.
+     *   Options are "values", "updates", and "debug". Default is "values".
+     *     values: Emit the current values of the state for each step.
+     *     updates: Emit only the updates to the state for each step.
+     *         Output is a dict with the node name as key and the updated values as value.
+     *     debug: Emit debug events for each step.
+     * @param options.outputKeys The keys to stream. Defaults to all non-context channels.
+     * @param options.interruptBefore Nodes to interrupt before.
+     * @param options.interruptAfter Nodes to interrupt after.
+     * @param options.debug Whether to print debug information during execution.
      */
     async invoke(input, options) {
         const streamMode = options?.streamMode ?? "values";

package/dist/pregel/io.d.ts

@@ -1,6 +1,6 @@
+import type { PendingWrite } from "@langchain/langgraph-checkpoint";
 import type { BaseChannel } from "../channels/base.js";
 import type { PregelExecutableTask } from "./types.js";
-import type { PendingWrite } from "../checkpoint/types.js";
 export declare function readChannel<C extends PropertyKey>(channels: Record<C, BaseChannel>, chan: C, catchErrors?: boolean, returnException?: boolean): unknown | null;
 export declare function readChannels<C extends PropertyKey>(channels: Record<C, BaseChannel>, select: C | Array<C>, skipEmpty?: boolean): Record<string, any> | any;
 /**

package/dist/pregel/loop.cjs

@@ -5,8 +5,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PregelLoop = void 0;
 const double_ended_queue_1 = __importDefault(require("double-ended-queue"));
-const base_js_1 = require("../checkpoint/base.cjs");
-const base_js_2 = require("../channels/base.cjs");
+const langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
+const base_js_1 = require("../channels/base.cjs");
 const constants_js_1 = require("../constants.cjs");
 const algo_js_1 = require("./algo.cjs");
 const utils_js_1 = require("../utils.cjs");
@@ -167,7 +167,7 @@ class PregelLoop {
     static async initialize(params) {
         const saved = (await params.checkpointer?.getTuple(params.config)) ?? {
             config: params.config,
-            checkpoint: (0, base_js_1.emptyCheckpoint)(),
+            checkpoint: (0, langgraph_checkpoint_1.emptyCheckpoint)(),
             metadata: {
                 source: "input",
                 step: -2,
@@ -183,10 +183,10 @@ class PregelLoop {
                 ...saved.config.configurable,
             },
         };
-        const checkpoint = (0, base_js_1.copyCheckpoint)(saved.checkpoint);
+        const checkpoint = (0, langgraph_checkpoint_1.copyCheckpoint)(saved.checkpoint);
         const checkpointMetadata = { ...saved.metadata };
         const checkpointPendingWrites = saved.pendingWrites ?? [];
-        const channels = (0, base_js_2.emptyChannels)(params.graph.channels, checkpoint);
+        const channels = (0, base_js_1.emptyChannels)(params.graph.channels, checkpoint);
         const step = (checkpointMetadata.step ?? 0) + 1;
         const stop = step + (params.config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;
         const checkpointPreviousVersions = { ...checkpoint.channel_versions };
@@ -289,7 +289,7 @@ class PregelLoop {
             this.status = "out_of_steps";
             return false;
         }
-        const [, nextTasks] = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.graph.nodes, this.channels, this.config, true, {
+        const nextTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.graph.nodes, this.channels, this.config, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             isResuming: this.input === INPUT_RESUMING,
@@ -308,6 +308,10 @@ class PregelLoop {
         // if there are pending writes from a previous loop, apply them
         if (this.checkpointPendingWrites.length > 0) {
             for (const [tid, k, v] of this.checkpointPendingWrites) {
+                // TODO: Do the same for INTERRUPT
+                if (k === constants_js_1.ERROR) {
+                    continue;
+                }
                 const task = this.tasks.find((t) => t.id === tid);
                 if (task) {
                     task.writes.push([k, v]);
@@ -364,7 +368,7 @@ class PregelLoop {
             if (inputWrites.length === 0) {
                 throw new errors_js_1.EmptyInputError(`Received no input writes for ${JSON.stringify(this.graph.inputChannels, null, 2)}`);
             }
-            const [, discardTasks] = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.graph.nodes, this.channels, this.config, true, { step: this.step });
+            const discardTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.graph.nodes, this.channels, this.config, true, { step: this.step });
             (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, discardTasks.concat([
                 {
                     name: constants_js_1.INPUT,
@@ -392,7 +396,7 @@ 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.checkpoint = (0, base_js_2.createCheckpoint)(this.checkpoint, this.channels, this.step
+            this.checkpoint = (0, base_js_1.createCheckpoint)(this.checkpoint, this.channels, this.step
             // id: this.isNested ? this.config.configurable?.checkpoint_id : undefined,
             );
             this.checkpointConfig = {
@@ -410,7 +414,7 @@ class PregelLoop {
             // ensuring checkpointers receive checkpoints in order
             void this._checkpointerPutAfterPrevious({
                 config: { ...this.checkpointConfig },
-                checkpoint: (0, base_js_1.copyCheckpoint)(this.checkpoint),
+                checkpoint: (0, langgraph_checkpoint_1.copyCheckpoint)(this.checkpoint),
                 metadata: { ...this.checkpointMetadata },
                 newVersions,
             });

package/dist/pregel/loop.d.ts

@@ -1,10 +1,9 @@
 import Deque from "double-ended-queue";
 import type { RunnableConfig } from "@langchain/core/runnables";
 import type { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { BaseCheckpointSaver, Checkpoint } from "../checkpoint/base.js";
+import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelExecutableTask, PregelInterface, StreamMode } from "./types.js";
-import { PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All } from "../checkpoint/types.js";
 export type PregelLoopInitializeParams = {
     input?: any;
     config: RunnableConfig;

package/dist/pregel/loop.js

@@ -1,7 +1,7 @@
 import Deque from "double-ended-queue";
-import { copyCheckpoint, emptyCheckpoint, } from "../checkpoint/base.js";
+import { copyCheckpoint, emptyCheckpoint, } from "@langchain/langgraph-checkpoint";
 import { createCheckpoint, emptyChannels, } from "../channels/base.js";
-import { CONFIG_KEY_READ, CONFIG_KEY_RESUMING, INPUT, INTERRUPT, } from "../constants.js";
+import { CONFIG_KEY_READ, CONFIG_KEY_RESUMING, ERROR, INPUT, INTERRUPT, } from "../constants.js";
 import { _applyWrites, _prepareNextTasks, increment, shouldInterrupt, } from "./algo.js";
 import { gatherIterator, prefixGenerator } from "../utils.js";
 import { mapInput, mapOutputUpdates, mapOutputValues } from "./io.js";
@@ -283,7 +283,7 @@ export class PregelLoop {
             this.status = "out_of_steps";
             return false;
         }
-        const [, nextTasks] = _prepareNextTasks(this.checkpoint, this.graph.nodes, this.channels, this.config, true, {
+        const nextTasks = _prepareNextTasks(this.checkpoint, this.graph.nodes, this.channels, this.config, true, {
             step: this.step,
             checkpointer: this.checkpointer,
             isResuming: this.input === INPUT_RESUMING,
@@ -302,6 +302,10 @@ export class PregelLoop {
         // if there are pending writes from a previous loop, apply them
         if (this.checkpointPendingWrites.length > 0) {
             for (const [tid, k, v] of this.checkpointPendingWrites) {
+                // TODO: Do the same for INTERRUPT
+                if (k === ERROR) {
+                    continue;
+                }
                 const task = this.tasks.find((t) => t.id === tid);
                 if (task) {
                     task.writes.push([k, v]);
@@ -358,7 +362,7 @@ export class PregelLoop {
             if (inputWrites.length === 0) {
                 throw new EmptyInputError(`Received no input writes for ${JSON.stringify(this.graph.inputChannels, null, 2)}`);
             }
-            const [, discardTasks] = _prepareNextTasks(this.checkpoint, this.graph.nodes, this.channels, this.config, true, { step: this.step });
+            const discardTasks = _prepareNextTasks(this.checkpoint, this.graph.nodes, this.channels, this.config, true, { step: this.step });
             _applyWrites(this.checkpoint, this.channels, discardTasks.concat([
                 {
                     name: INPUT,

package/dist/pregel/types.d.ts

@@ -1,6 +1,5 @@
 import type { Runnable, RunnableConfig } from "@langchain/core/runnables";
-import type { PendingWrite, CheckpointMetadata } from "../checkpoint/types.js";
-import type { BaseCheckpointSaver } from "../checkpoint/base.js";
+import type { PendingWrite, CheckpointMetadata, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import type { BaseChannel } from "../channels/base.js";
 import type { PregelNode } from "./read.js";
 export type StreamMode = "values" | "updates" | "debug";
@@ -85,6 +84,10 @@ export interface StateSnapshot {
      * @default undefined
      */
     readonly parentConfig?: RunnableConfig | undefined;
+    /**
+     * Tasks to execute in this step. If already attempted, may contain an error.
+     */
+    readonly tasks: PregelTaskDescription[];
 }
 export type All = "*";
 export {};

package/dist/pregel/utils.cjs

@@ -1,18 +1,23 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._coerceToDict = exports.getNewChannelVersions = void 0;
+exports._getIdMetadata = exports._coerceToDict = exports.getNewChannelVersions = exports.getNullChannelVersion = void 0;
+function getNullChannelVersion(currentVersions) {
+    const versionValues = Object.values(currentVersions);
+    const versionType = versionValues.length > 0 ? typeof versionValues[0] : undefined;
+    let nullVersion;
+    if (versionType === "number") {
+        nullVersion = 0;
+    }
+    else if (versionType === "string") {
+        nullVersion = "";
+    }
+    return nullVersion;
+}
+exports.getNullChannelVersion = getNullChannelVersion;
 function getNewChannelVersions(previousVersions, currentVersions) {
     // Get new channel versions
     if (Object.keys(previousVersions).length > 0) {
-        const versionValues = Object.values(currentVersions);
-        const versionType = versionValues.length > 0 ? typeof versionValues[0] : undefined;
-        let nullVersion;
-        if (versionType === "number") {
-            nullVersion = 0;
-        }
-        else if (versionType === "string") {
-            nullVersion = "";
-        }
+        const nullVersion = getNullChannelVersion(currentVersions);
         return Object.fromEntries(Object.entries(currentVersions).filter(([k, v]) => v > (previousVersions[k] ?? nullVersion)));
     }
     else {
@@ -31,3 +36,13 @@ function _coerceToDict(value, defaultKey) {
         : { [defaultKey]: value };
 }
 exports._coerceToDict = _coerceToDict;
+// Order matters
+function _getIdMetadata(metadata) {
+    return {
+        langgraph_step: metadata.langgraph_step,
+        langgraph_node: metadata.langgraph_node,
+        langgraph_triggers: metadata.langgraph_triggers,
+        langgraph_task_idx: metadata.langgraph_task_idx,
+    };
+}
+exports._getIdMetadata = _getIdMetadata;

package/dist/pregel/utils.d.ts

@@ -1,3 +1,10 @@
-import type { ChannelVersions } from "../checkpoint/base.js";
+import type { ChannelVersions } from "@langchain/langgraph-checkpoint";
+export declare function getNullChannelVersion(currentVersions: ChannelVersions): string | number | undefined;
 export declare function getNewChannelVersions(previousVersions: ChannelVersions, currentVersions: ChannelVersions): ChannelVersions;
 export declare function _coerceToDict(value: any, defaultKey: string): any;
+export declare function _getIdMetadata(metadata: Record<string, unknown>): {
+    langgraph_step: unknown;
+    langgraph_node: unknown;
+    langgraph_triggers: unknown;
+    langgraph_task_idx: unknown;
+};

package/dist/pregel/utils.js

@@ -1,15 +1,19 @@
+export function getNullChannelVersion(currentVersions) {
+    const versionValues = Object.values(currentVersions);
+    const versionType = versionValues.length > 0 ? typeof versionValues[0] : undefined;
+    let nullVersion;
+    if (versionType === "number") {
+        nullVersion = 0;
+    }
+    else if (versionType === "string") {
+        nullVersion = "";
+    }
+    return nullVersion;
+}
 export function getNewChannelVersions(previousVersions, currentVersions) {
     // Get new channel versions
     if (Object.keys(previousVersions).length > 0) {
-        const versionValues = Object.values(currentVersions);
-        const versionType = versionValues.length > 0 ? typeof versionValues[0] : undefined;
-        let nullVersion;
-        if (versionType === "number") {
-            nullVersion = 0;
-        }
-        else if (versionType === "string") {
-            nullVersion = "";
-        }
+        const nullVersion = getNullChannelVersion(currentVersions);
         return Object.fromEntries(Object.entries(currentVersions).filter(([k, v]) => v > (previousVersions[k] ?? nullVersion)));
     }
     else {
@@ -26,3 +30,12 @@ export function _coerceToDict(value, defaultKey) {
         ? value
         : { [defaultKey]: value };
 }
+// Order matters
+export function _getIdMetadata(metadata) {
+    return {
+        langgraph_step: metadata.langgraph_step,
+        langgraph_node: metadata.langgraph_node,
+        langgraph_triggers: metadata.langgraph_triggers,
+        langgraph_task_idx: metadata.langgraph_task_idx,
+    };
+}

package/dist/web.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Send = exports.EmptyChannelError = exports.InvalidUpdateError = exports.GraphValueError = exports.GraphRecursionError = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
+exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.EmptyChannelError = exports.InvalidUpdateError = exports.GraphValueError = exports.GraphRecursionError = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
 var index_js_1 = require("./graph/index.cjs");
 Object.defineProperty(exports, "END", { enumerable: true, get: function () { return index_js_1.END; } });
 Object.defineProperty(exports, "Graph", { enumerable: true, get: function () { return index_js_1.Graph; } });
@@ -9,12 +9,6 @@ Object.defineProperty(exports, "StateGraph", { enumerable: true, get: function (
 Object.defineProperty(exports, "MessageGraph", { enumerable: true, get: function () { return index_js_1.MessageGraph; } });
 Object.defineProperty(exports, "messagesStateReducer", { enumerable: true, get: function () { return index_js_1.messagesStateReducer; } });
 Object.defineProperty(exports, "Annotation", { enumerable: true, get: function () { return index_js_1.Annotation; } });
-var memory_js_1 = require("./checkpoint/memory.cjs");
-Object.defineProperty(exports, "MemorySaver", { enumerable: true, get: function () { return memory_js_1.MemorySaver; } });
-var base_js_1 = require("./checkpoint/base.cjs");
-Object.defineProperty(exports, "copyCheckpoint", { enumerable: true, get: function () { return base_js_1.copyCheckpoint; } });
-Object.defineProperty(exports, "emptyCheckpoint", { enumerable: true, get: function () { return base_js_1.emptyCheckpoint; } });
-Object.defineProperty(exports, "BaseCheckpointSaver", { enumerable: true, get: function () { return base_js_1.BaseCheckpointSaver; } });
 var errors_js_1 = require("./errors.cjs");
 Object.defineProperty(exports, "GraphRecursionError", { enumerable: true, get: function () { return errors_js_1.GraphRecursionError; } });
 Object.defineProperty(exports, "GraphValueError", { enumerable: true, get: function () { return errors_js_1.GraphValueError; } });
@@ -22,3 +16,8 @@ Object.defineProperty(exports, "InvalidUpdateError", { enumerable: true, get: fu
 Object.defineProperty(exports, "EmptyChannelError", { enumerable: true, get: function () { return errors_js_1.EmptyChannelError; } });
 var constants_js_1 = require("./constants.cjs");
 Object.defineProperty(exports, "Send", { enumerable: true, get: function () { return constants_js_1.Send; } });
+var langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
+Object.defineProperty(exports, "MemorySaver", { enumerable: true, get: function () { return langgraph_checkpoint_1.MemorySaver; } });
+Object.defineProperty(exports, "copyCheckpoint", { enumerable: true, get: function () { return langgraph_checkpoint_1.copyCheckpoint; } });
+Object.defineProperty(exports, "emptyCheckpoint", { enumerable: true, get: function () { return langgraph_checkpoint_1.emptyCheckpoint; } });
+Object.defineProperty(exports, "BaseCheckpointSaver", { enumerable: true, get: function () { return langgraph_checkpoint_1.BaseCheckpointSaver; } });

package/dist/web.d.ts

@@ -1,7 +1,4 @@
 export { END, Graph, type StateGraphArgs, START, StateGraph, type CompiledStateGraph, MessageGraph, messagesStateReducer, Annotation, type StateType, type UpdateType, type CompiledGraph, } from "./graph/index.js";
-export { MemorySaver } from "./checkpoint/memory.js";
-export { type Checkpoint, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "./checkpoint/base.js";
-export { type CheckpointMetadata } from "./checkpoint/types.js";
 export { GraphRecursionError, GraphValueError, InvalidUpdateError, EmptyChannelError, } from "./errors.js";
-export { type SerializerProtocol } from "./serde/base.js";
 export { Send } from "./constants.js";
+export { MemorySaver, type Checkpoint, type CheckpointMetadata, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "@langchain/langgraph-checkpoint";

package/dist/web.js

@@ -1,5 +1,4 @@
 export { END, Graph, START, StateGraph, MessageGraph, messagesStateReducer, Annotation, } from "./graph/index.js";
-export { MemorySaver } from "./checkpoint/memory.js";
-export { copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "./checkpoint/base.js";
 export { GraphRecursionError, GraphValueError, InvalidUpdateError, EmptyChannelError, } from "./errors.js";
 export { Send } from "./constants.js";
+export { MemorySaver, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "@langchain/langgraph-checkpoint";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.1.0-rc.0",
+  "version": "0.1.0",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -31,26 +31,34 @@
   "license": "MIT",
   "dependencies": {
     "@langchain/core": ">=0.2.20 <0.3.0",
+    "@langchain/langgraph-checkpoint": "0.0.0",
+    "@langchain/langgraph-checkpoint-sqlite": "0.0.0",
     "double-ended-queue": "^2.1.0-0",
     "uuid": "^10.0.0",
     "zod": "^3.23.8"
   },
+  "peerDependencies": {
+    "better-sqlite3": "^9.5.0"
+  },
+  "peerDependenciesMeta": {
+    "better-sqlite3": {
+      "optional": true
+    }
+  },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
     "@langchain/anthropic": "^0.2.12",
     "@langchain/community": "^0.2.25",
     "@langchain/openai": "^0.2.4",
-    "@langchain/scripts": "^0.0.21",
+    "@langchain/scripts": "^0.0.23",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
     "@tsconfig/recommended": "^1.0.3",
-    "@types/better-sqlite3": "^7.6.9",
     "@types/double-ended-queue": "^2",
     "@types/uuid": "^10",
     "@typescript-eslint/eslint-plugin": "^6.12.0",
     "@typescript-eslint/parser": "^6.12.0",
     "@xenova/transformers": "^2.17.2",
-    "better-sqlite3": "^9.5.0",
     "cheerio": "1.0.0-rc.12",
     "dotenv": "^16.3.1",
     "dpdm": "^3.12.0",
@@ -72,14 +80,6 @@
     "typescript": "^4.9.5 || ^5.4.5",
     "zod-to-json-schema": "^3.22.4"
   },
-  "peerDependencies": {
-    "better-sqlite3": "^9.5.0"
-  },
-  "peerDependenciesMeta": {
-    "better-sqlite3": {
-      "optional": true
-    }
-  },
   "publishConfig": {
     "access": "public",
     "registry": "https://registry.npmjs.org/"