@langchain/langgraph 0.2.62 → 0.2.63

package/dist/pregel/read.test.js

@@ -0,0 +1,192 @@
+import { describe, expect, it, jest } from "@jest/globals";
+import { CONFIG_KEY_READ } from "../constants.js";
+import { LastValue } from "../channels/last_value.js";
+import { ChannelRead, PregelNode } from "./read.js";
+import { ChannelWrite } from "./write.js";
+describe("ChannelRead", () => {
+    it("should read a single channel value", async () => {
+        // Setup mock read function
+        const mockRead = jest
+            .fn()
+            .mockImplementation((channel) => {
+            if (channel === "test_channel") {
+                return "test_value";
+            }
+            return null;
+        });
+        const config = {
+            configurable: {
+                [CONFIG_KEY_READ]: mockRead,
+            },
+        };
+        // Create channel read
+        const channelRead = new ChannelRead("test_channel");
+        // Run the channel read with our config
+        const result = await channelRead.invoke(null, config);
+        // Verify results
+        expect(result).toBe("test_value");
+    });
+    it("should read multiple channel values", async () => {
+        // Setup mock read function
+        const mockRead = jest
+            .fn()
+            .mockImplementation((channels) => {
+            if (Array.isArray(channels)) {
+                return {
+                    channel1: "value1",
+                    channel2: "value2",
+                };
+            }
+            return null;
+        });
+        const config = {
+            configurable: {
+                [CONFIG_KEY_READ]: mockRead,
+            },
+        };
+        // Create channel read for multiple channels
+        const channelRead = new ChannelRead(["channel1", "channel2"]);
+        // Run the channel read with our config
+        const result = await channelRead.invoke(null, config);
+        // Verify results
+        expect(result).toEqual({
+            channel1: "value1",
+            channel2: "value2",
+        });
+    });
+    it("should apply a mapper function to the channel value", async () => {
+        // Setup mock read function
+        const mockRead = jest.fn().mockImplementation(() => "test_value");
+        const config = {
+            configurable: {
+                [CONFIG_KEY_READ]: mockRead,
+            },
+        };
+        // Create mapper function
+        const mapper = (value) => `mapped_${value}`;
+        // Create channel read with mapper
+        const channelRead = new ChannelRead("test_channel", mapper);
+        // Run the channel read with our config
+        const result = await channelRead.invoke(null, config);
+        // Verify results
+        expect(result).toBe("mapped_test_value");
+    });
+    it("should throw an error if no read function is configured", async () => {
+        // Create channel read without configuring a read function
+        const channelRead = new ChannelRead("test_channel");
+        const config = {};
+        // Run the channel read with empty config
+        await expect(channelRead.invoke(null, config)).rejects.toThrow("not configured with a read function");
+    });
+});
+describe("PregelNode", () => {
+    it("should create a node that subscribes to channels", () => {
+        const node = new PregelNode({
+            channels: ["input", "context"],
+            triggers: ["input"],
+        });
+        expect(node.channels).toEqual(["input", "context"]);
+        expect(node.triggers).toEqual(["input"]);
+    });
+    it("should chain with ChannelWrite using pipe", () => {
+        const node = new PregelNode({
+            channels: ["input"],
+            triggers: ["input"],
+        });
+        const write = new ChannelWrite([
+            { channel: "output", value: "test_output" },
+        ]);
+        const pipeResult = node.pipe(write);
+        expect(pipeResult.writers).toHaveLength(1);
+        expect(pipeResult.writers[0]).toBe(write);
+    });
+    it("should combine multiple consecutive ChannelWrite instances", () => {
+        const node = new PregelNode({
+            channels: ["input"],
+            triggers: ["input"],
+        });
+        const write1 = new ChannelWrite([{ channel: "output1", value: "value1" }]);
+        const write2 = new ChannelWrite([{ channel: "output2", value: "value2" }]);
+        // Chain two writes
+        const pipeResult = node.pipe(write1).pipe(write2);
+        // Get optimized writers
+        const optimizedWriters = pipeResult.getWriters();
+        // Should be combined into a single ChannelWrite
+        expect(optimizedWriters).toHaveLength(1);
+        expect(optimizedWriters[0]).toBeInstanceOf(ChannelWrite);
+        expect(optimizedWriters[0].writes).toHaveLength(2);
+    });
+    it("should join additional channels", () => {
+        const node = new PregelNode({
+            channels: { input: "input", context: "context" },
+            triggers: ["input"],
+        });
+        const joinedNode = node.join(["history"]);
+        expect(joinedNode.channels).toEqual({
+            input: "input",
+            context: "context",
+            history: "history",
+        });
+    });
+});
+describe("Integrated Channel Read and Write", () => {
+    it("should perform direct channel operations", async () => {
+        // Use direct channel operations rather than depending on invoke
+        // Setup test environment with real channels
+        const channels = {
+            input: new LastValue(),
+            output: new LastValue(),
+        };
+        // Set initial value in input channel
+        channels.input.update(["test_input"]);
+        // Get value from input channel
+        const inputValue = channels.input.get();
+        expect(inputValue).toBe("test_input");
+        // Process value
+        const processedValue = `processed_${inputValue}`;
+        // Write to output channel
+        const updated = channels.output.update([processedValue]);
+        expect(updated).toBe(true);
+        // Read from output channel
+        const outputValue = channels.output.get();
+        expect(outputValue).toBe("processed_test_input");
+    });
+    it("should work with manual read and write operations", async () => {
+        // Setup test environment with real channels
+        const channels = {
+            input: new LastValue(),
+            output: new LastValue(),
+        };
+        // Initialize input channel with a value
+        channels.input.update(["test_input"]);
+        // Setup write tracking
+        let writtenValue = null;
+        // Manual read operation
+        const readFunc = (channel) => {
+            if (channel === "input") {
+                return channels.input.get();
+            }
+            return null;
+        };
+        // Manual write operation
+        const writeFunc = (values) => {
+            for (const [channel, value] of values) {
+                if (channel === "output") {
+                    writtenValue = value;
+                    channels.output.update([value]);
+                }
+            }
+        };
+        // Read from input channel
+        const inputValue = readFunc("input");
+        expect(inputValue).toBe("test_input");
+        // Process the value
+        const processedValue = `processed_${inputValue}`;
+        // Write to output channel
+        writeFunc([["output", processedValue]]);
+        // Verify the write happened
+        expect(writtenValue).toBe("processed_test_input");
+        expect(channels.output.get()).toBe("processed_test_input");
+    });
+});
+//# sourceMappingURL=read.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"read.test.js","sourceRoot":"","sources":["../../src/pregel/read.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAE3D,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,IAAI;aAClB,EAAE,EAAuD;aACzD,kBAAkB,CAAC,CAAC,OAA0B,EAAE,EAAE;YACjD,IAAI,OAAO,KAAK,cAAc,EAAE;gBAC9B,OAAO,YAAY,CAAC;aACrB;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEL,MAAM,MAAM,GAAmB;YAC7B,YAAY,EAAE;gBACZ,CAAC,eAAe,CAAC,EAAE,QAAQ;aAC5B;SACF,CAAC;QAEF,sBAAsB;QACtB,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,cAAc,CAAC,CAAC;QAEpD,uCAAuC;QACvC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEtD,iBAAiB;QACjB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,IAAI;aAClB,EAAE,EAIA;aACF,kBAAkB,CAAC,CAAC,QAA2B,EAAE,EAAE;YAClD,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAC3B,OAAO;oBACL,QAAQ,EAAE,QAAQ;oBAClB,QAAQ,EAAE,QAAQ;iBACnB,CAAC;aACH;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEL,MAAM,MAAM,GAAmB;YAC7B,YAAY,EAAE;gBACZ,CAAC,eAAe,CAAC,EAAE,QAAQ;aAC5B;SACF,CAAC;QAEF,4CAA4C;QAC5C,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;QAE9D,uCAAuC;QACvC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEtD,iBAAiB;QACjB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,QAAQ;YAClB,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;QAElE,MAAM,MAAM,GAAmB;YAC7B,YAAY,EAAE;gBACZ,CAAC,eAAe,CAAC,EAAE,QAAQ;aAC5B;SACF,CAAC;QAEF,yBAAyB;QACzB,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE,CAAC,UAAU,KAAK,EAAE,CAAC;QAEpD,kCAAkC;QAClC,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAE5D,uCAAuC;QACvC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEtD,iBAAiB;QACjB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,0DAA0D;QAC1D,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,yCAAyC;QACzC,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,qCAAqC,CACtC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC;YAC1B,QAAQ,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC;YAC9B,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC;YAC1B,QAAQ,EAAE,CAAC,OAAO,CAAC;YACnB,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC;YAC7B,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;SAC5C,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC;YAC1B,QAAQ,EAAE,CAAC,OAAO,CAAC;YACnB,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAE3E,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAE3E,mBAAmB;QACnB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAElD,wBAAwB;QACxB,MAAM,gBAAgB,GAAG,UAAU,CAAC,UAAU,EAAE,CAAC;QAEjD,gDAAgD;QAChD,MAAM,CAAC,gBAAgB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,CAAE,gBAAgB,CAAC,CAAC,CAAkB,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC;YAC1B,QAAQ,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE;YAChD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAE1C,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAClC,KAAK,EAAE,OAAO;YACd,OAAO,EAAE,SAAS;YAClB,OAAO,EAAE,SAAS;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,gEAAgE;QAEhE,4CAA4C;QAC5C,MAAM,QAAQ,GAAG;YACf,KAAK,EAAE,IAAI,SAAS,EAAU;YAC9B,MAAM,EAAE,IAAI,SAAS,EAAU;SAChC,CAAC;QAEF,qCAAqC;QACrC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAEtC,+BAA+B;QAC/B,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACxC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEtC,gBAAgB;QAChB,MAAM,cAAc,GAAG,aAAa,UAAU,EAAE,CAAC;QAEjD,0BAA0B;QAC1B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE3B,2BAA2B;QAC3B,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;QAC1C,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,4CAA4C;QAC5C,MAAM,QAAQ,GAAG;YACf,KAAK,EAAE,IAAI,SAAS,EAAU;YAC9B,MAAM,EAAE,IAAI,SAAS,EAAU;SAChC,CAAC;QAEF,wCAAwC;QACxC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAEtC,uBAAuB;QACvB,IAAI,YAAY,GAAkB,IAAI,CAAC;QAEvC,wBAAwB;QACxB,MAAM,QAAQ,GAAG,CAAC,OAAe,EAAiB,EAAE;YAClD,IAAI,OAAO,KAAK,OAAO,EAAE;gBACvB,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;aAC7B;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,yBAAyB;QACzB,MAAM,SAAS,GAAG,CAAC,MAA+B,EAAQ,EAAE;YAC1D,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE;gBACrC,IAAI,OAAO,KAAK,QAAQ,EAAE;oBACxB,YAAY,GAAG,KAAK,CAAC;oBACrB,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBACjC;aACF;QACH,CAAC,CAAC;QAEF,0BAA0B;QAC1B,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEtC,oBAAoB;QACpB,MAAM,cAAc,GAAG,aAAa,UAAU,EAAE,CAAC;QAEjD,0BAA0B;QAC1B,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;QAExC,4BAA4B;QAC5B,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/runner.test.cjs

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const runner_js_1 = require("./runner.cjs");
+(0, globals_1.describe)("PregelRunner", () => {
+    // Basic structure test
+    (0, globals_1.describe)("constructor", () => {
+        (0, globals_1.it)("should initialize without errors", () => {
+            const mockLoop = {};
+            const runner = new runner_js_1.PregelRunner({ loop: mockLoop });
+            (0, globals_1.expect)(runner).toBeInstanceOf(runner_js_1.PregelRunner);
+        });
+    });
+    // Simple behavior test with limited mocking
+    (0, globals_1.describe)("timeout option", () => {
+        (0, globals_1.it)("should pass timeout option to AbortSignal.timeout", async () => {
+            const mockLoop = {
+                config: {
+                    configurable: {
+                        thread_id: "1",
+                    },
+                },
+                tasks: {},
+                step: 1,
+                isNested: false,
+            };
+            const timeoutSpy = globals_1.jest.spyOn(AbortSignal, "timeout");
+            const runner = new runner_js_1.PregelRunner({ loop: mockLoop });
+            try {
+                await runner.tick({ timeout: 5000 });
+            }
+            catch (e) {
+                // Ignore errors
+            }
+            (0, globals_1.expect)(timeoutSpy).toHaveBeenCalledWith(5000);
+            timeoutSpy.mockRestore();
+        });
+    });
+    // Testing the onStepWrite callback behavior
+    (0, globals_1.describe)("onStepWrite callback", () => {
+        (0, globals_1.it)("should call onStepWrite with the step number and writes", async () => {
+            // Create a minimal implementation
+            const mockOnStepWrite = globals_1.jest.fn();
+            const mockLoop = {
+                config: {
+                    configurable: {
+                        thread_id: "1",
+                    },
+                },
+                tasks: {},
+                step: 42,
+                isNested: false,
+            };
+            const runner = new runner_js_1.PregelRunner({ loop: mockLoop });
+            try {
+                await runner.tick({ onStepWrite: mockOnStepWrite });
+            }
+            catch (e) {
+                // Ignore any errors from other parts of the code
+            }
+            // Verify the callback was called with the correct step number (42)
+            (0, globals_1.expect)(mockOnStepWrite).toHaveBeenCalledWith(42, globals_1.expect.any(Array));
+        });
+    });
+});
+//# sourceMappingURL=runner.test.js.map

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

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

package/dist/pregel/runner.test.js

@@ -0,0 +1,64 @@
+import { describe, expect, it, jest } from "@jest/globals";
+import { PregelRunner } from "./runner.js";
+describe("PregelRunner", () => {
+    // Basic structure test
+    describe("constructor", () => {
+        it("should initialize without errors", () => {
+            const mockLoop = {};
+            const runner = new PregelRunner({ loop: mockLoop });
+            expect(runner).toBeInstanceOf(PregelRunner);
+        });
+    });
+    // Simple behavior test with limited mocking
+    describe("timeout option", () => {
+        it("should pass timeout option to AbortSignal.timeout", async () => {
+            const mockLoop = {
+                config: {
+                    configurable: {
+                        thread_id: "1",
+                    },
+                },
+                tasks: {},
+                step: 1,
+                isNested: false,
+            };
+            const timeoutSpy = jest.spyOn(AbortSignal, "timeout");
+            const runner = new PregelRunner({ loop: mockLoop });
+            try {
+                await runner.tick({ timeout: 5000 });
+            }
+            catch (e) {
+                // Ignore errors
+            }
+            expect(timeoutSpy).toHaveBeenCalledWith(5000);
+            timeoutSpy.mockRestore();
+        });
+    });
+    // Testing the onStepWrite callback behavior
+    describe("onStepWrite callback", () => {
+        it("should call onStepWrite with the step number and writes", async () => {
+            // Create a minimal implementation
+            const mockOnStepWrite = jest.fn();
+            const mockLoop = {
+                config: {
+                    configurable: {
+                        thread_id: "1",
+                    },
+                },
+                tasks: {},
+                step: 42,
+                isNested: false,
+            };
+            const runner = new PregelRunner({ loop: mockLoop });
+            try {
+                await runner.tick({ onStepWrite: mockOnStepWrite });
+            }
+            catch (e) {
+                // Ignore any errors from other parts of the code
+            }
+            // Verify the callback was called with the correct step number (42)
+            expect(mockOnStepWrite).toHaveBeenCalledWith(42, expect.any(Array));
+        });
+    });
+});
+//# sourceMappingURL=runner.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"runner.test.js","sourceRoot":"","sources":["../../src/pregel/runner.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,uBAAuB;IACvB,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,QAAQ,GAAG,EAAgB,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4CAA4C;IAC5C,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE;oBACN,YAAY,EAAE;wBACZ,SAAS,EAAE,GAAG;qBACf;iBACF;gBACD,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,KAAK;aACS,CAAC;YAE3B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEpD,IAAI;gBACF,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;aACtC;YAAC,OAAO,CAAC,EAAE;gBACV,gBAAgB;aACjB;YAED,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC9C,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4CAA4C;IAC5C,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,kCAAkC;YAClC,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE;oBACN,YAAY,EAAE;wBACZ,SAAS,EAAE,GAAG;qBACf;iBACF;gBACD,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,KAAK;aACS,CAAC;YAE3B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEpD,IAAI;gBACF,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;aACrD;YAAC,OAAO,CAAC,EAAE;gBACV,iDAAiD;aAClD;YAED,mEAAmE;YACnE,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/utils/config.test.cjs

@@ -0,0 +1,214 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const singletons_1 = require("@langchain/core/singletons");
+const config_js_1 = require("./config.cjs");
+const constants_js_1 = require("../../constants.cjs");
+describe("ensureLangGraphConfig", () => {
+    // Save original to restore after tests
+    const originalGetRunnableConfig = singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig;
+    beforeEach(() => {
+        // Reset the mock before each test
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest.fn();
+    });
+    afterAll(() => {
+        // Restore the original after all tests
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig =
+            originalGetRunnableConfig;
+    });
+    it("should return a default config when no arguments provided", () => {
+        // Mock the AsyncLocalStorage to return undefined
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(undefined);
+        const result = (0, config_js_1.ensureLangGraphConfig)();
+        expect(result).toEqual({
+            tags: [],
+            metadata: {},
+            callbacks: undefined,
+            recursionLimit: 25,
+            configurable: {},
+        });
+    });
+    it("should merge multiple configs, with later configs taking precedence", () => {
+        // Mock the AsyncLocalStorage to return undefined
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config1 = {
+            tags: ["tag1"],
+            metadata: { key1: "value1" },
+            configurable: { option1: "value1" },
+        };
+        const config2 = {
+            tags: ["tag2"],
+            metadata: { key2: "value2" },
+            configurable: { option2: "value2" },
+        };
+        const result = (0, config_js_1.ensureLangGraphConfig)(config1, config2);
+        // The implementation completely replaces objects rather than merging them
+        expect(result).toEqual({
+            tags: ["tag2"],
+            metadata: { key2: "value2", option2: "value2" },
+            callbacks: undefined,
+            recursionLimit: 25,
+            configurable: { option2: "value2" },
+        });
+    });
+    it("should copy values from AsyncLocalStorage if available", () => {
+        // Mock values from AsyncLocalStorage
+        const asyncLocalStorageConfig = {
+            tags: ["storage-tag"],
+            metadata: { storage: "value" },
+            callbacks: { copy: () => ({ type: "copied-callback" }) },
+            configurable: { storageOption: "value" },
+        };
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(asyncLocalStorageConfig);
+        const result = (0, config_js_1.ensureLangGraphConfig)();
+        expect(result.tags).toEqual(["storage-tag"]);
+        expect(result.metadata || {}).toEqual({
+            storage: "value",
+            storageOption: "value",
+        });
+        expect(result.configurable).toEqual({ storageOption: "value" });
+        expect(result.callbacks).toEqual({ type: "copied-callback" });
+    });
+    it("should handle undefined config values", () => {
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config1 = undefined;
+        const config2 = {
+            tags: ["tag2"],
+            metadata: undefined,
+        };
+        const result = (0, config_js_1.ensureLangGraphConfig)(config1, config2);
+        expect(result.tags).toEqual(["tag2"]);
+        expect(result.metadata).toEqual({});
+    });
+    it("should copy scalar values to metadata from configurable", () => {
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config = {
+            configurable: {
+                stringValue: "string",
+                numberValue: 42,
+                booleanValue: true,
+                objectValue: { should: "not be copied" },
+                __privateValue: "should not be copied",
+            },
+        };
+        const result = (0, config_js_1.ensureLangGraphConfig)(config);
+        expect(result.metadata).toEqual({
+            stringValue: "string",
+            numberValue: 42,
+            booleanValue: true,
+            // objectValue and __privateValue should not be copied
+        });
+    });
+    it("should not overwrite existing metadata values with configurable values", () => {
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config = {
+            metadata: { key: "original value" },
+            configurable: {
+                key: "should not overwrite",
+            },
+        };
+        const result = (0, config_js_1.ensureLangGraphConfig)(config);
+        expect(result.metadata?.key).toEqual("original value");
+    });
+});
+describe("getStore, getWriter, getConfig", () => {
+    // Save original to restore after tests
+    const originalGetRunnableConfig = singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig;
+    beforeEach(() => {
+        // Reset the mock before each test
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest.fn();
+    });
+    afterAll(() => {
+        // Restore the original after all tests
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig =
+            originalGetRunnableConfig;
+    });
+    it("getStore should return store from config", () => {
+        const mockStore = {};
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue({
+            store: mockStore,
+        });
+        const result = (0, config_js_1.getStore)();
+        expect(result).toBe(mockStore);
+    });
+    it("getWriter should return writer from configurable", () => {
+        const mockWriter = () => { };
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue({
+            configurable: {
+                writer: mockWriter,
+            },
+        });
+        const result = (0, config_js_1.getWriter)();
+        expect(result).toBe(mockWriter);
+    });
+    it("getConfig should return the full config", () => {
+        const mockConfig = { key: "value" };
+        singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig = globals_1.jest
+            .fn()
+            .mockReturnValue(mockConfig);
+        const result = (0, config_js_1.getConfig)();
+        expect(result).toBe(mockConfig);
+    });
+});
+describe("recastCheckpointNamespace", () => {
+    it("should filter out numeric parts of the namespace", () => {
+        const namespace = `parent${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}123${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child`;
+        const result = (0, config_js_1.recastCheckpointNamespace)(namespace);
+        expect(result).toBe(`parent${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child`);
+    });
+    it("should remove parts after CHECKPOINT_NAMESPACE_END", () => {
+        const namespace = `part1${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}part2${constants_js_1.CHECKPOINT_NAMESPACE_END}extra`;
+        const result = (0, config_js_1.recastCheckpointNamespace)(namespace);
+        expect(result).toBe(`part1${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}part2`);
+    });
+    it("should handle complex namespace with numeric parts and CHECKPOINT_NAMESPACE_END", () => {
+        const namespace = `root${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}123${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child${constants_js_1.CHECKPOINT_NAMESPACE_END}extra${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}456`;
+        const result = (0, config_js_1.recastCheckpointNamespace)(namespace);
+        expect(result).toBe(`root${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child`);
+    });
+    it("should return the original namespace when no filtering needed", () => {
+        const namespace = `part1${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}part2`;
+        const result = (0, config_js_1.recastCheckpointNamespace)(namespace);
+        expect(result).toBe(namespace);
+    });
+});
+describe("getParentCheckpointNamespace", () => {
+    it("should return the parent namespace by removing the last part", () => {
+        const namespace = `parent${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child`;
+        const result = (0, config_js_1.getParentCheckpointNamespace)(namespace);
+        expect(result).toBe("parent");
+    });
+    it("should skip trailing numeric parts", () => {
+        const namespace = `parent${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}child${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}123${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}456`;
+        const result = (0, config_js_1.getParentCheckpointNamespace)(namespace);
+        expect(result).toBe("parent");
+    });
+    it("should return empty string for top-level namespace", () => {
+        const namespace = "singlePart";
+        const result = (0, config_js_1.getParentCheckpointNamespace)(namespace);
+        expect(result).toBe("");
+    });
+    it("should handle namespace with mixed numeric and non-numeric parts", () => {
+        const namespace = `root${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}sub1${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}123${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}sub2`;
+        const result = (0, config_js_1.getParentCheckpointNamespace)(namespace);
+        // The implementation stops at the first numeric part, not at the last non-numeric part
+        expect(result).toBe(`root${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}sub1${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}123`);
+    });
+});
+//# sourceMappingURL=config.test.js.map

package/dist/pregel/utils/config.test.d.ts

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