@langchain/langgraph 0.2.61 → 0.2.63

package/dist/pregel/debug.test.cjs

@@ -0,0 +1,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const debug_js_1 = require("./debug.cjs");
+const last_value_js_1 = require("../channels/last_value.cjs");
+const errors_js_1 = require("../errors.cjs");
+(0, globals_1.describe)("wrap", () => {
+    (0, globals_1.it)("should wrap text with color codes", () => {
+        const color = {
+            start: "\x1b[34m",
+            end: "\x1b[0m",
+        };
+        const text = "test text";
+        const result = (0, debug_js_1.wrap)(color, text);
+        (0, globals_1.expect)(result).toBe(`${color.start}${text}${color.end}`);
+    });
+});
+(0, globals_1.describe)("_readChannels", () => {
+    (0, globals_1.it)("should read values from channels", () => {
+        const channels = {
+            channel1: new last_value_js_1.LastValue(),
+            channel2: new last_value_js_1.LastValue(),
+        };
+        // Update channels with values
+        channels.channel1.update(["value1"]);
+        channels.channel2.update(["42"]);
+        const results = Array.from((0, debug_js_1._readChannels)(channels));
+        (0, globals_1.expect)(results).toEqual([
+            ["channel1", "value1"],
+            ["channel2", "42"],
+        ]);
+    });
+    (0, globals_1.it)("should skip empty channels", () => {
+        const mockEmptyChannel = {
+            lc_graph_name: "MockChannel",
+            lg_is_channel: true,
+            ValueType: "",
+            UpdateType: [],
+            get: globals_1.jest.fn().mockImplementation(() => {
+                throw new errors_js_1.EmptyChannelError("Empty channel");
+            }),
+            update: globals_1.jest.fn().mockReturnValue(true),
+            checkpoint: globals_1.jest.fn(),
+            fromCheckpoint: globals_1.jest
+                .fn()
+                .mockReturnThis(),
+            consume: globals_1.jest.fn().mockReturnValue(false),
+        };
+        const channels = {
+            channel1: new last_value_js_1.LastValue(),
+            emptyChannel: mockEmptyChannel,
+        };
+        // Update channel with value
+        channels.channel1.update(["value1"]);
+        const results = Array.from((0, debug_js_1._readChannels)(channels));
+        (0, globals_1.expect)(results).toEqual([["channel1", "value1"]]);
+    });
+    (0, globals_1.it)("should propagate non-empty channel errors", () => {
+        const mockErrorChannel = {
+            lc_graph_name: "MockChannel",
+            lg_is_channel: true,
+            ValueType: "",
+            UpdateType: [],
+            get: globals_1.jest.fn().mockImplementation(() => {
+                throw new Error("Other error");
+            }),
+            update: globals_1.jest.fn().mockReturnValue(true),
+            checkpoint: globals_1.jest.fn(),
+            fromCheckpoint: globals_1.jest
+                .fn()
+                .mockReturnThis(),
+            consume: globals_1.jest.fn().mockReturnValue(false),
+        };
+        const channels = {
+            channel1: new last_value_js_1.LastValue(),
+            errorChannel: mockErrorChannel,
+        };
+        channels.channel1.update(["value1"]);
+        (0, globals_1.expect)(() => Array.from((0, debug_js_1._readChannels)(channels))).toThrow("Other error");
+    });
+});
+(0, globals_1.describe)("tasksWithWrites", () => {
+    (0, globals_1.it)("should return task descriptions with no writes", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [];
+        const result = (0, debug_js_1.tasksWithWrites)(tasks, pendingWrites);
+        (0, globals_1.expect)(result).toEqual([
+            { id: "task1", name: "Task 1", path: ["PULL", "Task 1"], interrupts: [] },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    (0, globals_1.it)("should include error information", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [
+            ["task1", "__error__", { message: "Test error" }],
+        ];
+        const result = (0, debug_js_1.tasksWithWrites)(tasks, pendingWrites);
+        (0, globals_1.expect)(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                error: { message: "Test error" },
+                interrupts: [],
+            },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    (0, globals_1.it)("should include state information", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [];
+        const states = {
+            task1: { configurable: { key: "value" } },
+        };
+        const result = (0, debug_js_1.tasksWithWrites)(tasks, pendingWrites, states);
+        (0, globals_1.expect)(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+                state: { configurable: { key: "value" } },
+            },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    (0, globals_1.it)("should include interrupts", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [
+            ["task1", "__interrupt__", { value: "Interrupted", when: "during" }],
+        ];
+        const result = (0, debug_js_1.tasksWithWrites)(tasks, pendingWrites);
+        (0, globals_1.expect)(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [{ value: "Interrupted", when: "during" }],
+            },
+        ]);
+    });
+});
+//# sourceMappingURL=debug.test.js.map

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

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

package/dist/pregel/debug.test.js

@@ -0,0 +1,187 @@
+import { describe, expect, it, jest } from "@jest/globals";
+import { wrap, tasksWithWrites, _readChannels } from "./debug.js";
+import { LastValue } from "../channels/last_value.js";
+import { EmptyChannelError } from "../errors.js";
+describe("wrap", () => {
+    it("should wrap text with color codes", () => {
+        const color = {
+            start: "\x1b[34m",
+            end: "\x1b[0m",
+        };
+        const text = "test text";
+        const result = wrap(color, text);
+        expect(result).toBe(`${color.start}${text}${color.end}`);
+    });
+});
+describe("_readChannels", () => {
+    it("should read values from channels", () => {
+        const channels = {
+            channel1: new LastValue(),
+            channel2: new LastValue(),
+        };
+        // Update channels with values
+        channels.channel1.update(["value1"]);
+        channels.channel2.update(["42"]);
+        const results = Array.from(_readChannels(channels));
+        expect(results).toEqual([
+            ["channel1", "value1"],
+            ["channel2", "42"],
+        ]);
+    });
+    it("should skip empty channels", () => {
+        const mockEmptyChannel = {
+            lc_graph_name: "MockChannel",
+            lg_is_channel: true,
+            ValueType: "",
+            UpdateType: [],
+            get: jest.fn().mockImplementation(() => {
+                throw new EmptyChannelError("Empty channel");
+            }),
+            update: jest.fn().mockReturnValue(true),
+            checkpoint: jest.fn(),
+            fromCheckpoint: jest
+                .fn()
+                .mockReturnThis(),
+            consume: jest.fn().mockReturnValue(false),
+        };
+        const channels = {
+            channel1: new LastValue(),
+            emptyChannel: mockEmptyChannel,
+        };
+        // Update channel with value
+        channels.channel1.update(["value1"]);
+        const results = Array.from(_readChannels(channels));
+        expect(results).toEqual([["channel1", "value1"]]);
+    });
+    it("should propagate non-empty channel errors", () => {
+        const mockErrorChannel = {
+            lc_graph_name: "MockChannel",
+            lg_is_channel: true,
+            ValueType: "",
+            UpdateType: [],
+            get: jest.fn().mockImplementation(() => {
+                throw new Error("Other error");
+            }),
+            update: jest.fn().mockReturnValue(true),
+            checkpoint: jest.fn(),
+            fromCheckpoint: jest
+                .fn()
+                .mockReturnThis(),
+            consume: jest.fn().mockReturnValue(false),
+        };
+        const channels = {
+            channel1: new LastValue(),
+            errorChannel: mockErrorChannel,
+        };
+        channels.channel1.update(["value1"]);
+        expect(() => Array.from(_readChannels(channels))).toThrow("Other error");
+    });
+});
+describe("tasksWithWrites", () => {
+    it("should return task descriptions with no writes", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [];
+        const result = tasksWithWrites(tasks, pendingWrites);
+        expect(result).toEqual([
+            { id: "task1", name: "Task 1", path: ["PULL", "Task 1"], interrupts: [] },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    it("should include error information", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [
+            ["task1", "__error__", { message: "Test error" }],
+        ];
+        const result = tasksWithWrites(tasks, pendingWrites);
+        expect(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                error: { message: "Test error" },
+                interrupts: [],
+            },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    it("should include state information", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+            {
+                id: "task2",
+                name: "Task 2",
+                path: ["PULL", "Task 2"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [];
+        const states = {
+            task1: { configurable: { key: "value" } },
+        };
+        const result = tasksWithWrites(tasks, pendingWrites, states);
+        expect(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+                state: { configurable: { key: "value" } },
+            },
+            { id: "task2", name: "Task 2", path: ["PULL", "Task 2"], interrupts: [] },
+        ]);
+    });
+    it("should include interrupts", () => {
+        const tasks = [
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [],
+            },
+        ];
+        const pendingWrites = [
+            ["task1", "__interrupt__", { value: "Interrupted", when: "during" }],
+        ];
+        const result = tasksWithWrites(tasks, pendingWrites);
+        expect(result).toEqual([
+            {
+                id: "task1",
+                name: "Task 1",
+                path: ["PULL", "Task 1"],
+                interrupts: [{ value: "Interrupted", when: "during" }],
+            },
+        ]);
+    });
+});
+//# sourceMappingURL=debug.test.js.map

package/dist/pregel/debug.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.test.js","sourceRoot":"","sources":["../../src/pregel/debug.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAElE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEjD,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG;YACZ,KAAK,EAAE,UAAU;YACjB,GAAG,EAAE,SAAS;SACf,CAAC;QAEF,MAAM,IAAI,GAAG,WAAW,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAEjC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,GAAG,IAAI,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,IAAI,SAAS,EAAU;YACjC,QAAQ,EAAE,IAAI,SAAS,EAAU;SAClC,CAAC;QAEF,8BAA8B;QAC9B,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACrC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAEjC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEpD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;YACtB,CAAC,UAAU,EAAE,QAAQ,CAAC;YACtB,CAAC,UAAU,EAAE,IAAI,CAAC;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,gBAAgB,GAAwB;YAC5C,aAAa,EAAE,aAAa;YAC5B,aAAa,EAAE,IAAI;YACnB,SAAS,EAAE,EAAY;YACvB,UAAU,EAAE,EAAe;YAC3B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAgB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACnD,MAAM,IAAI,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC/C,CAAC,CAAC;YACF,MAAM,EAAE,IAAI,CAAC,EAAE,EAAkC,CAAC,eAAe,CAAC,IAAI,CAAC;YACvE,UAAU,EAAE,IAAI,CAAC,EAAE,EAAiB;YACpC,cAAc,EAAE,IAAI;iBACjB,EAAE,EAAiD;iBACnD,cAAc,EAAE;YACnB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAiB,CAAC,eAAe,CAAC,KAAK,CAAC;SACzD,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,IAAI,SAAS,EAAU;YACjC,YAAY,EAAE,gBAAgB;SAC/B,CAAC;QAEF,4BAA4B;QAC5B,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAErC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEpD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,gBAAgB,GAAwB;YAC5C,aAAa,EAAE,aAAa;YAC5B,aAAa,EAAE,IAAI;YACnB,SAAS,EAAE,EAAY;YACvB,UAAU,EAAE,EAAe;YAC3B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAgB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACnD,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;YACjC,CAAC,CAAC;YACF,MAAM,EAAE,IAAI,CAAC,EAAE,EAAkC,CAAC,eAAe,CAAC,IAAI,CAAC;YACvE,UAAU,EAAE,IAAI,CAAC,EAAE,EAAiB;YACpC,cAAc,EAAE,IAAI;iBACjB,EAAE,EAAiD;iBACnD,cAAc,EAAE;YACnB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAiB,CAAC,eAAe,CAAC,KAAK,CAAC;SACzD,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,IAAI,SAAS,EAAU;YACjC,YAAY,EAAE,gBAAgB;SAC/B,CAAC;QAEF,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAErC,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,KAAK,GAAG;YACZ;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;SACF,CAAC;QAEF,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;YACzE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;SAC1E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG;YACZ;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;SACF,CAAC;QAEF,MAAM,aAAa,GAAqC;YACtD,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;SAClD,CAAC;QAEF,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACxB,KAAK,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;gBAChC,UAAU,EAAE,EAAE;aACf;YACD,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;SAC1E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG;YACZ;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;SACF,CAAC;QAEF,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG;YACb,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;SAC1C,CAAC;QAEF,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACxB,UAAU,EAAE,EAAE;gBACd,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;aAC1C;YACD,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;SAC1E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,KAAK,GAAG;YACZ;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAqB;gBAC5C,UAAU,EAAE,EAAE;aACf;SACF,CAAC;QAEF,MAAM,aAAa,GAAqC;YACtD,CAAC,OAAO,EAAE,eAAe,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;SACrE,CAAC;QAEF,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACxB,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;aACvD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/index.cjs

@@ -1176,21 +1176,29 @@ class Pregel extends runnables_1.Runnable {
         // There is currently no way in _streamIterator to determine whether this was
         // set by by ensureConfig or manually by the user, so we specify the bound value here
         // and override if it is passed as an explicit param in `options`.
+        const abortController = new AbortController();
         const config = {
             recursionLimit: this.config?.recursionLimit,
             ...options,
+            signal: options?.signal
+                ? (0, index_js_1.combineAbortSignals)(options.signal, abortController.signal)
+                : abortController.signal,
         };
-        return super.stream(input, config);
+        return new stream_js_1.IterableReadableStreamWithAbortSignal(await super.stream(input, config), abortController);
     }
     streamEvents(input, options, streamOptions) {
-        // Similar to `stream`, we need to pass the `config.callbacks` here,
-        // otherwise the user-provided callback will get lost in `ensureLangGraphConfig`.
+        const abortController = new AbortController();
         const config = {
             recursionLimit: this.config?.recursionLimit,
+            // Similar to `stream`, we need to pass the `config.callbacks` here,
+            // otherwise the user-provided callback will get lost in `ensureLangGraphConfig`.
             callbacks: this.config?.callbacks,
             ...options,
+            signal: options?.signal
+                ? (0, index_js_1.combineAbortSignals)(options.signal, abortController.signal)
+                : abortController.signal,
         };
-        return super.streamEvents(input, config, streamOptions);
+        return new stream_js_1.IterableReadableStreamWithAbortSignal(super.streamEvents(input, config, streamOptions), abortController);
     }
     /**
      * Prepares channel specifications and managed values for graph execution.

package/dist/pregel/index.d.ts

@@ -437,6 +437,9 @@ export declare class Pregel<Nodes extends StrRecord<string, PregelNode>, Channel
      * @returns An async iterable stream of graph state updates
      */
     stream(input: InputType | Command | null, options?: Partial<PregelOptions<Nodes, Channels, ConfigurableFieldType>>): Promise<IterableReadableStream<PregelOutputType>>;
+    /**
+     * @inheritdoc
+     */
     streamEvents(input: InputType | Command | null, options: Partial<PregelOptions<Nodes, Channels, ConfigurableFieldType>> & {
         version: "v1" | "v2";
     }, streamOptions?: StreamEventsOptions): IterableReadableStream<StreamEvent>;

package/dist/pregel/index.js

@@ -10,7 +10,7 @@ import { ChannelWrite, PASSTHROUGH } from "./write.js";
 import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, CONFIG_KEY_TASK_ID, ERROR, INPUT, INTERRUPT, PUSH, CHECKPOINT_NAMESPACE_SEPARATOR, CHECKPOINT_NAMESPACE_END, CONFIG_KEY_STREAM, NULL_TASK_ID, COPY, END, CONFIG_KEY_NODE_FINISHED, } from "../constants.js";
 import { GraphRecursionError, GraphValueError, InvalidUpdateError, } from "../errors.js";
 import { _prepareNextTasks, _localRead, _applyWrites, } from "./algo.js";
-import { _coerceToDict, getNewChannelVersions, patchCheckpointMap, } from "./utils/index.js";
+import { _coerceToDict, combineAbortSignals, getNewChannelVersions, patchCheckpointMap, } from "./utils/index.js";
 import { findSubgraphPregel } from "./utils/subgraph.js";
 import { PregelLoop } from "./loop.js";
 import { ChannelKeyPlaceholder, isConfiguredManagedValue, ManagedValueMapping, NoopManagedValue, } from "../managed/base.js";
@@ -18,7 +18,7 @@ import { gatherIterator, patchConfigurable } from "../utils.js";
 import { ensureLangGraphConfig, recastCheckpointNamespace, } from "./utils/config.js";
 import { StreamMessagesHandler } from "./messages.js";
 import { PregelRunner } from "./runner.js";
-import { IterableReadableWritableStream } from "./stream.js";
+import { IterableReadableStreamWithAbortSignal, IterableReadableWritableStream, } from "./stream.js";
 function isString(value) {
     return typeof value === "string";
 }
@@ -1172,21 +1172,29 @@ export class Pregel extends Runnable {
         // There is currently no way in _streamIterator to determine whether this was
         // set by by ensureConfig or manually by the user, so we specify the bound value here
         // and override if it is passed as an explicit param in `options`.
+        const abortController = new AbortController();
         const config = {
             recursionLimit: this.config?.recursionLimit,
             ...options,
+            signal: options?.signal
+                ? combineAbortSignals(options.signal, abortController.signal)
+                : abortController.signal,
         };
-        return super.stream(input, config);
+        return new IterableReadableStreamWithAbortSignal(await super.stream(input, config), abortController);
     }
     streamEvents(input, options, streamOptions) {
-        // Similar to `stream`, we need to pass the `config.callbacks` here,
-        // otherwise the user-provided callback will get lost in `ensureLangGraphConfig`.
+        const abortController = new AbortController();
         const config = {
             recursionLimit: this.config?.recursionLimit,
+            // Similar to `stream`, we need to pass the `config.callbacks` here,
+            // otherwise the user-provided callback will get lost in `ensureLangGraphConfig`.
             callbacks: this.config?.callbacks,
             ...options,
+            signal: options?.signal
+                ? combineAbortSignals(options.signal, abortController.signal)
+                : abortController.signal,
         };
-        return super.streamEvents(input, config, streamOptions);
+        return new IterableReadableStreamWithAbortSignal(super.streamEvents(input, config, streamOptions), abortController);
     }
     /**
      * Prepares channel specifications and managed values for graph execution.