@langchain/langgraph 0.4.9 → 1.0.0-alpha.1

package/dist/pregel/read.test.js

@@ -1,192 +0,0 @@
-import { describe, expect, it, vi } from "vitest";
-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 = vi
-            .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 = vi
-            .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 = vi.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

@@ -1 +0,0 @@
-{"version":3,"file":"read.test.js","sourceRoot":"","sources":["../../src/pregel/read.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,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,EAAE;aAChB,EAAE,EAAuD;aACzD,kBAAkB,CAAC,CAAC,OAA0B,EAAE,EAAE;YACjD,IAAI,OAAO,KAAK,cAAc,EAAE,CAAC;gBAC/B,OAAO,YAAY,CAAC;YACtB,CAAC;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,EAAE;aAChB,EAAE,EAIA;aACF,kBAAkB,CAAC,CAAC,QAA2B,EAAE,EAAE;YAClD,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,OAAO;oBACL,QAAQ,EAAE,QAAQ;oBAClB,QAAQ,EAAE,QAAQ;iBACnB,CAAC;YACJ,CAAC;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,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;QAEhE,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,CAAC;gBACxB,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YAC9B,CAAC;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,CAAC;gBACtC,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACzB,YAAY,GAAG,KAAK,CAAC;oBACrB,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;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/retry.d.ts

@@ -1,17 +0,0 @@
-import { PregelExecutableTask } from "./types.js";
-import { type RetryPolicy } from "./utils/index.js";
-export declare const DEFAULT_INITIAL_INTERVAL = 500;
-export declare const DEFAULT_BACKOFF_FACTOR = 2;
-export declare const DEFAULT_MAX_INTERVAL = 128000;
-export declare const DEFAULT_MAX_RETRIES = 3;
-export type SettledPregelTask = {
-    task: PregelExecutableTask<any, any>;
-    error: Error;
-    signalAborted?: boolean;
-};
-export declare function _runWithRetry<N extends PropertyKey, C extends PropertyKey>(pregelTask: PregelExecutableTask<N, C>, retryPolicy?: RetryPolicy, configurable?: Record<string, unknown>, signal?: AbortSignal): Promise<{
-    task: PregelExecutableTask<N, C>;
-    result: unknown;
-    error: Error | undefined;
-    signalAborted?: boolean;
-}>;

package/dist/pregel/runnable_types.cjs

@@ -1,3 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=runnable_types.js.map

package/dist/pregel/runnable_types.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=runnable_types.js.map

package/dist/pregel/runnable_types.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"runnable_types.js","sourceRoot":"","sources":["../../src/pregel/runnable_types.ts"],"names":[],"mappings":""}

package/dist/pregel/runner.d.ts

@@ -1,79 +0,0 @@
-import { PendingWrite } from "@langchain/langgraph-checkpoint";
-import { RetryPolicy } from "./utils/index.js";
-import { PregelLoop } from "./loop.js";
-/**
- * Options for the {@link PregelRunner#tick} method.
- */
-export type TickOptions = {
-    /**
-     * The deadline before which all tasks must be completed.
-     */
-    timeout?: number;
-    /**
-     * An optional {@link AbortSignal} to cancel processing of tasks.
-     */
-    signal?: AbortSignal;
-    /**
-     * The {@link RetryPolicy} to use for the tick.
-     */
-    retryPolicy?: RetryPolicy;
-    /**
-     * An optional callback to be called after all task writes are completed.
-     */
-    onStepWrite?: (step: number, writes: PendingWrite[]) => void;
-    /**
-     * The maximum number of tasks to execute concurrently.
-     */
-    maxConcurrency?: number;
-};
-/**
- * Responsible for handling task execution on each tick of the {@link PregelLoop}.
- */
-export declare class PregelRunner {
-    private nodeFinished?;
-    private loop;
-    /**
-     * Construct a new PregelRunner, which executes tasks from the provided PregelLoop.
-     * @param loop - The PregelLoop that produces tasks for this runner to execute.
-     */
-    constructor({ loop, nodeFinished, }: {
-        loop: PregelLoop;
-        nodeFinished?: (id: string) => void;
-    });
-    /**
-     * Execute tasks from the current step of the PregelLoop.
-     *
-     * Note: this method does NOT call {@link PregelLoop}#tick. That must be handled externally.
-     * @param options - Options for the execution.
-     */
-    tick(options?: TickOptions): Promise<void>;
-    /**
-     * Initializes the current AbortSignals for the PregelRunner, handling the various ways that
-     * AbortSignals must be chained together so that the PregelLoop can be interrupted if necessary
-     * while still allowing nodes to gracefully exit.
-     *
-     * This method must only be called once per PregelRunner#tick. It has the side effect of updating
-     * the PregelLoop#config with the new AbortSignals so they may be propagated correctly to future
-     * ticks and subgraph calls.
-     *
-     * @param options - Options for the initialization.
-     * @returns The current abort signals.
-     * @internal
-     */
-    private _initializeAbortSignals;
-    /**
-     * Concurrently executes tasks with the requested retry policy, yielding a {@link SettledPregelTask} for each task as it completes.
-     * @param tasks - The tasks to execute.
-     * @param options - Options for the execution.
-     */
-    private _executeTasksWithRetry;
-    /**
-     * Determines what writes to apply based on whether the task completed successfully, and what type of error occurred.
-     *
-     * Throws an error if the error is a {@link GraphBubbleUp} error and {@link PregelLoop}#isNested is true.
-     *
-     * @param task - The task to commit.
-     * @param error - The error that occurred, if any.
-     */
-    private _commit;
-}

package/dist/pregel/runner.test.cjs

@@ -1,66 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const vitest_1 = require("vitest");
-const runner_js_1 = require("./runner.cjs");
-(0, vitest_1.describe)("PregelRunner", () => {
-    // Basic structure test
-    (0, vitest_1.describe)("constructor", () => {
-        (0, vitest_1.it)("should initialize without errors", () => {
-            const mockLoop = {};
-            const runner = new runner_js_1.PregelRunner({ loop: mockLoop });
-            (0, vitest_1.expect)(runner).toBeInstanceOf(runner_js_1.PregelRunner);
-        });
-    });
-    // Simple behavior test with limited mocking
-    (0, vitest_1.describe)("timeout option", () => {
-        (0, vitest_1.it)("should pass timeout option to AbortSignal.timeout", async () => {
-            const mockLoop = {
-                config: {
-                    configurable: {
-                        thread_id: "1",
-                    },
-                },
-                tasks: {},
-                step: 1,
-                isNested: false,
-            };
-            const timeoutSpy = vitest_1.vi.spyOn(AbortSignal, "timeout");
-            const runner = new runner_js_1.PregelRunner({ loop: mockLoop });
-            try {
-                await runner.tick({ timeout: 5000 });
-            }
-            catch (e) {
-                // Ignore errors
-            }
-            (0, vitest_1.expect)(timeoutSpy).toHaveBeenCalledWith(5000);
-            timeoutSpy.mockRestore();
-        });
-    });
-    // Testing the onStepWrite callback behavior
-    (0, vitest_1.describe)("onStepWrite callback", () => {
-        (0, vitest_1.it)("should call onStepWrite with the step number and writes", async () => {
-            // Create a minimal implementation
-            const mockOnStepWrite = vitest_1.vi.fn();
-            const mockLoop = {
-                config: {
-                    configurable: {
-                        thread_id: "1",
-                    },
-                },
-                tasks: {},
-                step: 42, // Use a unique value to verify it's passed correctly
-                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, vitest_1.expect)(mockOnStepWrite).toHaveBeenCalledWith(42, vitest_1.expect.any(Array));
-        });
-    });
-});
-//# sourceMappingURL=runner.test.js.map

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

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

package/dist/pregel/runner.test.js

@@ -1,64 +0,0 @@
-import { describe, expect, it, vi } from "vitest";
-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 = vi.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 = vi.fn();
-            const mockLoop = {
-                config: {
-                    configurable: {
-                        thread_id: "1",
-                    },
-                },
-                tasks: {},
-                step: 42, // Use a unique value to verify it's passed correctly
-                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

@@ -1 +0,0 @@
-{"version":3,"file":"runner.test.js","sourceRoot":"","sources":["../../src/pregel/runner.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,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,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEpD,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACvC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,gBAAgB;YAClB,CAAC;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,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE;oBACN,YAAY,EAAE;wBACZ,SAAS,EAAE,GAAG;qBACf;iBACF;gBACD,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,EAAE,EAAE,qDAAqD;gBAC/D,QAAQ,EAAE,KAAK;aACS,CAAC;YAE3B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEpD,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YACtD,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,iDAAiD;YACnD,CAAC;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/stream.d.ts

@@ -1,40 +0,0 @@
-import { IterableReadableStream } from "@langchain/core/utils/stream";
-import { StreamMode } from "./types.js";
-export type StreamChunk = [string[], StreamMode, unknown];
-/**
- * A wrapper around an IterableReadableStream that allows for aborting the stream when
- * {@link cancel} is called.
- */
-export declare class IterableReadableStreamWithAbortSignal<T> extends IterableReadableStream<T> {
-    protected _abortController: AbortController;
-    protected _innerReader: ReadableStreamDefaultReader<T>;
-    /**
-     * @param readableStream - The stream to wrap.
-     * @param abortController - The abort controller to use. Optional. One will be created if not provided.
-     */
-    constructor(readableStream: ReadableStream<T>, abortController?: AbortController);
-    /**
-     * Aborts the stream, abandoning any pending operations in progress. Calling this triggers an
-     * {@link AbortSignal} that is propagated to the tasks that are producing the data for this stream.
-     * @param reason - The reason for aborting the stream. Optional.
-     */
-    cancel(reason?: unknown): Promise<void>;
-    /**
-     * The {@link AbortSignal} for the stream. Aborted when {@link cancel} is called.
-     */
-    get signal(): AbortSignal;
-}
-export declare class IterableReadableWritableStream extends IterableReadableStream<StreamChunk> {
-    modes: Set<StreamMode>;
-    private controller;
-    private passthroughFn?;
-    private _closed;
-    get closed(): boolean;
-    constructor(params: {
-        passthroughFn?: (chunk: StreamChunk) => void;
-        modes: Set<StreamMode>;
-    });
-    push(chunk: StreamChunk): void;
-    close(): void;
-    error(e: any): void;
-}