@langchain/langgraph 0.2.73 → 0.3.0

package/dist/graph/message.js

@@ -1,6 +1,7 @@
 import { coerceMessageLikeToMessage, } from "@langchain/core/messages";
 import { v4 } from "uuid";
 import { StateGraph } from "./state.js";
+export const REMOVE_ALL_MESSAGES = "__remove_all__";
 /**
  * Prebuilt reducer that combines returned messages.
  * Can handle standard messages and special modifiers like {@link RemoveMessage}
@@ -19,12 +20,19 @@ export function messagesStateReducer(left, right) {
             m.lc_kwargs.id = m.id;
         }
     }
-    for (const m of rightMessages) {
+    let removeAllIdx;
+    for (let i = 0; i < rightMessages.length; i += 1) {
+        const m = rightMessages[i];
         if (m.id === null || m.id === undefined) {
             m.id = v4();
             m.lc_kwargs.id = m.id;
         }
+        if (m.getType() === "remove" && m.id === REMOVE_ALL_MESSAGES) {
+            removeAllIdx = i;
+        }
     }
+    if (removeAllIdx != null)
+        return rightMessages.slice(removeAllIdx + 1);
     // merge
     const merged = [...leftMessages];
     const mergedById = new Map(merged.map((m, i) => [m.id, i]));
@@ -32,7 +40,7 @@ export function messagesStateReducer(left, right) {
     for (const m of rightMessages) {
         const existingIdx = mergedById.get(m.id);
         if (existingIdx !== undefined) {
-            if (m._getType() === "remove") {
+            if (m.getType() === "remove") {
                 idsToRemove.add(m.id);
             }
             else {
@@ -41,7 +49,7 @@ export function messagesStateReducer(left, right) {
             }
         }
         else {
-            if (m._getType() === "remove") {
+            if (m.getType() === "remove") {
                 throw new Error(`Attempting to delete a message with an ID that doesn't exist ('${m.id}')`);
             }
             mergedById.set(m.id, merged.length);
@@ -63,4 +71,33 @@ export class MessageGraph extends StateGraph {
         });
     }
 }
+export function pushMessage(message, config, options) {
+    let stateKey = options?.stateKey ?? "messages";
+    if (options?.stateKey === null) {
+        stateKey = undefined;
+    }
+    // coerce to message
+    const validMessage = coerceMessageLikeToMessage(message);
+    if (!validMessage.id)
+        throw new Error("Message ID is required.");
+    const callbacks = (() => {
+        if (Array.isArray(config.callbacks)) {
+            return config.callbacks;
+        }
+        if (typeof config.callbacks !== "undefined") {
+            return config.callbacks.handlers;
+        }
+        return [];
+    })();
+    const messagesHandler = callbacks.find((cb) => "name" in cb && cb.name === "StreamMessagesHandler");
+    if (messagesHandler) {
+        const metadata = config.metadata ?? {};
+        const namespace = (metadata.langgraph_checkpoint_ns ?? "").split("|");
+        messagesHandler._emit([namespace, metadata], validMessage, undefined, false);
+    }
+    if (stateKey) {
+        config.configurable?.__pregel_send?.([[stateKey, validMessage]]);
+    }
+    return validMessage;
+}
 //# sourceMappingURL=message.js.map

package/dist/graph/message.js.map

@@ -1 +1 @@
-{"version":3,"file":"message.js","sourceRoot":"","sources":["../../src/graph/message.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAOxC;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAClC,IAAc,EACd,KAAe;IAEf,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC1D,oBAAoB;IACpB,MAAM,YAAY,GAAI,SAA+B,CAAC,GAAG,CACvD,0BAA0B,CAC3B,CAAC;IACF,MAAM,aAAa,GAAI,UAAgC,CAAC,GAAG,CACzD,0BAA0B,CAC3B,CAAC;IACF,qBAAqB;IACrB,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE;QAC5B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE;YACvC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;SACvB;KACF;IACD,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE;QAC7B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE;YACvC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;SACvB;KACF;IACD,QAAQ;IACR,MAAM,MAAM,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;IACjC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE;QAC7B,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzC,IAAI,WAAW,KAAK,SAAS,EAAE;YAC7B,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,QAAQ,EAAE;gBAC7B,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aACvB;iBAAM;gBACL,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;aACzB;SACF;aAAM;YACL,IAAI,CAAC,CAAC,QAAQ,EAAE,KAAK,QAAQ,EAAE;gBAC7B,MAAM,IAAI,KAAK,CACb,kEAAkE,CAAC,CAAC,EAAE,IAAI,CAC3E,CAAC;aACH;YACD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SAChB;KACF;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,cAAc;AACd,MAAM,OAAO,YAAa,SAAQ,UAIjC;IACC;QACE,KAAK,CAAC;YACJ,QAAQ,EAAE;gBACR,QAAQ,EAAE;oBACR,OAAO,EAAE,oBAAoB;oBAC7B,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;iBAClB;aACF;SACF,CAAC,CAAC;IACL,CAAC;CACF"}
+{"version":3,"file":"message.js","sourceRoot":"","sources":["../../src/graph/message.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAIxC,MAAM,CAAC,MAAM,mBAAmB,GAAG,gBAAgB,CAAC;AAOpD;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAClC,IAAc,EACd,KAAe;IAEf,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC1D,oBAAoB;IACpB,MAAM,YAAY,GAAI,SAA+B,CAAC,GAAG,CACvD,0BAA0B,CAC3B,CAAC;IACF,MAAM,aAAa,GAAI,UAAgC,CAAC,GAAG,CACzD,0BAA0B,CAC3B,CAAC;IACF,qBAAqB;IACrB,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACxC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,YAAgC,CAAC;IACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACjD,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACxC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC,EAAE,KAAK,mBAAmB,EAAE,CAAC;YAC7D,YAAY,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,IAAI,YAAY,IAAI,IAAI;QAAE,OAAO,aAAa,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;IAEvE,QAAQ;IACR,MAAM,MAAM,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;IACjC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;gBAC7B,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CACb,kEAAkE,CAAC,CAAC,EAAE,IAAI,CAC3E,CAAC;YACJ,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,cAAc;AACd,MAAM,OAAO,YAAa,SAAQ,UAIjC;IACC;QACE,KAAK,CAAC;YACJ,QAAQ,EAAE;gBACR,QAAQ,EAAE;oBACR,OAAO,EAAE,oBAAoB;oBAC7B,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;iBAClB;aACF;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,UAAU,WAAW,CACzB,OAAsC,EACtC,MAA+B,EAC/B,OAAsC;IAEtC,IAAI,QAAQ,GAAuB,OAAO,EAAE,QAAQ,IAAI,UAAU,CAAC;IACnE,IAAI,OAAO,EAAE,QAAQ,KAAK,IAAI,EAAE,CAAC;QAC/B,QAAQ,GAAG,SAAS,CAAC;IACvB,CAAC;IAED,oBAAoB;IACpB,MAAM,YAAY,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,CAAC,YAAY,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAEjE,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE;QACtB,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACpC,OAAO,MAAM,CAAC,SAAS,CAAC;QAC1B,CAAC;QAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,WAAW,EAAE,CAAC;YAC5C,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC;QACnC,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,EAAE,CAAC;IAEL,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,CAAC,EAAE,EAA+B,EAAE,CAClC,MAAM,IAAI,EAAE,IAAI,EAAE,CAAC,IAAI,KAAK,uBAAuB,CACtD,CAAC;IAEF,IAAI,eAAe,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;QACvC,MAAM,SAAS,GACb,CAAC,QAAQ,CAAC,uBAAuB,IAAI,EAAE,CACxC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAEb,eAAe,CAAC,KAAK,CACnB,CAAC,SAAS,EAAE,QAAQ,CAAC,EACrB,YAAY,EACZ,SAAS,EACT,KAAK,CACN,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,CAAC,YAAY,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC"}

package/dist/graph/message.test.cjs

@@ -0,0 +1,196 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const messages_1 = require("@langchain/core/messages");
+const message_js_1 = require("./message.cjs");
+const constants_js_1 = require("../constants.cjs");
+const state_js_1 = require("../graph/state.cjs");
+const messages_annotation_js_1 = require("../graph/messages_annotation.cjs");
+(0, vitest_1.describe)("messagesStateReducer", () => {
+    (0, vitest_1.it)("should add a single message", () => {
+        const left = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        const right = new messages_1.AIMessage({ id: "2", content: "Hi there!" });
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should add multiple messages", () => {
+        const left = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        const right = [
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+            new messages_1.SystemMessage({ id: "3", content: "System message" }),
+        ];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+            new messages_1.SystemMessage({ id: "3", content: "System message" }),
+        ];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should update existing message", () => {
+        const left = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        const right = new messages_1.HumanMessage({ id: "1", content: "Hello again" });
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [new messages_1.HumanMessage({ id: "1", content: "Hello again" })];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should assign missing IDs", () => {
+        const left = [new messages_1.HumanMessage({ content: "Hello" })];
+        const right = [new messages_1.AIMessage({ content: "Hi there!" })];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        (0, vitest_1.expect)(result).toHaveLength(2);
+        (0, vitest_1.expect)(result.every((m) => typeof m.id === "string" && m.id.length > 0)).toBe(true);
+    });
+    (0, vitest_1.it)("should handle duplicates in input", () => {
+        const left = [];
+        const right = [
+            new messages_1.AIMessage({ id: "1", content: "Hi there!" }),
+            new messages_1.AIMessage({ id: "1", content: "Hi there again!" }),
+        ];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        (0, vitest_1.expect)(result).toHaveLength(1);
+        (0, vitest_1.expect)(result[0].id).toBe("1");
+        (0, vitest_1.expect)(result[0].content).toBe("Hi there again!");
+    });
+    (0, vitest_1.it)("should handle duplicates with remove", () => {
+        const left = [new messages_1.AIMessage({ id: "1", content: "Hello!" })];
+        const right = [
+            new messages_1.RemoveMessage({ id: "1" }),
+            new messages_1.AIMessage({ id: "1", content: "Hi there!" }),
+            new messages_1.AIMessage({ id: "1", content: "Hi there again!" }),
+        ];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        (0, vitest_1.expect)(result).toHaveLength(1);
+        (0, vitest_1.expect)(result[0].id).toBe("1");
+        (0, vitest_1.expect)(result[0].content).toBe("Hi there again!");
+    });
+    (0, vitest_1.it)("should remove message", () => {
+        const left = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = new messages_1.RemoveMessage({ id: "2" });
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should handle duplicate remove messages", () => {
+        const left = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = [
+            new messages_1.RemoveMessage({ id: "2" }),
+            new messages_1.RemoveMessage({ id: "2" }),
+        ];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should throw on removing nonexistent message", () => {
+        const left = [new messages_1.HumanMessage({ id: "1", content: "Hello" })];
+        const right = new messages_1.RemoveMessage({ id: "2" });
+        (0, vitest_1.expect)(() => (0, message_js_1.messagesStateReducer)(left, right)).toThrow("Attempting to delete a message with an ID that doesn't exist");
+    });
+    (0, vitest_1.it)("should handle mixed operations", () => {
+        const left = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = [
+            new messages_1.HumanMessage({ id: "1", content: "Updated hello" }),
+            new messages_1.RemoveMessage({ id: "2" }),
+            new messages_1.SystemMessage({ id: "3", content: "New message" }),
+        ];
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [
+            new messages_1.HumanMessage({ id: "1", content: "Updated hello" }),
+            new messages_1.SystemMessage({ id: "3", content: "New message" }),
+        ];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should handle empty inputs", () => {
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([], [])).toEqual([]);
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([], [new messages_1.HumanMessage({ id: "1", content: "Hello" })])).toEqual([new messages_1.HumanMessage({ id: "1", content: "Hello" })]);
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([new messages_1.HumanMessage({ id: "1", content: "Hello" })], [])).toEqual([new messages_1.HumanMessage({ id: "1", content: "Hello" })]);
+    });
+    (0, vitest_1.it)("should handle non-array inputs", () => {
+        const left = new messages_1.HumanMessage({ id: "1", content: "Hello" });
+        const right = new messages_1.AIMessage({ id: "2", content: "Hi there!" });
+        const result = (0, message_js_1.messagesStateReducer)(left, right);
+        const expected = [
+            new messages_1.HumanMessage({ id: "1", content: "Hello" }),
+            new messages_1.AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        (0, vitest_1.expect)(result).toEqual(expected);
+    });
+    (0, vitest_1.it)("should remove all messages", () => {
+        // simple removal
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([new messages_1.HumanMessage("Hello"), new messages_1.AIMessage("Hi there!")], [new messages_1.RemoveMessage({ id: message_js_1.REMOVE_ALL_MESSAGES })])).toEqual([]);
+        // removal and update (i.e. overwriting)
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([new messages_1.HumanMessage("Hello"), new messages_1.AIMessage("Hi there!")], [
+            new messages_1.RemoveMessage({ id: message_js_1.REMOVE_ALL_MESSAGES }),
+            new messages_1.HumanMessage({ id: "1", content: "Updated hello" }),
+        ])).toEqual([new messages_1.HumanMessage({ id: "1", content: "Updated hello" })]);
+        // test removing preceding messages in the right list
+        (0, vitest_1.expect)((0, message_js_1.messagesStateReducer)([new messages_1.HumanMessage("Hello"), new messages_1.AIMessage("Hi there!")], [
+            new messages_1.HumanMessage("Updated hello"),
+            new messages_1.RemoveMessage({ id: message_js_1.REMOVE_ALL_MESSAGES }),
+            new messages_1.HumanMessage({ id: "1", content: "Updated hi there" }),
+        ])).toEqual([new messages_1.HumanMessage({ id: "1", content: "Updated hi there" })]);
+    });
+});
+(0, vitest_1.describe)("pushMessage", () => {
+    (0, vitest_1.it)("should throw on message without ID", () => {
+        const message = new messages_1.AIMessage("No ID");
+        const config = { callbacks: [] };
+        (0, vitest_1.expect)(() => (0, message_js_1.pushMessage)(message, config)).toThrow("Message ID is required");
+    });
+    (0, vitest_1.it)("should handle message with ID", () => {
+        const message = new messages_1.AIMessage({ id: "1", content: "With ID" });
+        const config = { callbacks: [] };
+        const result = (0, message_js_1.pushMessage)(message, config);
+        (0, vitest_1.expect)(result).toEqual(message);
+    });
+    (0, vitest_1.it)("should handle message with custom state key", () => {
+        const message = new messages_1.AIMessage({ id: "1", content: "With ID" });
+        const config = {
+            callbacks: [],
+            configurable: {
+                __pregel_send: (messages) => {
+                    (0, vitest_1.expect)(messages).toEqual([["custom", message]]);
+                },
+            },
+        };
+        (0, message_js_1.pushMessage)(message, config, { stateKey: "custom" });
+    });
+    (0, vitest_1.it)("should push messages in graph", async () => {
+        const graph = new state_js_1.StateGraph(messages_annotation_js_1.MessagesAnnotation)
+            .addNode("chat", (state, config) => {
+            (0, vitest_1.expect)(() => (0, message_js_1.pushMessage)(new messages_1.AIMessage("No ID"), config)).toThrow();
+            (0, message_js_1.pushMessage)(new messages_1.AIMessage({ id: "1", content: "First" }), config);
+            (0, message_js_1.pushMessage)(new messages_1.HumanMessage({ id: "2", content: "Second" }), config);
+            (0, message_js_1.pushMessage)(new messages_1.AIMessage({ id: "3", content: "Third" }), config);
+            return state;
+        })
+            .addEdge(constants_js_1.START, "chat")
+            .compile();
+        const messages = [];
+        let values;
+        for await (const [event, chunk] of await graph.stream({ messages: [] }, { streamMode: ["messages", "values"] })) {
+            if (event === "values") {
+                values = chunk.messages;
+            }
+            else if (event === "messages") {
+                const [message] = chunk;
+                messages.push(message);
+            }
+        }
+        (0, vitest_1.expect)(values).toEqual(messages);
+    });
+});
+//# sourceMappingURL=message.test.js.map

package/dist/graph/message.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/graph/message.test.js

@@ -0,0 +1,194 @@
+import { describe, it, expect } from "vitest";
+import { AIMessage, HumanMessage, SystemMessage, RemoveMessage, } from "@langchain/core/messages";
+import { messagesStateReducer, pushMessage, REMOVE_ALL_MESSAGES, } from "./message.js";
+import { START } from "../constants.js";
+import { StateGraph } from "../graph/state.js";
+import { MessagesAnnotation } from "../graph/messages_annotation.js";
+describe("messagesStateReducer", () => {
+    it("should add a single message", () => {
+        const left = [new HumanMessage({ id: "1", content: "Hello" })];
+        const right = new AIMessage({ id: "2", content: "Hi there!" });
+        const result = messagesStateReducer(left, right);
+        const expected = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        expect(result).toEqual(expected);
+    });
+    it("should add multiple messages", () => {
+        const left = [new HumanMessage({ id: "1", content: "Hello" })];
+        const right = [
+            new AIMessage({ id: "2", content: "Hi there!" }),
+            new SystemMessage({ id: "3", content: "System message" }),
+        ];
+        const result = messagesStateReducer(left, right);
+        const expected = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+            new SystemMessage({ id: "3", content: "System message" }),
+        ];
+        expect(result).toEqual(expected);
+    });
+    it("should update existing message", () => {
+        const left = [new HumanMessage({ id: "1", content: "Hello" })];
+        const right = new HumanMessage({ id: "1", content: "Hello again" });
+        const result = messagesStateReducer(left, right);
+        const expected = [new HumanMessage({ id: "1", content: "Hello again" })];
+        expect(result).toEqual(expected);
+    });
+    it("should assign missing IDs", () => {
+        const left = [new HumanMessage({ content: "Hello" })];
+        const right = [new AIMessage({ content: "Hi there!" })];
+        const result = messagesStateReducer(left, right);
+        expect(result).toHaveLength(2);
+        expect(result.every((m) => typeof m.id === "string" && m.id.length > 0)).toBe(true);
+    });
+    it("should handle duplicates in input", () => {
+        const left = [];
+        const right = [
+            new AIMessage({ id: "1", content: "Hi there!" }),
+            new AIMessage({ id: "1", content: "Hi there again!" }),
+        ];
+        const result = messagesStateReducer(left, right);
+        expect(result).toHaveLength(1);
+        expect(result[0].id).toBe("1");
+        expect(result[0].content).toBe("Hi there again!");
+    });
+    it("should handle duplicates with remove", () => {
+        const left = [new AIMessage({ id: "1", content: "Hello!" })];
+        const right = [
+            new RemoveMessage({ id: "1" }),
+            new AIMessage({ id: "1", content: "Hi there!" }),
+            new AIMessage({ id: "1", content: "Hi there again!" }),
+        ];
+        const result = messagesStateReducer(left, right);
+        expect(result).toHaveLength(1);
+        expect(result[0].id).toBe("1");
+        expect(result[0].content).toBe("Hi there again!");
+    });
+    it("should remove message", () => {
+        const left = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = new RemoveMessage({ id: "2" });
+        const result = messagesStateReducer(left, right);
+        const expected = [new HumanMessage({ id: "1", content: "Hello" })];
+        expect(result).toEqual(expected);
+    });
+    it("should handle duplicate remove messages", () => {
+        const left = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = [
+            new RemoveMessage({ id: "2" }),
+            new RemoveMessage({ id: "2" }),
+        ];
+        const result = messagesStateReducer(left, right);
+        const expected = [new HumanMessage({ id: "1", content: "Hello" })];
+        expect(result).toEqual(expected);
+    });
+    it("should throw on removing nonexistent message", () => {
+        const left = [new HumanMessage({ id: "1", content: "Hello" })];
+        const right = new RemoveMessage({ id: "2" });
+        expect(() => messagesStateReducer(left, right)).toThrow("Attempting to delete a message with an ID that doesn't exist");
+    });
+    it("should handle mixed operations", () => {
+        const left = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        const right = [
+            new HumanMessage({ id: "1", content: "Updated hello" }),
+            new RemoveMessage({ id: "2" }),
+            new SystemMessage({ id: "3", content: "New message" }),
+        ];
+        const result = messagesStateReducer(left, right);
+        const expected = [
+            new HumanMessage({ id: "1", content: "Updated hello" }),
+            new SystemMessage({ id: "3", content: "New message" }),
+        ];
+        expect(result).toEqual(expected);
+    });
+    it("should handle empty inputs", () => {
+        expect(messagesStateReducer([], [])).toEqual([]);
+        expect(messagesStateReducer([], [new HumanMessage({ id: "1", content: "Hello" })])).toEqual([new HumanMessage({ id: "1", content: "Hello" })]);
+        expect(messagesStateReducer([new HumanMessage({ id: "1", content: "Hello" })], [])).toEqual([new HumanMessage({ id: "1", content: "Hello" })]);
+    });
+    it("should handle non-array inputs", () => {
+        const left = new HumanMessage({ id: "1", content: "Hello" });
+        const right = new AIMessage({ id: "2", content: "Hi there!" });
+        const result = messagesStateReducer(left, right);
+        const expected = [
+            new HumanMessage({ id: "1", content: "Hello" }),
+            new AIMessage({ id: "2", content: "Hi there!" }),
+        ];
+        expect(result).toEqual(expected);
+    });
+    it("should remove all messages", () => {
+        // simple removal
+        expect(messagesStateReducer([new HumanMessage("Hello"), new AIMessage("Hi there!")], [new RemoveMessage({ id: REMOVE_ALL_MESSAGES })])).toEqual([]);
+        // removal and update (i.e. overwriting)
+        expect(messagesStateReducer([new HumanMessage("Hello"), new AIMessage("Hi there!")], [
+            new RemoveMessage({ id: REMOVE_ALL_MESSAGES }),
+            new HumanMessage({ id: "1", content: "Updated hello" }),
+        ])).toEqual([new HumanMessage({ id: "1", content: "Updated hello" })]);
+        // test removing preceding messages in the right list
+        expect(messagesStateReducer([new HumanMessage("Hello"), new AIMessage("Hi there!")], [
+            new HumanMessage("Updated hello"),
+            new RemoveMessage({ id: REMOVE_ALL_MESSAGES }),
+            new HumanMessage({ id: "1", content: "Updated hi there" }),
+        ])).toEqual([new HumanMessage({ id: "1", content: "Updated hi there" })]);
+    });
+});
+describe("pushMessage", () => {
+    it("should throw on message without ID", () => {
+        const message = new AIMessage("No ID");
+        const config = { callbacks: [] };
+        expect(() => pushMessage(message, config)).toThrow("Message ID is required");
+    });
+    it("should handle message with ID", () => {
+        const message = new AIMessage({ id: "1", content: "With ID" });
+        const config = { callbacks: [] };
+        const result = pushMessage(message, config);
+        expect(result).toEqual(message);
+    });
+    it("should handle message with custom state key", () => {
+        const message = new AIMessage({ id: "1", content: "With ID" });
+        const config = {
+            callbacks: [],
+            configurable: {
+                __pregel_send: (messages) => {
+                    expect(messages).toEqual([["custom", message]]);
+                },
+            },
+        };
+        pushMessage(message, config, { stateKey: "custom" });
+    });
+    it("should push messages in graph", async () => {
+        const graph = new StateGraph(MessagesAnnotation)
+            .addNode("chat", (state, config) => {
+            expect(() => pushMessage(new AIMessage("No ID"), config)).toThrow();
+            pushMessage(new AIMessage({ id: "1", content: "First" }), config);
+            pushMessage(new HumanMessage({ id: "2", content: "Second" }), config);
+            pushMessage(new AIMessage({ id: "3", content: "Third" }), config);
+            return state;
+        })
+            .addEdge(START, "chat")
+            .compile();
+        const messages = [];
+        let values;
+        for await (const [event, chunk] of await graph.stream({ messages: [] }, { streamMode: ["messages", "values"] })) {
+            if (event === "values") {
+                values = chunk.messages;
+            }
+            else if (event === "messages") {
+                const [message] = chunk;
+                messages.push(message);
+            }
+        }
+        expect(values).toEqual(messages);
+    });
+});
+//# sourceMappingURL=message.test.js.map

package/dist/graph/message.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message.test.js","sourceRoot":"","sources":["../../src/graph/message.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,SAAS,EACT,YAAY,EACZ,aAAa,EACb,aAAa,GAEd,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,oBAAoB,EACpB,WAAW,EACX,mBAAmB,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAErE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG;YACf,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SACjD,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG;YACZ,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;YAChD,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;SAC1D,CAAC;QACF,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG;YACf,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;YAChD,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;SAC1D,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,CAAC,IAAI,SAAS,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,IAAI,GAAkB,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG;YACZ,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;YAChD,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC;SACvD,CAAC;QACF,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,IAAI,GAAG,CAAC,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG;YACZ,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC9B,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;YAChD,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC;SACvD,CAAC;QACF,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,IAAI,GAAG;YACX,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SACjD,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,IAAI,GAAG;YACX,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SACjD,CAAC;QACF,MAAM,KAAK,GAAG;YACZ,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC9B,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;SAC/B,CAAC;QACF,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,CACrD,8DAA8D,CAC/D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,IAAI,GAAG;YACX,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SACjD,CAAC;QACF,MAAM,KAAK,GAAG;YACZ,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;YACvD,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC9B,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;SACvD,CAAC;QACF,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG;YACf,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;YACvD,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;SACvD,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,oBAAoB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,CACJ,oBAAoB,CAClB,EAAE,EACF,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAClD,CACF,CAAC,OAAO,CAAC,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7D,MAAM,CACJ,oBAAoB,CAClB,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,EACjD,EAAE,CACH,CACF,CAAC,OAAO,CAAC,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG;YACf,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC/C,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SACjD,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,iBAAiB;QACjB,MAAM,CACJ,oBAAoB,CAClB,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC,EACvD,CAAC,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC,CAAC,CACjD,CACF,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEd,wCAAwC;QACxC,MAAM,CACJ,oBAAoB,CAClB,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC,EACvD;YACE,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC;YAC9C,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;SACxD,CACF,CACF,CAAC,OAAO,CAAC,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC;QAErE,qDAAqD;QACrD,MAAM,CACJ,oBAAoB,CAClB,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE,IAAI,SAAS,CAAC,WAAW,CAAC,CAAC,EACvD;YACE,IAAI,YAAY,CAAC,eAAe,CAAC;YACjC,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC;YAC9C,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC;SAC3D,CACF,CACF,CAAC,OAAO,CAAC,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAG,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;QACjC,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAChD,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,OAAO,GAAG,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,OAAO,GAAG,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG;YACb,SAAS,EAAE,EAAE;YACb,YAAY,EAAE;gBACZ,aAAa,EAAE,CAAC,QAAiC,EAAE,EAAE;oBACnD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;gBAClD,CAAC;aACF;SACF,CAAC;QACF,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,kBAAkB,CAAC;aAC7C,OAAO,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACjC,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAEpE,WAAW,CAAC,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAClE,WAAW,CAAC,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YACtE,WAAW,CAAC,IAAI,SAAS,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAElE,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;aACD,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;aACtB,OAAO,EAAE,CAAC;QAEb,MAAM,QAAQ,GAAkB,EAAE,CAAC;QACnC,IAAI,MAAiC,CAAC;QAEtC,IAAI,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,KAAK,CAAC,MAAM,CACnD,EAAE,QAAQ,EAAE,EAAE,EAAE,EAChB,EAAE,UAAU,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,CACvC,EAAE,CAAC;YACF,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACvB,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC;YAC1B,CAAC;iBAAM,IAAI,KAAK,KAAK,UAAU,EAAE,CAAC;gBAChC,MAAM,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC;gBACxB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/graph/messages_annotation.cjs

@@ -1,9 +1,11 @@
 "use strict";
 /* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MessagesAnnotation = void 0;
+exports.MessagesZodState = exports.MessagesAnnotation = void 0;
+const zod_1 = require("zod");
 const annotation_js_1 = require("./annotation.cjs");
 const message_js_1 = require("./message.cjs");
+const state_js_1 = require("./zod/state.cjs");
 /**
  * Prebuilt state annotation that combines returned messages.
  * Can handle standard messages and special modifiers like {@link RemoveMessage}
@@ -45,4 +47,55 @@ exports.MessagesAnnotation = annotation_js_1.Annotation.Root({
         default: () => [],
     }),
 });
+/**
+ * Prebuilt state object that uses Zod to combine returned messages.
+ * This utility is synonymous with the `MessagesAnnotation` annotation,
+ * but uses Zod as the way to express messages state.
+ *
+ * You can use import and use this prebuilt schema like this:
+ *
+ * @example
+ * ```ts
+ * import { MessagesZodState, StateGraph } from "@langchain/langgraph";
+ *
+ * const graph = new StateGraph(MessagesZodState)
+ *   .addNode(...)
+ *   ...
+ * ```
+ *
+ * Which is equivalent to initializing the schema object manually like this:
+ *
+ * @example
+ * ```ts
+ * import { z } from "zod";
+ * import type { BaseMessage, BaseMessageLike } from "@langchain/core/messages";
+ * import { StateGraph, messagesStateReducer } from "@langchain/langgraph";
+ * import "@langchain/langgraph/zod";
+ *
+ * const AgentState = z.object({
+ *   messages: z
+ *     .custom<BaseMessage[]>()
+ *     .default(() => [])
+ *     .langgraph.reducer(
+ *        messagesStateReducer,
+ *        z.custom<BaseMessageLike | BaseMessageLike[]>()
+ *     ),
+ * });
+ * const graph = new StateGraph(AgentState)
+ *   .addNode(...)
+ *   ...
+ * ```
+ */
+exports.MessagesZodState = zod_1.z.object({
+    messages: (0, state_js_1.withLangGraph)(zod_1.z.custom(), {
+        reducer: {
+            schema: zod_1.z.custom(),
+            fn: message_js_1.messagesStateReducer,
+        },
+        jsonSchemaExtra: {
+            langgraph_type: "messages",
+        },
+        default: () => [],
+    }),
+});
 //# sourceMappingURL=messages_annotation.js.map

package/dist/graph/messages_annotation.d.ts

@@ -1,4 +1,5 @@
 import { BaseMessage } from "@langchain/core/messages";
+import { z } from "zod";
 import { Messages } from "./message.js";
 /**
  * Prebuilt state annotation that combines returned messages.
@@ -38,3 +39,49 @@ import { Messages } from "./message.js";
 export declare const MessagesAnnotation: import("./annotation.js").AnnotationRoot<{
     messages: import("../web.js").BinaryOperatorAggregate<BaseMessage[], Messages>;
 }>;
+/**
+ * Prebuilt state object that uses Zod to combine returned messages.
+ * This utility is synonymous with the `MessagesAnnotation` annotation,
+ * but uses Zod as the way to express messages state.
+ *
+ * You can use import and use this prebuilt schema like this:
+ *
+ * @example
+ * ```ts
+ * import { MessagesZodState, StateGraph } from "@langchain/langgraph";
+ *
+ * const graph = new StateGraph(MessagesZodState)
+ *   .addNode(...)
+ *   ...
+ * ```
+ *
+ * Which is equivalent to initializing the schema object manually like this:
+ *
+ * @example
+ * ```ts
+ * import { z } from "zod";
+ * import type { BaseMessage, BaseMessageLike } from "@langchain/core/messages";
+ * import { StateGraph, messagesStateReducer } from "@langchain/langgraph";
+ * import "@langchain/langgraph/zod";
+ *
+ * const AgentState = z.object({
+ *   messages: z
+ *     .custom<BaseMessage[]>()
+ *     .default(() => [])
+ *     .langgraph.reducer(
+ *        messagesStateReducer,
+ *        z.custom<BaseMessageLike | BaseMessageLike[]>()
+ *     ),
+ * });
+ * const graph = new StateGraph(AgentState)
+ *   .addNode(...)
+ *   ...
+ * ```
+ */
+export declare const MessagesZodState: z.ZodObject<{
+    messages: z.ZodType<BaseMessage[], z.ZodTypeDef, Messages>;
+}, "strip", z.ZodTypeAny, {
+    messages: BaseMessage[];
+}, {
+    messages: Messages;
+}>;