@langchain/langgraph 0.0.13 → 0.0.15

package/checkpoint/sqlite.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/checkpoint/sqlite.cjs');

package/checkpoint/sqlite.d.cts

@@ -0,0 +1 @@
+export * from '../dist/checkpoint/sqlite.js'

package/checkpoint/sqlite.d.ts

@@ -0,0 +1 @@
+export * from '../dist/checkpoint/sqlite.js'

package/checkpoint/sqlite.js

@@ -0,0 +1 @@
+export * from '../dist/checkpoint/sqlite.js'

package/dist/checkpoint/base.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
+const base_js_1 = require("../serde/base.cjs");
 const id_js_1 = require("./id.cjs");
 function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
@@ -43,7 +44,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

@@ -60,6 +60,6 @@ export declare abstract class BaseCheckpointSaver {
     constructor(serde?: SerializerProtocol<unknown>);
     get(config: RunnableConfig): Promise<Checkpoint | undefined>;
     abstract getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
-    abstract list(config: RunnableConfig): AsyncGenerator<CheckpointTuple>;
+    abstract list(config: RunnableConfig, limit?: number, before?: RunnableConfig): AsyncGenerator<CheckpointTuple>;
     abstract put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;
 }

package/dist/checkpoint/base.js

@@ -1,3 +1,4 @@
+import { DefaultSerializer } from "../serde/base.js";
 import { uuid6 } from "./id.js";
 export function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
@@ -37,7 +38,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,22 +33,25 @@ 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])),
                 };
             }
         }
         return undefined;
     }
-    async *list(config) {
+    async *list(config, limit, before) {
         const thread_id = config.configurable?.thread_id;
         const checkpoints = this.storage[thread_id] ?? {};
         // sort in desc order
-        for (const [checkpoint_id, checkpoint] of Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]))) {
+        for (const [checkpoint_id, checkpoint] of Object.entries(checkpoints)
+            .filter((c) => before ? c[0] < before.configurable?.checkpoint_id : true)
+            .sort((a, b) => b[0].localeCompare(a[0]))
+            .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.d.ts

@@ -5,6 +5,6 @@ export declare class MemorySaver extends BaseCheckpointSaver {
     storage: Record<string, Record<string, [string, string]>>;
     constructor(serde?: SerializerProtocol<unknown>);
     getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
-    list(config: RunnableConfig): AsyncGenerator<CheckpointTuple>;
+    list(config: RunnableConfig, limit?: number, before?: RunnableConfig): AsyncGenerator<CheckpointTuple>;
     put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;
 }

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,22 +30,25 @@ 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])),
                 };
             }
         }
         return undefined;
     }
-    async *list(config) {
+    async *list(config, limit, before) {
         const thread_id = config.configurable?.thread_id;
         const checkpoints = this.storage[thread_id] ?? {};
         // sort in desc order
-        for (const [checkpoint_id, checkpoint] of Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]))) {
+        for (const [checkpoint_id, checkpoint] of Object.entries(checkpoints)
+            .filter((c) => before ? c[0] < before.configurable?.checkpoint_id : true)
+            .sort((a, b) => b[0].localeCompare(a[0]))
+            .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: {
@@ -106,13 +106,16 @@ CREATE TABLE IF NOT EXISTS checkpoints (
         }
         return undefined;
     }
-    async *list(config) {
+    async *list(config, limit, before) {
         this.setup();
         const thread_id = config.configurable?.thread_id;
+        let sql = `SELECT thread_id, checkpoint_id, parent_id, checkpoint, metadata FROM checkpoints WHERE thread_id = ? ${before ? "AND checkpoint_id < ?" : ""} ORDER BY checkpoint_id DESC`;
+        if (limit) {
+            sql += ` LIMIT ${limit}`;
+        }
+        const args = [thread_id, before?.configurable?.checkpoint_id].filter(Boolean);
         try {
-            const rows = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_id, checkpoint, metadata FROM checkpoints WHERE thread_id = ? ORDER BY checkpoint_id DESC`)
-                .all(thread_id);
+            const rows = this.db.prepare(sql).all(...args);
             if (rows) {
                 for (const row of rows) {
                     yield {
@@ -122,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.d.ts

@@ -9,6 +9,6 @@ export declare class SqliteSaver extends BaseCheckpointSaver {
     static fromConnString(connStringOrLocalPath: string): SqliteSaver;
     private setup;
     getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
-    list(config: RunnableConfig): AsyncGenerator<CheckpointTuple>;
+    list(config: RunnableConfig, limit?: number, before?: RunnableConfig): AsyncGenerator<CheckpointTuple>;
     put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;
 }

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: {
@@ -100,13 +100,16 @@ CREATE TABLE IF NOT EXISTS checkpoints (
         }
         return undefined;
     }
-    async *list(config) {
+    async *list(config, limit, before) {
         this.setup();
         const thread_id = config.configurable?.thread_id;
+        let sql = `SELECT thread_id, checkpoint_id, parent_id, checkpoint, metadata FROM checkpoints WHERE thread_id = ? ${before ? "AND checkpoint_id < ?" : ""} ORDER BY checkpoint_id DESC`;
+        if (limit) {
+            sql += ` LIMIT ${limit}`;
+        }
+        const args = [thread_id, before?.configurable?.checkpoint_id].filter(Boolean);
         try {
-            const rows = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_id, checkpoint, metadata FROM checkpoints WHERE thread_id = ? ORDER BY checkpoint_id DESC`)
-                .all(thread_id);
+            const rows = this.db.prepare(sql).all(...args);
             if (rows) {
                 for (const row of rows) {
                     yield {
@@ -116,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/index.cjs

@@ -1,17 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.EmptyChannelError = exports.InvalidUpdateError = exports.GraphValueError = exports.GraphRecursionError = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.MemorySaver = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
+exports.EmptyChannelError = exports.InvalidUpdateError = exports.GraphValueError = exports.GraphRecursionError = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = 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; } });
 Object.defineProperty(exports, "START", { enumerable: true, get: function () { return index_js_1.START; } });
 Object.defineProperty(exports, "StateGraph", { enumerable: true, get: function () { return index_js_1.StateGraph; } });
 Object.defineProperty(exports, "MessageGraph", { enumerable: true, get: function () { return index_js_1.MessageGraph; } });
-var index_js_2 = require("./checkpoint/index.cjs");
-Object.defineProperty(exports, "MemorySaver", { enumerable: true, get: function () { return index_js_2.MemorySaver; } });
-var index_js_3 = require("./checkpoint/index.cjs");
-Object.defineProperty(exports, "emptyCheckpoint", { enumerable: true, get: function () { return index_js_3.emptyCheckpoint; } });
-Object.defineProperty(exports, "BaseCheckpointSaver", { enumerable: true, get: function () { return index_js_3.BaseCheckpointSaver; } });
+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; } });

package/dist/index.d.ts

@@ -1,4 +1,4 @@
 export { END, Graph, type StateGraphArgs, START, StateGraph, MessageGraph, } from "./graph/index.js";
-export { MemorySaver } from "./checkpoint/index.js";
-export { type Checkpoint, emptyCheckpoint, BaseCheckpointSaver, } from "./checkpoint/index.js";
+export { MemorySaver } from "./checkpoint/memory.js";
+export { type Checkpoint, type CheckpointMetadata, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "./checkpoint/base.js";
 export { GraphRecursionError, GraphValueError, InvalidUpdateError, EmptyChannelError, } from "./errors.js";

package/dist/index.js

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

package/dist/pregel/index.cjs

@@ -208,6 +208,108 @@ class Pregel extends runnables_1.Runnable {
             return Object.keys(this.channels);
         }
     }
+    async getState(config) {
+        if (!this.checkpointer) {
+            throw new errors_js_1.GraphValueError("No checkpointer set");
+        }
+        const saved = await this.checkpointer.getTuple(config);
+        const checkpoint = saved ? saved.checkpoint : (0, base_js_2.emptyCheckpoint)();
+        const channels = (0, base_js_1.emptyChannels)(this.channels, checkpoint);
+        // eslint-disable-next-line @typescript-eslint/no-unused-vars
+        const [_, nextTasks] = _prepareNextTasks(checkpoint, this.nodes, channels, false);
+        return {
+            values: (0, io_js_1.readChannels)(channels, this.streamChannelsAsIs),
+            next: nextTasks.map((task) => task.name),
+            metadata: saved?.metadata,
+            config: saved ? saved.config : config,
+            parentConfig: saved?.parentConfig,
+        };
+    }
+    async *getStateHistory(config, limit, before) {
+        if (!this.checkpointer) {
+            throw new errors_js_1.GraphValueError("No checkpointer set");
+        }
+        for await (const saved of this.checkpointer.list(config, limit, before)) {
+            const channels = (0, base_js_1.emptyChannels)(this.channels, saved.checkpoint);
+            // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            const [_, nextTasks] = _prepareNextTasks(saved.checkpoint, this.nodes, channels, false);
+            yield {
+                values: (0, io_js_1.readChannels)(channels, this.streamChannelsAsIs),
+                next: nextTasks.map((task) => task.name),
+                metadata: saved.metadata,
+                config: saved.config,
+                parentConfig: saved.parentConfig,
+            };
+        }
+    }
+    async updateState(config, values, asNode) {
+        if (!this.checkpointer) {
+            throw new errors_js_1.GraphValueError("No checkpointer set");
+        }
+        // Get the latest checkpoint
+        const saved = await this.checkpointer.getTuple(config);
+        const checkpoint = saved
+            ? (0, base_js_2.copyCheckpoint)(saved.checkpoint)
+            : (0, base_js_2.emptyCheckpoint)();
+        // Find last that updated the state, if not provided
+        const maxSeens = Object.entries(checkpoint.versions_seen).reduce((acc, [node, versions]) => {
+            const maxSeen = Math.max(...Object.values(versions));
+            if (maxSeen) {
+                if (!acc[maxSeen]) {
+                    acc[maxSeen] = [];
+                }
+                acc[maxSeen].push(node);
+            }
+            return acc;
+        }, {});
+        if (!asNode && !Object.keys(maxSeens).length) {
+            if (!Array.isArray(this.inputs) && this.inputs in this.nodes) {
+                asNode = this.inputs;
+            }
+        }
+        else if (!asNode) {
+            const maxSeen = Math.max(...Object.keys(maxSeens).map(Number));
+            const nodes = maxSeens[maxSeen];
+            if (nodes.length === 1) {
+                asNode = nodes[0];
+            }
+        }
+        if (!asNode) {
+            throw new errors_js_1.InvalidUpdateError("Ambiguous update, specify as_node");
+        }
+        // update channels
+        const channels = (0, base_js_1.emptyChannels)(this.channels, checkpoint);
+        // create task to run all writers of the chosen node
+        const writers = this.nodes[asNode].getWriters();
+        if (!writers.length) {
+            throw new errors_js_1.InvalidUpdateError(`No writers found for node ${asNode}`);
+        }
+        const task = {
+            name: asNode,
+            input: values,
+            proc: 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            writers.length > 1 ? runnables_1.RunnableSequence.from(writers) : writers[0],
+            writes: [],
+            config: undefined,
+        };
+        // execute task
+        await task.proc.invoke(task.input, (0, runnables_1.patchConfig)(config, {
+            runName: `${this.name}UpdateState`,
+            configurable: {
+                [constants_js_1.CONFIG_KEY_SEND]: (items) => task.writes.push(...items),
+                [constants_js_1.CONFIG_KEY_READ]: _localRead.bind(undefined, checkpoint, channels, task.writes),
+            },
+        }));
+        // apply to checkpoint and save
+        _applyWrites(checkpoint, channels, task.writes);
+        const step = (saved?.metadata?.step ?? -2) + 1;
+        return await this.checkpointer.put(saved?.config ?? config, (0, base_js_1.createCheckpoint)(checkpoint, channels, step), {
+            source: "update",
+            step,
+            writes: { [asNode]: values },
+        });
+    }
     _defaults(config) {
         const { debug, streamMode, inputKeys, outputKeys, interruptAfter, interruptBefore, ...rest } = config;
         const defaultDebug = debug !== undefined ? debug : this.debug;

package/dist/pregel/index.d.ts

@@ -5,7 +5,7 @@ import { BaseChannel } from "../channels/base.js";
 import { BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint } from "../checkpoint/base.js";
 import { PregelNode } from "./read.js";
 import { ChannelWrite } from "./write.js";
-import { All, PregelExecutableTask, PregelTaskDescription } from "./types.js";
+import { All, PregelExecutableTask, PregelTaskDescription, StateSnapshot } from "./types.js";
 type WriteValue = Runnable | RunnableFunc<unknown, unknown> | unknown;
 export declare class Channel {
     static subscribeTo(channels: string, options?: {
@@ -85,6 +85,9 @@ export declare class Pregel<const Nn extends StrRecord<string, PregelNode>, cons
     validate(): this;
     get streamChannelsList(): Array<keyof Cc>;
     get streamChannelsAsIs(): keyof Cc | Array<keyof Cc>;
+    getState(config: RunnableConfig): Promise<StateSnapshot>;
+    getStateHistory(config: RunnableConfig, limit?: number, before?: RunnableConfig): AsyncIterableIterator<StateSnapshot>;
+    updateState(config: RunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn): Promise<RunnableConfig>;
     _defaults(config: PregelOptions<Nn, Cc>): [
         boolean,
         StreamMode,

package/dist/pregel/index.js

@@ -1,5 +1,5 @@
 /* eslint-disable no-param-reassign */
-import { Runnable, _coerceToRunnable, ensureConfig, patchConfig, } from "@langchain/core/runnables";
+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";
@@ -204,6 +204,108 @@ export class Pregel extends Runnable {
             return Object.keys(this.channels);
         }
     }
+    async getState(config) {
+        if (!this.checkpointer) {
+            throw new GraphValueError("No checkpointer set");
+        }
+        const saved = await this.checkpointer.getTuple(config);
+        const checkpoint = saved ? saved.checkpoint : emptyCheckpoint();
+        const channels = emptyChannels(this.channels, checkpoint);
+        // eslint-disable-next-line @typescript-eslint/no-unused-vars
+        const [_, nextTasks] = _prepareNextTasks(checkpoint, this.nodes, channels, false);
+        return {
+            values: readChannels(channels, this.streamChannelsAsIs),
+            next: nextTasks.map((task) => task.name),
+            metadata: saved?.metadata,
+            config: saved ? saved.config : config,
+            parentConfig: saved?.parentConfig,
+        };
+    }
+    async *getStateHistory(config, limit, before) {
+        if (!this.checkpointer) {
+            throw new GraphValueError("No checkpointer set");
+        }
+        for await (const saved of this.checkpointer.list(config, limit, before)) {
+            const channels = emptyChannels(this.channels, saved.checkpoint);
+            // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            const [_, nextTasks] = _prepareNextTasks(saved.checkpoint, this.nodes, channels, false);
+            yield {
+                values: readChannels(channels, this.streamChannelsAsIs),
+                next: nextTasks.map((task) => task.name),
+                metadata: saved.metadata,
+                config: saved.config,
+                parentConfig: saved.parentConfig,
+            };
+        }
+    }
+    async updateState(config, values, asNode) {
+        if (!this.checkpointer) {
+            throw new GraphValueError("No checkpointer set");
+        }
+        // Get the latest checkpoint
+        const saved = await this.checkpointer.getTuple(config);
+        const checkpoint = saved
+            ? copyCheckpoint(saved.checkpoint)
+            : emptyCheckpoint();
+        // Find last that updated the state, if not provided
+        const maxSeens = Object.entries(checkpoint.versions_seen).reduce((acc, [node, versions]) => {
+            const maxSeen = Math.max(...Object.values(versions));
+            if (maxSeen) {
+                if (!acc[maxSeen]) {
+                    acc[maxSeen] = [];
+                }
+                acc[maxSeen].push(node);
+            }
+            return acc;
+        }, {});
+        if (!asNode && !Object.keys(maxSeens).length) {
+            if (!Array.isArray(this.inputs) && this.inputs in this.nodes) {
+                asNode = this.inputs;
+            }
+        }
+        else if (!asNode) {
+            const maxSeen = Math.max(...Object.keys(maxSeens).map(Number));
+            const nodes = maxSeens[maxSeen];
+            if (nodes.length === 1) {
+                asNode = nodes[0];
+            }
+        }
+        if (!asNode) {
+            throw new InvalidUpdateError("Ambiguous update, specify as_node");
+        }
+        // update channels
+        const channels = emptyChannels(this.channels, checkpoint);
+        // create task to run all writers of the chosen node
+        const writers = this.nodes[asNode].getWriters();
+        if (!writers.length) {
+            throw new InvalidUpdateError(`No writers found for node ${asNode}`);
+        }
+        const task = {
+            name: asNode,
+            input: values,
+            proc: 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            writers.length > 1 ? RunnableSequence.from(writers) : writers[0],
+            writes: [],
+            config: undefined,
+        };
+        // execute task
+        await task.proc.invoke(task.input, patchConfig(config, {
+            runName: `${this.name}UpdateState`,
+            configurable: {
+                [CONFIG_KEY_SEND]: (items) => task.writes.push(...items),
+                [CONFIG_KEY_READ]: _localRead.bind(undefined, checkpoint, channels, task.writes),
+            },
+        }));
+        // apply to checkpoint and save
+        _applyWrites(checkpoint, channels, task.writes);
+        const step = (saved?.metadata?.step ?? -2) + 1;
+        return await this.checkpointer.put(saved?.config ?? config, createCheckpoint(checkpoint, channels, step), {
+            source: "update",
+            step,
+            writes: { [asNode]: values },
+        });
+    }
     _defaults(config) {
         const { debug, streamMode, inputKeys, outputKeys, interruptAfter, interruptBefore, ...rest } = config;
         const defaultDebug = debug !== undefined ? debug : this.debug;

package/dist/pregel/types.d.ts

@@ -1,4 +1,5 @@
 import { Runnable, RunnableConfig } from "@langchain/core/runnables";
+import { CheckpointMetadata } from "../checkpoint/base.js";
 export interface PregelTaskDescription {
     readonly name: string;
     readonly input: unknown;
@@ -23,6 +24,10 @@ export interface StateSnapshot {
      * Config used to fetch this snapshot
      */
     readonly config: RunnableConfig;
+    /**
+     * Metadata about the checkpoint
+     */
+    readonly metadata?: CheckpointMetadata;
     /**
      * Config used to fetch the parent snapshot, if any
      * @default undefined

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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.0.13",
+  "version": "0.0.15",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -38,7 +38,6 @@
   "license": "MIT",
   "dependencies": {
     "@langchain/core": "^0.1.61",
-    "better-sqlite3": "^9.5.0",
     "uuid": "^9.0.1"
   },
   "devDependencies": {
@@ -53,6 +52,7 @@
     "@types/uuid": "^9",
     "@typescript-eslint/eslint-plugin": "^6.12.0",
     "@typescript-eslint/parser": "^6.12.0",
+    "better-sqlite3": "^9.5.0",
     "dotenv": "^16.3.1",
     "dpdm": "^3.12.0",
     "eslint": "^8.33.0",
@@ -73,6 +73,14 @@
     "zod": "^3.22.4",
     "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/"
@@ -105,6 +113,15 @@
       "import": "./prebuilt.js",
       "require": "./prebuilt.cjs"
     },
+    "./checkpoint/sqlite": {
+      "types": {
+        "import": "./checkpoint/sqlite.d.ts",
+        "require": "./checkpoint/sqlite.d.cts",
+        "default": "./checkpoint/sqlite.d.ts"
+      },
+      "import": "./checkpoint/sqlite.js",
+      "require": "./checkpoint/sqlite.cjs"
+    },
     "./package.json": "./package.json"
   },
   "files": [
@@ -120,6 +137,10 @@
     "prebuilt.cjs",
     "prebuilt.js",
     "prebuilt.d.ts",
-    "prebuilt.d.cts"
+    "prebuilt.d.cts",
+    "checkpoint/sqlite.cjs",
+    "checkpoint/sqlite.js",
+    "checkpoint/sqlite.d.ts",
+    "checkpoint/sqlite.d.cts"
   ]
 }