@langchain/langgraph 0.2.61 → 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/retry.cjs

@@ -57,6 +57,10 @@ pregelTask, retryPolicy, configurable, signal) {
     if (configurable) {
         config = (0, index_js_1.patchConfigurable)(config, configurable);
     }
+    config = {
+        ...config,
+        signal,
+    };
     // eslint-disable-next-line no-constant-condition
     while (true) {
         if (signal?.aborted) {
@@ -132,6 +136,7 @@ pregelTask, retryPolicy, configurable, signal) {
         task: pregelTask,
         result,
         error: error,
+        signalAborted: signal?.aborted,
     };
 }
 exports._runWithRetry = _runWithRetry;

package/dist/pregel/retry.d.ts

@@ -7,9 +7,11 @@ 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/retry.js

@@ -54,6 +54,10 @@ pregelTask, retryPolicy, configurable, signal) {
     if (configurable) {
         config = patchConfigurable(config, configurable);
     }
+    config = {
+        ...config,
+        signal,
+    };
     // eslint-disable-next-line no-constant-condition
     while (true) {
         if (signal?.aborted) {
@@ -129,6 +133,7 @@ pregelTask, retryPolicy, configurable, signal) {
         task: pregelTask,
         result,
         error: error,
+        signalAborted: signal?.aborted,
     };
 }
 //# sourceMappingURL=retry.js.map

package/dist/pregel/retry.js.map

@@ -1 +1 @@
-{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/pregel/retry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAoB,MAAM,kBAAkB,CAAC;AAEvE,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AAC5C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC;AACxC,MAAM,CAAC,MAAM,oBAAoB,GAAG,MAAM,CAAC;AAC3C,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,uBAAuB,GAAG;IAC9B,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG,EAAE,WAAW;CACjB,CAAC;AAEF,8DAA8D;AAC9D,MAAM,wBAAwB,GAAG,CAAC,KAAU,EAAE,EAAE;IAC9C,IACE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;QAClC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC;QACtC,KAAK,CAAC,IAAI,KAAK,YAAY,EAC3B;QACA,OAAO,KAAK,CAAC;KACd;IACD,8DAA8D;IAC9D,IAAK,KAAa,EAAE,IAAI,KAAK,cAAc,EAAE;QAC3C,OAAO,KAAK,CAAC;KACd;IACD,MAAM,MAAM;IACV,8DAA8D;IAC7D,KAAa,EAAE,QAAQ,EAAE,MAAM,IAAK,KAAa,EAAE,MAAM,CAAC;IAC7D,IAAI,MAAM,IAAI,uBAAuB,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE;QACvD,OAAO,KAAK,CAAC;KACd;IACD,8DAA8D;IAC9D,IAAK,KAAa,EAAE,KAAK,EAAE,IAAI,KAAK,oBAAoB,EAAE;QACxD,OAAO,KAAK,CAAC;KACd;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAQF,MAAM,CAAC,KAAK,UAAU,aAAa;AAIjC,8DAA8D;AAC9D,UAAsC,EACtC,WAAyB,EACzB,YAAsC,EACtC,MAAoB;IAMpB,MAAM,mBAAmB,GAAG,UAAU,CAAC,YAAY,IAAI,WAAW,CAAC;IACnE,IAAI,QAAQ,GACV,mBAAmB,KAAK,SAAS;QAC/B,CAAC,CAAC,mBAAmB,CAAC,eAAe,IAAI,wBAAwB;QACjE,CAAC,CAAC,CAAC,CAAC;IACR,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,KAAK,CAAC;IACV,IAAI,MAAM,CAAC;IAEX,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;IAC5B,IAAI,YAAY,EAAE;QAChB,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;KAClD;IACD,iDAAiD;IACjD,OAAO,IAAI,EAAE;QACX,IAAI,MAAM,EAAE,OAAO,EAAE;YACnB,gEAAgE;YAChE,wEAAwE;YACxE,MAAM;SACP;QACD,0CAA0C;QAC1C,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtD,KAAK,GAAG,SAAS,CAAC;QAClB,IAAI;YACF,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAChE,MAAM;SACP;QAAC,OAAO,CAAU,EAAE;YACnB,KAAK,GAAG,CAAC,CAAC;YACT,KAAkC,CAAC,YAAY,GAAG,UAAU,CAAC,EAAE,CAAC;YACjE,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE;gBAC1B,MAAM,EAAE,GAAW,MAAM,EAAE,YAAY,EAAE,aAAa,CAAC;gBACvD,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC;gBAC1B,IAAI,GAAG,CAAC,KAAK,KAAK,EAAE,EAAE;oBACpB,mDAAmD;oBACnD,KAAK,MAAM,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE;wBACvC,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;qBAClC;oBACD,KAAK,GAAG,SAAS,CAAC;oBAClB,MAAM;iBACP;qBAAM,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE;oBACvC,gEAAgE;oBAChE,MAAM,QAAQ,GAAG,4BAA4B,CAAC,EAAE,CAAC,CAAC;oBAClD,KAAK,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC;wBAC1B,GAAG,KAAK,CAAC,OAAO;wBAChB,KAAK,EAAE,QAAQ;qBAChB,CAAC,CAAC;iBACJ;aACF;YACD,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE;gBAC1B,MAAM;aACP;YACD,IAAI,mBAAmB,KAAK,SAAS,EAAE;gBACrC,MAAM;aACP;YACD,QAAQ,IAAI,CAAC,CAAC;YACd,6BAA6B;YAC7B,IACE,QAAQ,IAAI,CAAC,mBAAmB,CAAC,WAAW,IAAI,mBAAmB,CAAC,EACpE;gBACA,MAAM;aACP;YACD,MAAM,OAAO,GAAG,mBAAmB,CAAC,OAAO,IAAI,wBAAwB,CAAC;YACxE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;gBACnB,MAAM;aACP;YACD,QAAQ,GAAG,IAAI,CAAC,GAAG,CACjB,mBAAmB,CAAC,WAAW,IAAI,oBAAoB,EACvD,QAAQ,GAAG,CAAC,mBAAmB,CAAC,aAAa,IAAI,sBAAsB,CAAC,CACzE,CAAC;YACF,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,MAAM;gBACnD,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;gBAC7C,CAAC,CAAC,QAAQ,CAAC;YACb,wBAAwB;YACxB,sDAAsD;YACtD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC;YACxE,gBAAgB;YAChB,MAAM,SAAS,GACZ,KAAe,CAAC,IAAI;gBACrB,8DAA8D;gBAC5D,KAAe,CAAC,WAAmB,CAAC,iBAAiB;gBACtD,KAAe,CAAC,WAAW,CAAC,IAAI,CAAC;YACpC,IAAI,mBAAmB,EAAE,UAAU,IAAI,IAAI,EAAE;gBAC3C,OAAO,CAAC,GAAG,CACT,kBAAkB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,QAAQ,CAAC,OAAO,CAClE,CAAC,CACF,eAAe,QAAQ,WAAW,SAAS,KAAK,KAAK,EAAE,CACzD,CAAC;aACH;YAED,4CAA4C;YAC5C,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,EAAE,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;SACrE;KACF;IACD,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,MAAM;QACN,KAAK,EAAE,KAA0B;KAClC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/pregel/retry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAoB,MAAM,kBAAkB,CAAC;AAEvE,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AAC5C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC;AACxC,MAAM,CAAC,MAAM,oBAAoB,GAAG,MAAM,CAAC;AAC3C,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,uBAAuB,GAAG;IAC9B,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG,EAAE,WAAW;CACjB,CAAC;AAEF,8DAA8D;AAC9D,MAAM,wBAAwB,GAAG,CAAC,KAAU,EAAE,EAAE;IAC9C,IACE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;QAClC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC;QACtC,KAAK,CAAC,IAAI,KAAK,YAAY,EAC3B;QACA,OAAO,KAAK,CAAC;KACd;IACD,8DAA8D;IAC9D,IAAK,KAAa,EAAE,IAAI,KAAK,cAAc,EAAE;QAC3C,OAAO,KAAK,CAAC;KACd;IACD,MAAM,MAAM;IACV,8DAA8D;IAC7D,KAAa,EAAE,QAAQ,EAAE,MAAM,IAAK,KAAa,EAAE,MAAM,CAAC;IAC7D,IAAI,MAAM,IAAI,uBAAuB,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE;QACvD,OAAO,KAAK,CAAC;KACd;IACD,8DAA8D;IAC9D,IAAK,KAAa,EAAE,KAAK,EAAE,IAAI,KAAK,oBAAoB,EAAE;QACxD,OAAO,KAAK,CAAC;KACd;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AASF,MAAM,CAAC,KAAK,UAAU,aAAa;AAIjC,8DAA8D;AAC9D,UAAsC,EACtC,WAAyB,EACzB,YAAsC,EACtC,MAAoB;IAOpB,MAAM,mBAAmB,GAAG,UAAU,CAAC,YAAY,IAAI,WAAW,CAAC;IACnE,IAAI,QAAQ,GACV,mBAAmB,KAAK,SAAS;QAC/B,CAAC,CAAC,mBAAmB,CAAC,eAAe,IAAI,wBAAwB;QACjE,CAAC,CAAC,CAAC,CAAC;IACR,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,KAAK,CAAC;IACV,IAAI,MAAM,CAAC;IAEX,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;IAC5B,IAAI,YAAY,EAAE;QAChB,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;KAClD;IAED,MAAM,GAAG;QACP,GAAG,MAAM;QACT,MAAM;KACP,CAAC;IAEF,iDAAiD;IACjD,OAAO,IAAI,EAAE;QACX,IAAI,MAAM,EAAE,OAAO,EAAE;YACnB,gEAAgE;YAChE,wEAAwE;YACxE,MAAM;SACP;QACD,0CAA0C;QAC1C,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtD,KAAK,GAAG,SAAS,CAAC;QAClB,IAAI;YACF,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAChE,MAAM;SACP;QAAC,OAAO,CAAU,EAAE;YACnB,KAAK,GAAG,CAAC,CAAC;YACT,KAAkC,CAAC,YAAY,GAAG,UAAU,CAAC,EAAE,CAAC;YACjE,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE;gBAC1B,MAAM,EAAE,GAAW,MAAM,EAAE,YAAY,EAAE,aAAa,CAAC;gBACvD,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC;gBAC1B,IAAI,GAAG,CAAC,KAAK,KAAK,EAAE,EAAE;oBACpB,mDAAmD;oBACnD,KAAK,MAAM,MAAM,IAAI,UAAU,CAAC,OAAO,EAAE;wBACvC,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;qBAClC;oBACD,KAAK,GAAG,SAAS,CAAC;oBAClB,MAAM;iBACP;qBAAM,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE;oBACvC,gEAAgE;oBAChE,MAAM,QAAQ,GAAG,4BAA4B,CAAC,EAAE,CAAC,CAAC;oBAClD,KAAK,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC;wBAC1B,GAAG,KAAK,CAAC,OAAO;wBAChB,KAAK,EAAE,QAAQ;qBAChB,CAAC,CAAC;iBACJ;aACF;YACD,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE;gBAC1B,MAAM;aACP;YACD,IAAI,mBAAmB,KAAK,SAAS,EAAE;gBACrC,MAAM;aACP;YACD,QAAQ,IAAI,CAAC,CAAC;YACd,6BAA6B;YAC7B,IACE,QAAQ,IAAI,CAAC,mBAAmB,CAAC,WAAW,IAAI,mBAAmB,CAAC,EACpE;gBACA,MAAM;aACP;YACD,MAAM,OAAO,GAAG,mBAAmB,CAAC,OAAO,IAAI,wBAAwB,CAAC;YACxE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;gBACnB,MAAM;aACP;YACD,QAAQ,GAAG,IAAI,CAAC,GAAG,CACjB,mBAAmB,CAAC,WAAW,IAAI,oBAAoB,EACvD,QAAQ,GAAG,CAAC,mBAAmB,CAAC,aAAa,IAAI,sBAAsB,CAAC,CACzE,CAAC;YACF,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,MAAM;gBACnD,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;gBAC7C,CAAC,CAAC,QAAQ,CAAC;YACb,wBAAwB;YACxB,sDAAsD;YACtD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC;YACxE,gBAAgB;YAChB,MAAM,SAAS,GACZ,KAAe,CAAC,IAAI;gBACrB,8DAA8D;gBAC5D,KAAe,CAAC,WAAmB,CAAC,iBAAiB;gBACtD,KAAe,CAAC,WAAW,CAAC,IAAI,CAAC;YACpC,IAAI,mBAAmB,EAAE,UAAU,IAAI,IAAI,EAAE;gBAC3C,OAAO,CAAC,GAAG,CACT,kBAAkB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,QAAQ,CAAC,OAAO,CAClE,CAAC,CACF,eAAe,QAAQ,WAAW,SAAS,KAAK,KAAK,EAAE,CACzD,CAAC;aACH;YAED,4CAA4C;YAC5C,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,EAAE,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;SACrE;KACF;IACD,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,MAAM;QACN,KAAK,EAAE,KAA0B;QACjC,aAAa,EAAE,MAAM,EAAE,OAAO;KAC/B,CAAC;AACJ,CAAC"}

package/dist/pregel/runner.cjs

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PregelRunner = void 0;
 const types_js_1 = require("./types.cjs");
+const index_js_1 = require("./utils/index.cjs");
 const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 const retry_js_1 = require("./retry.cjs");
@@ -36,18 +37,23 @@ class PregelRunner {
      * @param options - Options for the execution.
      */
     async tick(options = {}) {
-        const { timeout, signal, retryPolicy, onStepWrite, maxConcurrency } = options;
+        const { timeout, retryPolicy, onStepWrite, maxConcurrency } = options;
         const nodeErrors = new Set();
         let graphBubbleUp;
+        const exceptionSignalController = new AbortController();
         // Start task execution
         const pendingTasks = Object.values(this.loop.tasks).filter((t) => t.writes.length === 0);
+        const currentSignals = this._initializeAbortSignals({
+            exceptionSignalController,
+            timeout,
+            signal: options.signal,
+        });
         const taskStream = this._executeTasksWithRetry(pendingTasks, {
-            stepTimeout: timeout,
-            signal,
+            signals: currentSignals,
             retryPolicy,
             maxConcurrency,
         });
-        for await (const { task, error } of taskStream) {
+        for await (const { task, error, signalAborted } of taskStream) {
             this._commit(task, error);
             if ((0, errors_js_1.isGraphInterrupt)(error)) {
                 graphBubbleUp = error;
@@ -55,7 +61,22 @@ class PregelRunner {
             else if ((0, errors_js_1.isGraphBubbleUp)(error) && !(0, errors_js_1.isGraphInterrupt)(graphBubbleUp)) {
                 graphBubbleUp = error;
             }
-            else if (error) {
+            else if (error && (nodeErrors.size === 0 || !signalAborted)) {
+                /*
+                 * The goal here is to capture the exception that causes the graph to terminate early. In
+                 * theory it's possible for multiple nodes to throw, so this also handles the edge case of
+                 * capturing concurrent exceptions thrown before the node saw an abort. This is checked via
+                 * the signalAborted flag, which records the state of the abort signal at the time the node
+                 * execution finished.
+                 *
+                 * There is a case however where one node throws some error causing us to trigger an abort,
+                 * which then causes other concurrently executing nodes to throw their own AbortErrors. In
+                 * this case we don't care about reporting the abort errors thrown by the other nodes,
+                 * because they don't tell the user anything about what caused the graph execution to
+                 * terminate early, so we ignore them (and any other errors that occur after the node sees
+                 * an abort signal).
+                 */
+                exceptionSignalController.abort();
                 nodeErrors.add(error);
             }
         }
@@ -75,14 +96,60 @@ class PregelRunner {
             throw graphBubbleUp;
         }
     }
+    /**
+     * 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
+     */
+    _initializeAbortSignals({ exceptionSignalController, timeout, signal, }) {
+        const previousSignals = this.loop.config.configurable?.[constants_js_1.CONFIG_KEY_ABORT_SIGNALS] ?? {};
+        // This is true when a node calls a subgraph and, rather than forwarding its own AbortSignal,
+        // it creates a new AbortSignal and passes that along instead.
+        const subgraphCalledWithSignalCreatedByNode = signal &&
+            previousSignals.composedAbortSignal &&
+            signal !== previousSignals.composedAbortSignal;
+        const externalAbortSignal = subgraphCalledWithSignalCreatedByNode
+            ? // Chain the signals here to make sure that the subgraph receives the external abort signal in
+                // addition to the signal created by the node.
+                (0, index_js_1.combineAbortSignals)(previousSignals.externalAbortSignal, signal)
+            : // Otherwise, just keep using the external abort signal, or initialize it if it hasn't been
+                // assigned yet
+                previousSignals.externalAbortSignal ?? signal;
+        const errorAbortSignal = previousSignals.errorAbortSignal
+            ? // Chaining here rather than always using a fresh one handles the case where a subgraph is
+                // called in a parallel branch to some other node in the parent graph.
+                (0, index_js_1.combineAbortSignals)(previousSignals.errorAbortSignal, exceptionSignalController.signal)
+            : exceptionSignalController.signal;
+        const timeoutAbortSignal = timeout
+            ? AbortSignal.timeout(timeout)
+            : undefined;
+        const composedAbortSignal = (0, index_js_1.combineAbortSignals)(...(externalAbortSignal ? [externalAbortSignal] : []), ...(timeoutAbortSignal ? [timeoutAbortSignal] : []), errorAbortSignal);
+        const currentSignals = {
+            externalAbortSignal,
+            errorAbortSignal,
+            timeoutAbortSignal,
+            composedAbortSignal,
+        };
+        this.loop.config = (0, index_js_1.patchConfigurable)(this.loop.config, {
+            [constants_js_1.CONFIG_KEY_ABORT_SIGNALS]: currentSignals,
+        });
+        return currentSignals;
+    }
     /**
      * 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.
      */
     async *_executeTasksWithRetry(tasks, options) {
-        const { stepTimeout, retryPolicy, maxConcurrency } = options ?? {};
-        let signal = options?.signal;
+        const { retryPolicy, maxConcurrency, signals } = options ?? {};
         const promiseAddedSymbol = Symbol.for("promiseAdded");
         let addedPromiseSignal;
         let addedPromiseWait;
@@ -196,29 +263,26 @@ class PregelRunner {
             }
             return Promise.resolve();
         }
-        if (stepTimeout && signal) {
-            if ("any" in AbortSignal) {
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                signal = AbortSignal.any([
-                    signal,
-                    AbortSignal.timeout(stepTimeout),
-                ]);
-            }
-        }
-        else if (stepTimeout) {
-            signal = AbortSignal.timeout(stepTimeout);
-        }
-        if (signal?.aborted) {
+        if (signals?.composedAbortSignal?.aborted) {
             // note: don't use throwIfAborted here because it throws a DOMException,
             // which isn't consistent with how we throw on abort below.
             throw new Error("Abort");
         }
         let startedTasksCount = 0;
         let listener;
-        const signalPromise = new Promise((_resolve, reject) => {
-            listener = () => reject(new Error("Abort"));
-            signal?.addEventListener("abort", listener);
-        }).finally(() => signal?.removeEventListener("abort", listener));
+        const timeoutOrCancelSignal = signals?.externalAbortSignal || signals?.timeoutAbortSignal
+            ? (0, index_js_1.combineAbortSignals)(...(signals.externalAbortSignal
+                ? [signals.externalAbortSignal]
+                : []), ...(signals.timeoutAbortSignal ? [signals.timeoutAbortSignal] : []))
+            : undefined;
+        const abortPromise = timeoutOrCancelSignal
+            ? new Promise((_resolve, reject) => {
+                listener = () => reject(new Error("Abort"));
+                timeoutOrCancelSignal.addEventListener("abort", listener, {
+                    once: true,
+                });
+            })
+            : undefined;
         while ((startedTasksCount === 0 || Object.keys(executingTasksMap).length > 0) &&
             tasks.length) {
             for (; Object.values(executingTasksMap).length <
@@ -227,13 +291,17 @@ class PregelRunner {
                 executingTasksMap[task.id] = (0, retry_js_1._runWithRetry)(task, retryPolicy, {
                     [constants_js_1.CONFIG_KEY_SEND]: writer?.bind(null, this, task),
                     [constants_js_1.CONFIG_KEY_CALL]: call?.bind(null, this, task),
-                }).catch((error) => {
-                    return { task, error };
+                }, signals?.composedAbortSignal).catch((error) => {
+                    return {
+                        task,
+                        error,
+                        signalAborted: signals?.composedAbortSignal?.aborted,
+                    };
                 });
             }
             const settledTask = await Promise.race([
                 ...Object.values(executingTasksMap),
-                signalPromise,
+                ...(abortPromise ? [abortPromise] : []),
                 addedPromiseWait,
             ]);
             if (settledTask === promiseAddedSymbol) {

package/dist/pregel/runner.d.ts

@@ -47,6 +47,20 @@ export declare class PregelRunner {
      * @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.