@langchain/langgraph 0.2.62 → 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/io.mapCommand.test.cjs

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

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