@langchain/langgraph 0.2.62 → 0.2.63

package/dist/pregel/io.mapCommand.test.js

@@ -0,0 +1,149 @@
+import { describe, expect, it } from "@jest/globals";
+import { Command, Send } from "../constants.js";
+import { mapCommand } from "./io.js";
+import { InvalidUpdateError } from "../errors.js";
+describe("mapCommand", () => {
+    it("should handle Command with goto (string)", () => {
+        const cmd = new Command({
+            goto: "nextNode",
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            [
+                "00000000-0000-0000-0000-000000000000",
+                "branch:__start__:__self__:nextNode",
+                "__start__",
+            ],
+        ]);
+    });
+    it("should handle Command with goto (Send object)", () => {
+        const send = new Send("targetNode", { arg1: "value1" });
+        const cmd = new Command({
+            goto: send,
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
+        ]);
+    });
+    it("should handle Command with goto (array of strings and Send objects)", () => {
+        const send = new Send("targetNode", { arg1: "value1" });
+        const cmd = new Command({
+            goto: ["nextNode1", send, "nextNode2"],
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            [
+                "00000000-0000-0000-0000-000000000000",
+                "branch:__start__:__self__:nextNode1",
+                "__start__",
+            ],
+            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
+            [
+                "00000000-0000-0000-0000-000000000000",
+                "branch:__start__:__self__:nextNode2",
+                "__start__",
+            ],
+        ]);
+    });
+    it("should throw error for invalid goto value", () => {
+        const cmd = new Command({
+            // @ts-expect-error Testing invalid input
+            goto: { invalidType: true },
+        });
+        const pendingWrites = [];
+        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("In Command.send, expected Send or string, got object");
+    });
+    it("should handle Command with resume (single value)", () => {
+        const cmd = new Command({
+            resume: "resumeValue",
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
+        ]);
+    });
+    it("should handle Command with resume (object of task IDs)", () => {
+        // Using a valid UUID-like structure
+        const cmd = new Command({
+            resume: {
+                "123e4567-e89b-12d3-a456-426614174000": "resumeValue1",
+                "123e4567-e89b-12d3-a456-426614174001": "resumeValue2",
+            },
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            ["123e4567-e89b-12d3-a456-426614174000", "__resume__", ["resumeValue1"]],
+            ["123e4567-e89b-12d3-a456-426614174001", "__resume__", ["resumeValue2"]],
+        ]);
+    });
+    it("should handle Command with update (object)", () => {
+        const cmd = new Command({
+            update: {
+                channel1: "value1",
+                channel2: "value2",
+            },
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
+            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
+        ]);
+    });
+    it("should handle Command with update (array of tuples)", () => {
+        const cmd = new Command({
+            update: [
+                ["channel1", "value1"],
+                ["channel2", "value2"],
+            ],
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
+            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
+        ]);
+    });
+    it("should throw error for invalid update type", () => {
+        const cmd = new Command({
+            // @ts-expect-error Testing invalid input
+            update: "invalidUpdateType",
+        });
+        const pendingWrites = [];
+        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("Expected cmd.update to be a dict mapping channel names to update values");
+    });
+    it("should throw error for parent graph reference when none exists", () => {
+        const cmd = new Command({
+            graph: Command.PARENT,
+            goto: "nextNode",
+        });
+        const pendingWrites = [];
+        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow(InvalidUpdateError);
+        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("There is no parent graph.");
+    });
+    it("should handle multiple command attributes together", () => {
+        const cmd = new Command({
+            goto: "nextNode",
+            resume: "resumeValue",
+            update: { channel1: "value1" },
+        });
+        const pendingWrites = [];
+        const result = Array.from(mapCommand(cmd, pendingWrites));
+        expect(result).toEqual([
+            [
+                "00000000-0000-0000-0000-000000000000",
+                "branch:__start__:__self__:nextNode",
+                "__start__",
+            ],
+            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
+            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
+        ]);
+    });
+});
+//# sourceMappingURL=io.mapCommand.test.js.map

package/dist/pregel/io.mapCommand.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"io.mapCommand.test.js","sourceRoot":"","sources":["../../src/pregel/io.mapCommand.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,oCAAoC;gBACpC,WAAW;aACZ;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,gBAAgB,EAAE,IAAI,CAAC;SACjE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,EAAE,WAAW,CAAC;SACvC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,qCAAqC;gBACrC,WAAW;aACZ;YACD,CAAC,sCAAsC,EAAE,gBAAgB,EAAE,IAAI,CAAC;YAChE;gBACE,sCAAsC;gBACtC,qCAAqC;gBACrC,WAAW;aACZ;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,yCAAyC;YACzC,IAAI,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;SAC5B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,sDAAsD,CACvD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,YAAY,EAAE,aAAa,CAAC;SACtE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,oCAAoC;QACpC,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,sCAAsC,EAAE,cAAc;gBACtD,sCAAsC,EAAE,cAAc;aACvD;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;YACxE,CAAC,sCAAsC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;SACzE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,QAAQ,EAAE,QAAQ;aACnB;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;YAC9D,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,CAAC,UAAU,EAAE,QAAQ,CAAC;gBACtB,CAAC,UAAU,EAAE,QAAQ,CAAC;aACvB;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;YAC9D,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,yCAAyC;YACzC,MAAM,EAAE,mBAAmB;SAC5B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,yEAAyE,CAC1E,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,KAAK,EAAE,OAAO,CAAC,MAAM;YACrB,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,kBAAkB,CACnB,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,2BAA2B,CAC5B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,UAAU;YAChB,MAAM,EAAE,aAAa;YACrB,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE;SAC/B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,oCAAoC;gBACpC,WAAW;aACZ;YACD,CAAC,sCAAsC,EAAE,YAAY,EAAE,aAAa,CAAC;YACrE,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/messages.test.cjs

@@ -0,0 +1,351 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const messages_1 = require("@langchain/core/messages");
+const outputs_1 = require("@langchain/core/outputs");
+const messages_js_1 = require("./messages.cjs");
+const constants_js_1 = require("../constants.cjs");
+(0, globals_1.describe)("StreamMessagesHandler", () => {
+    (0, globals_1.describe)("constructor", () => {
+        (0, globals_1.it)("should properly initialize the handler", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            (0, globals_1.expect)(handler.name).toBe("StreamMessagesHandler");
+            (0, globals_1.expect)(handler.streamFn).toBe(streamFn);
+            (0, globals_1.expect)(handler.metadatas).toEqual({});
+            (0, globals_1.expect)(handler.seen).toEqual({});
+            (0, globals_1.expect)(handler.emittedChatModelRunIds).toEqual({});
+            (0, globals_1.expect)(handler.stableMessageIdMap).toEqual({});
+            (0, globals_1.expect)(handler.lc_prefer_streaming).toBe(true);
+        });
+    });
+    (0, globals_1.describe)("_emit", () => {
+        (0, globals_1.it)("should emit a message with metadata", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const meta = [
+                ["ns1", "ns2"],
+                { name: "test", tags: [] },
+            ];
+            const message = new messages_1.AIMessage({ content: "Hello world" });
+            const runId = "run-123";
+            handler._emit(meta, message, runId);
+            (0, globals_1.expect)(streamFn).toHaveBeenCalledWith([
+                ["ns1", "ns2"],
+                "messages",
+                [message, { name: "test", tags: [] }],
+            ]);
+            // Should store the message in seen if it has an ID
+            message.id = "msg-123";
+            handler._emit(meta, message, runId);
+            (0, globals_1.expect)(handler.seen["msg-123"]).toBe(message);
+        });
+        (0, globals_1.it)("should deduplicate messages when dedupe=true and message has been seen", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const meta = [
+                ["ns1"],
+                { name: "test" },
+            ];
+            const message = new messages_1.AIMessage({ content: "Hello world", id: "msg-123" });
+            const runId = "run-123";
+            // First emit should work
+            handler._emit(meta, message, runId);
+            (0, globals_1.expect)(streamFn).toHaveBeenCalledTimes(1);
+            // Second emit with same ID and dedupe=true should be ignored
+            streamFn.mockClear();
+            handler._emit(meta, message, runId, true);
+            (0, globals_1.expect)(streamFn).not.toHaveBeenCalled();
+        });
+        (0, globals_1.it)("should assign proper ID to tool messages", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const meta = [
+                ["ns1"],
+                { name: "test" },
+            ];
+            const toolMessage = new messages_1.ToolMessage({
+                content: "Tool result",
+                tool_call_id: "tc-123",
+            });
+            const runId = "run-456";
+            handler._emit(meta, toolMessage, runId);
+            // Should assign an ID based on the tool call ID
+            (0, globals_1.expect)(toolMessage.id).toBe(`run-${runId}-tool-tc-123`);
+            (0, globals_1.expect)(streamFn).toHaveBeenCalled();
+        });
+        (0, globals_1.it)("should maintain stable message IDs for the same run", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const meta = [
+                ["ns1"],
+                { name: "test" },
+            ];
+            const runId = "run-789";
+            // First message with auto-generated ID
+            const message1 = new messages_1.AIMessage({ content: "First chunk" });
+            handler._emit(meta, message1, runId);
+            const stableId = message1.id;
+            // Second message with no ID should get the same stable ID
+            const message2 = new messages_1.AIMessage({ content: "Second chunk" });
+            handler._emit(meta, message2, runId);
+            (0, globals_1.expect)(message2.id).toBe(stableId);
+            (0, globals_1.expect)(handler.stableMessageIdMap[runId]).toBe(stableId);
+        });
+    });
+    (0, globals_1.describe)("handleChatModelStart", () => {
+        (0, globals_1.it)("should store metadata when provided", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "run-123";
+            const metadata = {
+                langgraph_checkpoint_ns: "ns1|ns2",
+                other_meta: "value",
+            };
+            handler.handleChatModelStart({}, // llm
+            [], // messages
+            runId, undefined, // parentRunId
+            {}, // extraParams
+            [], // tags
+            metadata, // metadata
+            "ModelName" // name
+            );
+            (0, globals_1.expect)(handler.metadatas[runId]).toEqual([
+                ["ns1", "ns2"],
+                { tags: [], name: "ModelName", ...metadata },
+            ]);
+        });
+        (0, globals_1.it)("should not store metadata when TAG_NOSTREAM is present", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "run-123";
+            const metadata = {
+                langgraph_checkpoint_ns: "ns1|ns2",
+            };
+            handler.handleChatModelStart({}, [], runId, undefined, {}, [constants_js_1.TAG_NOSTREAM], // nostream tag
+            metadata, "ModelName");
+            // Should not store metadata due to TAG_NOSTREAM
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+    });
+    (0, globals_1.describe)("handleLLMNewToken", () => {
+        (0, globals_1.it)("should emit message chunk when metadata exists", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "run-123";
+            handler.metadatas[runId] = [["ns1", "ns2"], { name: "test" }];
+            // Spy on _emit
+            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, // idx
+            runId);
+            // Should mark run as emitted
+            (0, globals_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
+            // Should emit AIMessageChunk when no chunk is provided
+            (0, globals_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], globals_1.expect.any(messages_1.AIMessageChunk), runId);
+            (0, globals_1.expect)(emitSpy.mock.calls[0][1].content).toBe("token");
+        });
+        (0, globals_1.it)("should emit provided chunk when available", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "run-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Spy on _emit
+            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            // Create a chunk
+            const chunk = new outputs_1.ChatGenerationChunk({
+                message: new messages_1.AIMessageChunk({ content: "chunk content" }),
+                text: "chunk content", // Add text field to satisfy ChatGenerationChunkFields
+            });
+            handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, runId, undefined, undefined, { chunk } // provide the chunk
+            );
+            // Should emit the chunk's message
+            (0, globals_1.expect)(emitSpy).toHaveBeenCalledWith(handler.metadatas[runId], chunk.message, runId);
+        });
+        (0, globals_1.it)("should not emit when metadata is missing", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "run-123";
+            // No metadata for this runId
+            // Spy on _emit
+            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            handler.handleLLMNewToken("token", { prompt: 0, completion: 0 }, runId);
+            // Should mark run as emitted
+            (0, globals_1.expect)(handler.emittedChatModelRunIds[runId]).toBe(true);
+            // But should not call _emit
+            (0, globals_1.expect)(emitSpy).not.toHaveBeenCalled();
+        });
+    });
+    (0, globals_1.describe)("handleLLMEnd", () => {
+        (0, globals_1.it)("should emit message from non-streaming run", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "run-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Not marked as emitted yet
+            // Mock _emit directly instead of spying
+            handler._emit = globals_1.jest.fn();
+            const message = new messages_1.AIMessage({ content: "final result" });
+            handler.handleLLMEnd({
+                generations: [[{ text: "test output", message }]],
+            }, runId);
+            // Should emit the message with dedupe=true
+            (0, globals_1.expect)(handler._emit).toHaveBeenCalledWith(globals_1.expect.anything(), globals_1.expect.objectContaining({ content: "final result" }), runId, true);
+            // Should clean up
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+        (0, globals_1.it)("should not emit for streaming runs that already emitted", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "run-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Mark as already emitted
+            handler.emittedChatModelRunIds[runId] = true;
+            // Mock _emit directly
+            handler._emit = globals_1.jest.fn();
+            handler.handleLLMEnd({
+                generations: [
+                    [
+                        {
+                            text: "test output",
+                            message: new messages_1.AIMessage({ content: "result" }),
+                        },
+                    ],
+                ],
+            }, runId);
+            // Should not emit anything
+            (0, globals_1.expect)(handler._emit).not.toHaveBeenCalled();
+            // Should clean up metadata
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+    });
+    (0, globals_1.describe)("handleLLMError", () => {
+        (0, globals_1.it)("should clean up metadata on error", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "run-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            handler.handleLLMError(new Error("Test error"), runId);
+            // Should clean up metadata
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+    });
+    (0, globals_1.describe)("handleChainStart", () => {
+        (0, globals_1.it)("should store metadata for matching node name", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "chain-123";
+            const metadata = {
+                langgraph_checkpoint_ns: "ns1|ns2",
+                langgraph_node: "NodeName", // Matches name parameter
+            };
+            handler.handleChainStart({}, {}, runId, undefined, [], metadata, undefined, "NodeName" // Name matches langgraph_node
+            );
+            (0, globals_1.expect)(handler.metadatas[runId]).toEqual([
+                ["ns1", "ns2"],
+                { tags: [], name: "NodeName", ...metadata },
+            ]);
+        });
+        (0, globals_1.it)("should not store metadata when node name doesn't match", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "chain-123";
+            const metadata = {
+                langgraph_checkpoint_ns: "ns1|ns2",
+                langgraph_node: "NodeName", // Doesn't match name parameter
+            };
+            handler.handleChainStart({}, {}, runId, undefined, [], metadata, undefined, "DifferentName" // Different from langgraph_node
+            );
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+        (0, globals_1.it)("should not store metadata when TAG_HIDDEN is present", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "chain-123";
+            const metadata = {
+                langgraph_checkpoint_ns: "ns1|ns2",
+                langgraph_node: "NodeName",
+            };
+            handler.handleChainStart({}, {}, runId, undefined, [constants_js_1.TAG_HIDDEN], // Hidden tag
+            metadata, undefined, "NodeName");
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+    });
+    (0, globals_1.describe)("handleChainEnd", () => {
+        (0, globals_1.it)("should emit a single message output", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "chain-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Mock _emit directly
+            handler._emit = globals_1.jest.fn();
+            const message = new messages_1.AIMessage({ content: "chain result" });
+            handler.handleChainEnd(message, runId);
+            // Should emit the message with dedupe=true
+            (0, globals_1.expect)(handler._emit).toHaveBeenCalledWith(globals_1.expect.anything(), globals_1.expect.objectContaining({ content: "chain result" }), runId, true);
+            // Should clean up
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+        (0, globals_1.it)("should emit messages from an array output", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "chain-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Mock _emit directly
+            handler._emit = globals_1.jest.fn();
+            const message1 = new messages_1.AIMessage({ content: "result 1" });
+            const message2 = new messages_1.AIMessage({ content: "result 2" });
+            const notAMessage = "not a message";
+            handler.handleChainEnd([message1, message2, notAMessage], runId);
+            // Should emit both messages
+            (0, globals_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
+            // Verify calls in a way that's less brittle
+            const callArgs = handler._emit.mock.calls;
+            const emittedContents = callArgs.map((args) => args[1].content);
+            (0, globals_1.expect)(emittedContents).toContain("result 1");
+            (0, globals_1.expect)(emittedContents).toContain("result 2");
+            // Should clean up
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+        (0, globals_1.it)("should emit messages from object output properties", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "chain-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            // Mock _emit directly
+            handler._emit = globals_1.jest.fn();
+            const message = new messages_1.AIMessage({ content: "direct result" });
+            const arrayMessage = new messages_1.AIMessage({ content: "array result" });
+            handler.handleChainEnd({
+                directMessage: message,
+                arrayMessages: [arrayMessage, "not a message"],
+                otherProp: "something else",
+            }, runId);
+            // Should emit both messages
+            (0, globals_1.expect)(handler._emit).toHaveBeenCalledTimes(2);
+            // Verify calls in a way that's less brittle
+            const callArgs = handler._emit.mock.calls;
+            const emittedContents = callArgs.map((args) => args[1].content);
+            (0, globals_1.expect)(emittedContents).toContain("direct result");
+            (0, globals_1.expect)(emittedContents).toContain("array result");
+            // Should clean up
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+        (0, globals_1.it)("should do nothing when metadata is missing", () => {
+            const streamFn = globals_1.jest.fn();
+            const handler = new messages_js_1.StreamMessagesHandler(streamFn);
+            const runId = "chain-123";
+            // No metadata for this runId
+            // Spy on _emit
+            const emitSpy = globals_1.jest.spyOn(handler, "_emit");
+            const message = new messages_1.AIMessage({ content: "result" });
+            handler.handleChainEnd(message, runId);
+            // Should not emit anything
+            (0, globals_1.expect)(emitSpy).not.toHaveBeenCalled();
+        });
+    });
+    (0, globals_1.describe)("handleChainError", () => {
+        (0, globals_1.it)("should clean up metadata on error", () => {
+            const handler = new messages_js_1.StreamMessagesHandler(globals_1.jest.fn());
+            const runId = "chain-123";
+            handler.metadatas[runId] = [["ns1"], { name: "test" }];
+            handler.handleChainError(new Error("Test error"), runId);
+            // Should clean up metadata
+            (0, globals_1.expect)(handler.metadatas[runId]).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=messages.test.js.map

package/dist/pregel/messages.test.d.ts

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