@langchain/langgraph 0.1.0-rc.0 → 0.1.0

package/LICENSE

@@ -1,6 +1,6 @@
 The MIT License
 
-Copyright (c) 2023 LangChain
+Copyright (c) 2024 LangChain
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/dist/channels/base.cjs

@@ -1,8 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createCheckpoint = exports.emptyChannels = exports.BaseChannel = void 0;
-const base_js_1 = require("../checkpoint/base.cjs");
-const id_js_1 = require("../checkpoint/id.cjs");
+const langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
 const errors_js_1 = require("../errors.cjs");
 class BaseChannel {
     constructor() {
@@ -42,28 +41,34 @@ function emptyChannels(channels, checkpoint) {
 exports.emptyChannels = emptyChannels;
 function createCheckpoint(checkpoint, channels, step) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const values = {};
-    for (const k of Object.keys(channels)) {
-        try {
-            values[k] = channels[k].checkpoint();
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            if (error.name === errors_js_1.EmptyChannelError.unminifiable_name) {
-                // no-op
+    let values;
+    if (channels === undefined) {
+        values = checkpoint.channel_values;
+    }
+    else {
+        values = {};
+        for (const k of Object.keys(channels)) {
+            try {
+                values[k] = channels[k].checkpoint();
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
             }
-            else {
-                throw error; // Rethrow unexpected errors
+            catch (error) {
+                if (error.name === errors_js_1.EmptyChannelError.unminifiable_name) {
+                    // no-op
+                }
+                else {
+                    throw error; // Rethrow unexpected errors
+                }
             }
         }
     }
     return {
         v: 1,
-        id: (0, id_js_1.uuid6)(step),
+        id: (0, langgraph_checkpoint_1.uuid6)(step),
         ts: new Date().toISOString(),
         channel_values: values,
         channel_versions: { ...checkpoint.channel_versions },
-        versions_seen: (0, base_js_1.deepCopy)(checkpoint.versions_seen),
+        versions_seen: (0, langgraph_checkpoint_1.deepCopy)(checkpoint.versions_seen),
         pending_sends: checkpoint.pending_sends ?? [],
     };
 }

package/dist/channels/base.d.ts

@@ -1,5 +1,4 @@
-import { ReadonlyCheckpoint } from "../checkpoint/base.js";
-import { Checkpoint } from "../checkpoint/index.js";
+import { ReadonlyCheckpoint, Checkpoint } from "@langchain/langgraph-checkpoint";
 export declare abstract class BaseChannel<ValueType = unknown, UpdateType = unknown, CheckpointType = unknown> {
     ValueType: ValueType;
     UpdateType: UpdateType;
@@ -52,4 +51,4 @@ export declare abstract class BaseChannel<ValueType = unknown, UpdateType = unkn
     consume(): boolean;
 }
 export declare function emptyChannels<Cc extends Record<string, BaseChannel>>(channels: Cc, checkpoint: ReadonlyCheckpoint): Cc;
-export declare function createCheckpoint<ValueType>(checkpoint: ReadonlyCheckpoint, channels: Record<string, BaseChannel<ValueType>>, step: number): Checkpoint;
+export declare function createCheckpoint<ValueType>(checkpoint: ReadonlyCheckpoint, channels: Record<string, BaseChannel<ValueType>> | undefined, step: number): Checkpoint;

package/dist/channels/base.js

@@ -1,5 +1,4 @@
-import { deepCopy } from "../checkpoint/base.js";
-import { uuid6 } from "../checkpoint/id.js";
+import { deepCopy, uuid6, } from "@langchain/langgraph-checkpoint";
 import { EmptyChannelError } from "../errors.js";
 export class BaseChannel {
     constructor() {
@@ -37,18 +36,24 @@ export function emptyChannels(channels, checkpoint) {
 }
 export function createCheckpoint(checkpoint, channels, step) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const values = {};
-    for (const k of Object.keys(channels)) {
-        try {
-            values[k] = channels[k].checkpoint();
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            if (error.name === EmptyChannelError.unminifiable_name) {
-                // no-op
+    let values;
+    if (channels === undefined) {
+        values = checkpoint.channel_values;
+    }
+    else {
+        values = {};
+        for (const k of Object.keys(channels)) {
+            try {
+                values[k] = channels[k].checkpoint();
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
             }
-            else {
-                throw error; // Rethrow unexpected errors
+            catch (error) {
+                if (error.name === EmptyChannelError.unminifiable_name) {
+                    // no-op
+                }
+                else {
+                    throw error; // Rethrow unexpected errors
+                }
             }
         }
     }

package/dist/checkpoint/sqlite.cjs

@@ -1,202 +1,17 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SqliteSaver = void 0;
-const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
-const base_js_1 = require("./base.cjs");
-class SqliteSaver extends base_js_1.BaseCheckpointSaver {
-    constructor(db, serde) {
-        super(serde);
-        Object.defineProperty(this, "db", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "isSetup", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.db = db;
-        this.isSetup = false;
-    }
-    static fromConnString(connStringOrLocalPath) {
-        return new SqliteSaver(new better_sqlite3_1.default(connStringOrLocalPath));
-    }
-    setup() {
-        if (this.isSetup) {
-            return;
-        }
-        this.db.pragma("journal_mode=WAL");
-        this.db.exec(`
-CREATE TABLE IF NOT EXISTS checkpoints (
-  thread_id TEXT NOT NULL,
-  checkpoint_ns TEXT NOT NULL DEFAULT '',
-  checkpoint_id TEXT NOT NULL,
-  parent_checkpoint_id TEXT,
-  type TEXT,
-  checkpoint BLOB,
-  metadata BLOB,
-  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)
-);`);
-        this.db.exec(`
-CREATE TABLE IF NOT EXISTS writes (
-  thread_id TEXT NOT NULL,
-  checkpoint_ns TEXT NOT NULL DEFAULT '',
-  checkpoint_id TEXT NOT NULL,
-  task_id TEXT NOT NULL,
-  idx INTEGER NOT NULL,
-  channel TEXT NOT NULL,
-  type TEXT,
-  value BLOB,
-  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)
-);`);
-        this.isSetup = true;
-    }
-    async getTuple(config) {
-        this.setup();
-        const { thread_id, checkpoint_ns = "", checkpoint_id, } = config.configurable ?? {};
-        let row;
-        if (checkpoint_id) {
-            row = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata FROM checkpoints WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id = ?`)
-                .get(thread_id, checkpoint_ns, checkpoint_id);
-        }
-        else {
-            row = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata FROM checkpoints WHERE thread_id = ? AND checkpoint_ns = ? ORDER BY checkpoint_id DESC LIMIT 1`)
-                .get(thread_id, checkpoint_ns);
-        }
-        if (row === undefined) {
-            return undefined;
-        }
-        let finalConfig = config;
-        if (!checkpoint_id) {
-            finalConfig = {
-                configurable: {
-                    thread_id: row.thread_id,
-                    checkpoint_ns,
-                    checkpoint_id: row.checkpoint_id,
-                },
-            };
-        }
-        if (finalConfig.configurable?.thread_id === undefined ||
-            finalConfig.configurable?.checkpoint_id === undefined) {
-            throw new Error("Missing thread_id or checkpoint_id");
-        }
-        // find any pending writes
-        const pendingWritesRows = this.db
-            .prepare(`SELECT task_id, channel, type, value FROM writes WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id = ?`)
-            .all(finalConfig.configurable.thread_id.toString(), checkpoint_ns, finalConfig.configurable.checkpoint_id.toString());
-        const pendingWrites = await Promise.all(pendingWritesRows.map(async (row) => {
-            return [
-                row.task_id,
-                row.channel,
-                await this.serde.parse(row.value ?? ""),
-            ];
-        }));
-        return {
-            config: finalConfig,
-            checkpoint: (await this.serde.parse(row.checkpoint)),
-            metadata: (await this.serde.parse(row.metadata)),
-            parentConfig: row.parent_checkpoint_id
-                ? {
-                    configurable: {
-                        thread_id: row.thread_id,
-                        checkpoint_ns,
-                        checkpoint_id: row.parent_checkpoint_id,
-                    },
-                }
-                : undefined,
-            pendingWrites,
-        };
-    }
-    async *list(config, options) {
-        const { limit, before } = options ?? {};
-        this.setup();
-        const thread_id = config.configurable?.thread_id;
-        let sql = `SELECT thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, 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);
-        const rows = this.db
-            .prepare(sql)
-            .all(...args);
-        if (rows) {
-            for (const row of rows) {
-                yield {
-                    config: {
-                        configurable: {
-                            thread_id: row.thread_id,
-                            checkpoint_ns: row.checkpoint_ns,
-                            checkpoint_id: row.checkpoint_id,
-                        },
-                    },
-                    checkpoint: (await this.serde.parse(row.checkpoint)),
-                    metadata: (await this.serde.parse(row.metadata)),
-                    parentConfig: row.parent_checkpoint_id
-                        ? {
-                            configurable: {
-                                thread_id: row.thread_id,
-                                checkpoint_ns: row.checkpoint_ns,
-                                checkpoint_id: row.parent_checkpoint_id,
-                            },
-                        }
-                        : undefined,
-                };
-            }
-        }
-    }
-    async put(config, checkpoint, metadata) {
-        this.setup();
-        const row = [
-            config.configurable?.thread_id?.toString(),
-            config.configurable?.checkpoint_ns,
-            checkpoint.id,
-            config.configurable?.checkpoint_id,
-            "Checkpoint",
-            this.serde.stringify(checkpoint),
-            this.serde.stringify(metadata),
-        ];
-        this.db
-            .prepare(`INSERT OR REPLACE INTO checkpoints (thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`)
-            .run(...row);
-        return {
-            configurable: {
-                thread_id: config.configurable?.thread_id,
-                checkpoint_ns: config.configurable?.checkpoint_ns,
-                checkpoint_id: checkpoint.id,
-            },
-        };
-    }
-    async putWrites(config, writes, taskId) {
-        this.setup();
-        const stmt = this.db.prepare(`
-      INSERT OR REPLACE INTO writes 
-      (thread_id, checkpoint_ns, checkpoint_id, task_id, idx, channel, type, value) 
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    `);
-        const transaction = this.db.transaction((rows) => {
-            for (const row of rows) {
-                stmt.run(...row);
-            }
-        });
-        const rows = writes.map((write, idx) => [
-            config.configurable?.thread_id,
-            config.configurable?.checkpoint_ns,
-            config.configurable?.checkpoint_id,
-            taskId,
-            idx,
-            write[0],
-            "Checkpoint",
-            this.serde.stringify(write[1]),
-        ]);
-        transaction(rows);
-    }
-}
-exports.SqliteSaver = SqliteSaver;
+__exportStar(require("@langchain/langgraph-checkpoint-sqlite"), exports);

package/dist/checkpoint/sqlite.d.ts

@@ -1,16 +1 @@
-import { Database as DatabaseType } from "better-sqlite3";
-import { RunnableConfig } from "@langchain/core/runnables";
-import { BaseCheckpointSaver, Checkpoint, CheckpointListOptions, CheckpointTuple } from "./base.js";
-import { SerializerProtocol } from "../serde/base.js";
-import type { PendingWrite, CheckpointMetadata } from "../checkpoint/types.js";
-export declare class SqliteSaver extends BaseCheckpointSaver {
-    db: DatabaseType;
-    protected isSetup: boolean;
-    constructor(db: DatabaseType, serde?: SerializerProtocol<Checkpoint>);
-    static fromConnString(connStringOrLocalPath: string): SqliteSaver;
-    protected setup(): void;
-    getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
-    list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple>;
-    put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;
-    putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void>;
-}
+export * from "@langchain/langgraph-checkpoint-sqlite";

package/dist/checkpoint/sqlite.js

@@ -1,195 +1 @@
-import Database from "better-sqlite3";
-import { BaseCheckpointSaver, } from "./base.js";
-export class SqliteSaver extends BaseCheckpointSaver {
-    constructor(db, serde) {
-        super(serde);
-        Object.defineProperty(this, "db", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "isSetup", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.db = db;
-        this.isSetup = false;
-    }
-    static fromConnString(connStringOrLocalPath) {
-        return new SqliteSaver(new Database(connStringOrLocalPath));
-    }
-    setup() {
-        if (this.isSetup) {
-            return;
-        }
-        this.db.pragma("journal_mode=WAL");
-        this.db.exec(`
-CREATE TABLE IF NOT EXISTS checkpoints (
-  thread_id TEXT NOT NULL,
-  checkpoint_ns TEXT NOT NULL DEFAULT '',
-  checkpoint_id TEXT NOT NULL,
-  parent_checkpoint_id TEXT,
-  type TEXT,
-  checkpoint BLOB,
-  metadata BLOB,
-  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)
-);`);
-        this.db.exec(`
-CREATE TABLE IF NOT EXISTS writes (
-  thread_id TEXT NOT NULL,
-  checkpoint_ns TEXT NOT NULL DEFAULT '',
-  checkpoint_id TEXT NOT NULL,
-  task_id TEXT NOT NULL,
-  idx INTEGER NOT NULL,
-  channel TEXT NOT NULL,
-  type TEXT,
-  value BLOB,
-  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)
-);`);
-        this.isSetup = true;
-    }
-    async getTuple(config) {
-        this.setup();
-        const { thread_id, checkpoint_ns = "", checkpoint_id, } = config.configurable ?? {};
-        let row;
-        if (checkpoint_id) {
-            row = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata FROM checkpoints WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id = ?`)
-                .get(thread_id, checkpoint_ns, checkpoint_id);
-        }
-        else {
-            row = this.db
-                .prepare(`SELECT thread_id, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata FROM checkpoints WHERE thread_id = ? AND checkpoint_ns = ? ORDER BY checkpoint_id DESC LIMIT 1`)
-                .get(thread_id, checkpoint_ns);
-        }
-        if (row === undefined) {
-            return undefined;
-        }
-        let finalConfig = config;
-        if (!checkpoint_id) {
-            finalConfig = {
-                configurable: {
-                    thread_id: row.thread_id,
-                    checkpoint_ns,
-                    checkpoint_id: row.checkpoint_id,
-                },
-            };
-        }
-        if (finalConfig.configurable?.thread_id === undefined ||
-            finalConfig.configurable?.checkpoint_id === undefined) {
-            throw new Error("Missing thread_id or checkpoint_id");
-        }
-        // find any pending writes
-        const pendingWritesRows = this.db
-            .prepare(`SELECT task_id, channel, type, value FROM writes WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id = ?`)
-            .all(finalConfig.configurable.thread_id.toString(), checkpoint_ns, finalConfig.configurable.checkpoint_id.toString());
-        const pendingWrites = await Promise.all(pendingWritesRows.map(async (row) => {
-            return [
-                row.task_id,
-                row.channel,
-                await this.serde.parse(row.value ?? ""),
-            ];
-        }));
-        return {
-            config: finalConfig,
-            checkpoint: (await this.serde.parse(row.checkpoint)),
-            metadata: (await this.serde.parse(row.metadata)),
-            parentConfig: row.parent_checkpoint_id
-                ? {
-                    configurable: {
-                        thread_id: row.thread_id,
-                        checkpoint_ns,
-                        checkpoint_id: row.parent_checkpoint_id,
-                    },
-                }
-                : undefined,
-            pendingWrites,
-        };
-    }
-    async *list(config, options) {
-        const { limit, before } = options ?? {};
-        this.setup();
-        const thread_id = config.configurable?.thread_id;
-        let sql = `SELECT thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, 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);
-        const rows = this.db
-            .prepare(sql)
-            .all(...args);
-        if (rows) {
-            for (const row of rows) {
-                yield {
-                    config: {
-                        configurable: {
-                            thread_id: row.thread_id,
-                            checkpoint_ns: row.checkpoint_ns,
-                            checkpoint_id: row.checkpoint_id,
-                        },
-                    },
-                    checkpoint: (await this.serde.parse(row.checkpoint)),
-                    metadata: (await this.serde.parse(row.metadata)),
-                    parentConfig: row.parent_checkpoint_id
-                        ? {
-                            configurable: {
-                                thread_id: row.thread_id,
-                                checkpoint_ns: row.checkpoint_ns,
-                                checkpoint_id: row.parent_checkpoint_id,
-                            },
-                        }
-                        : undefined,
-                };
-            }
-        }
-    }
-    async put(config, checkpoint, metadata) {
-        this.setup();
-        const row = [
-            config.configurable?.thread_id?.toString(),
-            config.configurable?.checkpoint_ns,
-            checkpoint.id,
-            config.configurable?.checkpoint_id,
-            "Checkpoint",
-            this.serde.stringify(checkpoint),
-            this.serde.stringify(metadata),
-        ];
-        this.db
-            .prepare(`INSERT OR REPLACE INTO checkpoints (thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`)
-            .run(...row);
-        return {
-            configurable: {
-                thread_id: config.configurable?.thread_id,
-                checkpoint_ns: config.configurable?.checkpoint_ns,
-                checkpoint_id: checkpoint.id,
-            },
-        };
-    }
-    async putWrites(config, writes, taskId) {
-        this.setup();
-        const stmt = this.db.prepare(`
-      INSERT OR REPLACE INTO writes 
-      (thread_id, checkpoint_ns, checkpoint_id, task_id, idx, channel, type, value) 
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    `);
-        const transaction = this.db.transaction((rows) => {
-            for (const row of rows) {
-                stmt.run(...row);
-            }
-        });
-        const rows = writes.map((write, idx) => [
-            config.configurable?.thread_id,
-            config.configurable?.checkpoint_ns,
-            config.configurable?.checkpoint_id,
-            taskId,
-            idx,
-            write[0],
-            "Checkpoint",
-            this.serde.stringify(write[1]),
-        ]);
-        transaction(rows);
-    }
-}
+export * from "@langchain/langgraph-checkpoint-sqlite";

package/dist/graph/graph.d.ts

@@ -1,9 +1,9 @@
 import { Runnable, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
 import { Graph as RunnableGraph } from "@langchain/core/runnables/graph";
+import { BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import { PregelNode } from "../pregel/read.js";
 import { Pregel } from "../pregel/index.js";
 import type { PregelParams } from "../pregel/types.js";
-import { BaseCheckpointSaver } from "../checkpoint/base.js";
 import { BaseChannel } from "../channels/base.js";
 import { All } from "../pregel/types.js";
 import { Send } from "../constants.js";

package/dist/graph/message.d.ts

@@ -5,7 +5,4 @@ export declare function messagesStateReducer(left: BaseMessage[], right: Message
 export declare class MessageGraph extends StateGraph<BaseMessage[], BaseMessage[], Messages> {
     constructor();
 }
-export interface MessagesState {
-    messages: BaseMessage[];
-}
 export {};

package/dist/graph/messages_annotation.cjs

@@ -0,0 +1,12 @@
+"use strict";
+/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MessagesAnnotation = void 0;
+const annotation_js_1 = require("./annotation.cjs");
+const message_js_1 = require("./message.cjs");
+exports.MessagesAnnotation = annotation_js_1.Annotation.Root({
+    messages: (0, annotation_js_1.Annotation)({
+        reducer: message_js_1.messagesStateReducer,
+        default: () => [],
+    }),
+});

package/dist/graph/messages_annotation.d.ts

@@ -0,0 +1,4 @@
+import { BaseMessage } from "@langchain/core/messages";
+export declare const MessagesAnnotation: import("./annotation.js").AnnotationRoot<{
+    messages: import("../channels/binop.js").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
+}>;

package/dist/graph/messages_annotation.js

@@ -0,0 +1,9 @@
+/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
+import { Annotation } from "./annotation.js";
+import { messagesStateReducer } from "./message.js";
+export const MessagesAnnotation = Annotation.Root({
+    messages: Annotation({
+        reducer: messagesStateReducer,
+        default: () => [],
+    }),
+});

package/dist/graph/state.cjs

@@ -155,7 +155,8 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
                 return write_js_1.SKIP_WRITE;
             }
             else if (typeof input !== "object" || Array.isArray(input)) {
-                throw new errors_js_1.InvalidUpdateError(`Expected dict, got ${typeof input}`);
+                const typeofInput = Array.isArray(input) ? "array" : typeof input;
+                throw new errors_js_1.InvalidUpdateError(`Expected object, got ${typeofInput}`);
             }
             else {
                 return key in input ? input[key] : write_js_1.SKIP_WRITE;

package/dist/graph/state.d.ts

@@ -1,7 +1,7 @@
 import { Runnable, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
+import { BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { END, CompiledGraph, Graph, START, Branch } from "./graph.js";
-import { BaseCheckpointSaver } from "../checkpoint/base.js";
 import { All } from "../pregel/types.js";
 import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType } from "./annotation.js";
 export type ChannelReducers<Channels extends object> = {

package/dist/graph/state.js

@@ -151,7 +151,8 @@ export class CompiledStateGraph extends CompiledGraph {
                 return SKIP_WRITE;
             }
             else if (typeof input !== "object" || Array.isArray(input)) {
-                throw new InvalidUpdateError(`Expected dict, got ${typeof input}`);
+                const typeofInput = Array.isArray(input) ? "array" : typeof input;
+                throw new InvalidUpdateError(`Expected object, got ${typeofInput}`);
             }
             else {
                 return key in input ? input[key] : SKIP_WRITE;

package/dist/index.cjs

@@ -15,7 +15,10 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.MessagesAnnotation = void 0;
 const async_local_storage_js_1 = require("./setup/async_local_storage.cjs");
 // Initialize global async local storage instance for tracing
 (0, async_local_storage_js_1.initializeAsyncLocalStorageSingleton)();
 __exportStar(require("./web.cjs"), exports);
+var messages_annotation_js_1 = require("./graph/messages_annotation.cjs");
+Object.defineProperty(exports, "MessagesAnnotation", { enumerable: true, get: function () { return messages_annotation_js_1.MessagesAnnotation; } });

package/dist/index.d.ts

@@ -1 +1,2 @@
 export * from "./web.js";
+export { MessagesAnnotation } from "./graph/messages_annotation.js";

package/dist/index.js

@@ -3,3 +3,4 @@ import { initializeAsyncLocalStorageSingleton } from "./setup/async_local_storag
 // Initialize global async local storage instance for tracing
 initializeAsyncLocalStorageSingleton();
 export * from "./web.js";
+export { MessagesAnnotation } from "./graph/messages_annotation.js";