@langchain/langgraph 0.2.61 → 0.2.63

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 {};

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

@@ -0,0 +1,212 @@
+import { jest } from "@jest/globals";
+import { AsyncLocalStorageProviderSingleton } from "@langchain/core/singletons";
+import { ensureLangGraphConfig, getStore, getWriter, getConfig, recastCheckpointNamespace, getParentCheckpointNamespace, } from "./config.js";
+import { CHECKPOINT_NAMESPACE_SEPARATOR, CHECKPOINT_NAMESPACE_END, } from "../../constants.js";
+describe("ensureLangGraphConfig", () => {
+    // Save original to restore after tests
+    const originalGetRunnableConfig = AsyncLocalStorageProviderSingleton.getRunnableConfig;
+    beforeEach(() => {
+        // Reset the mock before each test
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest.fn();
+    });
+    afterAll(() => {
+        // Restore the original after all tests
+        AsyncLocalStorageProviderSingleton.getRunnableConfig =
+            originalGetRunnableConfig;
+    });
+    it("should return a default config when no arguments provided", () => {
+        // Mock the AsyncLocalStorage to return undefined
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue(undefined);
+        const result = 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
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = 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 = 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" },
+        };
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue(asyncLocalStorageConfig);
+        const result = 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", () => {
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config1 = undefined;
+        const config2 = {
+            tags: ["tag2"],
+            metadata: undefined,
+        };
+        const result = ensureLangGraphConfig(config1, config2);
+        expect(result.tags).toEqual(["tag2"]);
+        expect(result.metadata).toEqual({});
+    });
+    it("should copy scalar values to metadata from configurable", () => {
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = 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 = 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", () => {
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue(undefined);
+        const config = {
+            metadata: { key: "original value" },
+            configurable: {
+                key: "should not overwrite",
+            },
+        };
+        const result = ensureLangGraphConfig(config);
+        expect(result.metadata?.key).toEqual("original value");
+    });
+});
+describe("getStore, getWriter, getConfig", () => {
+    // Save original to restore after tests
+    const originalGetRunnableConfig = AsyncLocalStorageProviderSingleton.getRunnableConfig;
+    beforeEach(() => {
+        // Reset the mock before each test
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest.fn();
+    });
+    afterAll(() => {
+        // Restore the original after all tests
+        AsyncLocalStorageProviderSingleton.getRunnableConfig =
+            originalGetRunnableConfig;
+    });
+    it("getStore should return store from config", () => {
+        const mockStore = {};
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue({
+            store: mockStore,
+        });
+        const result = getStore();
+        expect(result).toBe(mockStore);
+    });
+    it("getWriter should return writer from configurable", () => {
+        const mockWriter = () => { };
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue({
+            configurable: {
+                writer: mockWriter,
+            },
+        });
+        const result = getWriter();
+        expect(result).toBe(mockWriter);
+    });
+    it("getConfig should return the full config", () => {
+        const mockConfig = { key: "value" };
+        AsyncLocalStorageProviderSingleton.getRunnableConfig = jest
+            .fn()
+            .mockReturnValue(mockConfig);
+        const result = getConfig();
+        expect(result).toBe(mockConfig);
+    });
+});
+describe("recastCheckpointNamespace", () => {
+    it("should filter out numeric parts of the namespace", () => {
+        const namespace = `parent${CHECKPOINT_NAMESPACE_SEPARATOR}123${CHECKPOINT_NAMESPACE_SEPARATOR}child`;
+        const result = recastCheckpointNamespace(namespace);
+        expect(result).toBe(`parent${CHECKPOINT_NAMESPACE_SEPARATOR}child`);
+    });
+    it("should remove parts after CHECKPOINT_NAMESPACE_END", () => {
+        const namespace = `part1${CHECKPOINT_NAMESPACE_SEPARATOR}part2${CHECKPOINT_NAMESPACE_END}extra`;
+        const result = recastCheckpointNamespace(namespace);
+        expect(result).toBe(`part1${CHECKPOINT_NAMESPACE_SEPARATOR}part2`);
+    });
+    it("should handle complex namespace with numeric parts and CHECKPOINT_NAMESPACE_END", () => {
+        const namespace = `root${CHECKPOINT_NAMESPACE_SEPARATOR}123${CHECKPOINT_NAMESPACE_SEPARATOR}child${CHECKPOINT_NAMESPACE_END}extra${CHECKPOINT_NAMESPACE_SEPARATOR}456`;
+        const result = recastCheckpointNamespace(namespace);
+        expect(result).toBe(`root${CHECKPOINT_NAMESPACE_SEPARATOR}child`);
+    });
+    it("should return the original namespace when no filtering needed", () => {
+        const namespace = `part1${CHECKPOINT_NAMESPACE_SEPARATOR}part2`;
+        const result = recastCheckpointNamespace(namespace);
+        expect(result).toBe(namespace);
+    });
+});
+describe("getParentCheckpointNamespace", () => {
+    it("should return the parent namespace by removing the last part", () => {
+        const namespace = `parent${CHECKPOINT_NAMESPACE_SEPARATOR}child`;
+        const result = getParentCheckpointNamespace(namespace);
+        expect(result).toBe("parent");
+    });
+    it("should skip trailing numeric parts", () => {
+        const namespace = `parent${CHECKPOINT_NAMESPACE_SEPARATOR}child${CHECKPOINT_NAMESPACE_SEPARATOR}123${CHECKPOINT_NAMESPACE_SEPARATOR}456`;
+        const result = getParentCheckpointNamespace(namespace);
+        expect(result).toBe("parent");
+    });
+    it("should return empty string for top-level namespace", () => {
+        const namespace = "singlePart";
+        const result = getParentCheckpointNamespace(namespace);
+        expect(result).toBe("");
+    });
+    it("should handle namespace with mixed numeric and non-numeric parts", () => {
+        const namespace = `root${CHECKPOINT_NAMESPACE_SEPARATOR}sub1${CHECKPOINT_NAMESPACE_SEPARATOR}123${CHECKPOINT_NAMESPACE_SEPARATOR}sub2`;
+        const result = getParentCheckpointNamespace(namespace);
+        // The implementation stops at the first numeric part, not at the last non-numeric part
+        expect(result).toBe(`root${CHECKPOINT_NAMESPACE_SEPARATOR}sub1${CHECKPOINT_NAMESPACE_SEPARATOR}123`);
+    });
+});
+//# sourceMappingURL=config.test.js.map

package/dist/pregel/utils/config.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.test.js","sourceRoot":"","sources":["../../../src/pregel/utils/config.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,kCAAkC,EAAE,MAAM,4BAA4B,CAAC;AAEhF,OAAO,EACL,qBAAqB,EACrB,QAAQ,EACR,SAAS,EACT,SAAS,EACT,yBAAyB,EACzB,4BAA4B,GAC7B,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,8BAA8B,EAC9B,wBAAwB,GACzB,MAAM,oBAAoB,CAAC;AAE5B,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,uCAAuC;IACvC,MAAM,yBAAyB,GAC7B,kCAAkC,CAAC,iBAAiB,CAAC;IAEvD,UAAU,CAAC,GAAG,EAAE;QACd,kCAAkC;QAClC,kCAAkC,CAAC,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,GAAG,EAAE;QACZ,uCAAuC;QACvC,kCAAkC,CAAC,iBAAiB;YAClD,yBAAyB,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,iDAAiD;QACjD,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,SAAS,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG,qBAAqB,EAAE,CAAC;QAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,SAAS;YACpB,cAAc,EAAE,EAAE;YAClB,YAAY,EAAE,EAAE;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,iDAAiD;QACjD,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,SAAS,CAAC,CAAC;QAE9B,MAAM,OAAO,GAAG;YACd,IAAI,EAAE,CAAC,MAAM,CAAC;YACd,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC5B,YAAY,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,IAAI,EAAE,CAAC,MAAM,CAAC;YACd,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC5B,YAAY,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SACpC,CAAC;QAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAEvD,0EAA0E;QAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,CAAC,MAAM,CAAC;YACd,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE;YAC/C,SAAS,EAAE,SAAS;YACpB,cAAc,EAAE,EAAE;YAClB,YAAY,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SACpC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,qCAAqC;QACrC,MAAM,uBAAuB,GAAG;YAC9B,IAAI,EAAE,CAAC,aAAa,CAAC;YACrB,QAAQ,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;YAC9B,SAAS,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,EAAE;YACxD,YAAY,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE;SACzC,CAAC;QAEF,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,uBAAuB,CAAC,CAAC;QAE5C,MAAM,MAAM,GAAG,qBAAqB,EAAE,CAAC;QAEvC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;YACpC,OAAO,EAAE,OAAO;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,SAAS,CAAC,CAAC;QAE9B,MAAM,OAAO,GAAG,SAAS,CAAC;QAC1B,MAAM,OAAO,GAAG;YACd,IAAI,EAAE,CAAC,MAAM,CAAC;YACd,QAAQ,EAAE,SAAS;SACpB,CAAC;QAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,SAAS,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG;YACb,YAAY,EAAE;gBACZ,WAAW,EAAE,QAAQ;gBACrB,WAAW,EAAE,EAAE;gBACf,YAAY,EAAE,IAAI;gBAClB,WAAW,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE;gBACxC,cAAc,EAAE,sBAAsB;aACvC;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAC9B,WAAW,EAAE,QAAQ;YACrB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,IAAI;YAClB,sDAAsD;SACvD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;QAChF,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,SAAS,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,EAAE,GAAG,EAAE,gBAAgB,EAAE;YACnC,YAAY,EAAE;gBACZ,GAAG,EAAE,sBAAsB;aAC5B;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,uCAAuC;IACvC,MAAM,yBAAyB,GAC7B,kCAAkC,CAAC,iBAAiB,CAAC;IAEvD,UAAU,CAAC,GAAG,EAAE;QACd,kCAAkC;QAClC,kCAAkC,CAAC,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,GAAG,EAAE;QACZ,uCAAuC;QACvC,kCAAkC,CAAC,iBAAiB;YAClD,yBAAyB,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,SAAS,GAAG,EAAe,CAAC;QAClC,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC;YACf,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QAEL,MAAM,MAAM,GAAG,QAAQ,EAAE,CAAC;QAE1B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,UAAU,GAAG,GAAG,EAAE,GAAE,CAAC,CAAC;QAC5B,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC;YACf,YAAY,EAAE;gBACZ,MAAM,EAAE,UAAU;aACnB;SACF,CAAC,CAAC;QAEL,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAE3B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,UAAU,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;QACpC,kCAAkC,CAAC,iBAAiB,GAAG,IAAI;aACxD,EAAE,EAAE;aACJ,eAAe,CAAC,UAAU,CAAC,CAAC;QAE/B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAE3B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,SAAS,GAAG,SAAS,8BAA8B,MAAM,8BAA8B,OAAO,CAAC;QACrG,MAAM,MAAM,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,8BAA8B,OAAO,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,SAAS,GAAG,QAAQ,8BAA8B,QAAQ,wBAAwB,OAAO,CAAC;QAChG,MAAM,MAAM,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,8BAA8B,OAAO,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;QACzF,MAAM,SAAS,GAAG,OAAO,8BAA8B,MAAM,8BAA8B,QAAQ,wBAAwB,QAAQ,8BAA8B,KAAK,CAAC;QACvK,MAAM,MAAM,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,8BAA8B,OAAO,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,SAAS,GAAG,QAAQ,8BAA8B,OAAO,CAAC;QAChE,MAAM,MAAM,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,SAAS,GAAG,SAAS,8BAA8B,OAAO,CAAC;QACjE,MAAM,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,SAAS,GAAG,SAAS,8BAA8B,QAAQ,8BAA8B,MAAM,8BAA8B,KAAK,CAAC;QACzI,MAAM,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,SAAS,GAAG,YAAY,CAAC;QAC/B,MAAM,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,SAAS,GAAG,OAAO,8BAA8B,OAAO,8BAA8B,MAAM,8BAA8B,MAAM,CAAC;QACvI,MAAM,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAC;QAEvD,uFAAuF;QACvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACjB,OAAO,8BAA8B,OAAO,8BAA8B,KAAK,CAChF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/utils/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.patchCheckpointMap = exports.patchConfigurable = exports._coerceToDict = exports.getNewChannelVersions = exports.getNullChannelVersion = void 0;
+exports.combineAbortSignals = exports.patchCheckpointMap = exports.patchConfigurable = exports._coerceToDict = exports.getNewChannelVersions = exports.getNullChannelVersion = void 0;
 const constants_js_1 = require("../../constants.cjs");
 function getNullChannelVersion(currentVersions) {
     const versionValues = Object.values(currentVersions);
@@ -69,4 +69,29 @@ function patchCheckpointMap(config, metadata) {
     }
 }
 exports.patchCheckpointMap = patchCheckpointMap;
+/**
+ * Combine multiple abort signals into a single abort signal.
+ * @param signals - The abort signals to combine.
+ * @returns A single abort signal that is aborted if any of the input signals are aborted.
+ */
+function combineAbortSignals(...signals) {
+    if (signals.length === 1) {
+        return signals[0];
+    }
+    if ("any" in AbortSignal) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return AbortSignal.any(signals);
+    }
+    const combinedController = new AbortController();
+    const listener = () => {
+        combinedController.abort();
+        signals.forEach((s) => s.removeEventListener("abort", listener));
+    };
+    signals.forEach((s) => s.addEventListener("abort", listener));
+    if (signals.some((s) => s.aborted)) {
+        combinedController.abort();
+    }
+    return combinedController.signal;
+}
+exports.combineAbortSignals = combineAbortSignals;
 //# sourceMappingURL=index.js.map

package/dist/pregel/utils/index.d.ts

@@ -33,3 +33,9 @@ export type RetryPolicy = {
 };
 export declare function patchConfigurable(config: RunnableConfig | undefined, patch: Record<string, any>): RunnableConfig;
 export declare function patchCheckpointMap(config: RunnableConfig, metadata?: CheckpointMetadata): RunnableConfig;
+/**
+ * Combine multiple abort signals into a single abort signal.
+ * @param signals - The abort signals to combine.
+ * @returns A single abort signal that is aborted if any of the input signals are aborted.
+ */
+export declare function combineAbortSignals(...signals: AbortSignal[]): AbortSignal;

package/dist/pregel/utils/index.js

@@ -61,4 +61,28 @@ export function patchCheckpointMap(config, metadata) {
         return config;
     }
 }
+/**
+ * Combine multiple abort signals into a single abort signal.
+ * @param signals - The abort signals to combine.
+ * @returns A single abort signal that is aborted if any of the input signals are aborted.
+ */
+export function combineAbortSignals(...signals) {
+    if (signals.length === 1) {
+        return signals[0];
+    }
+    if ("any" in AbortSignal) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return AbortSignal.any(signals);
+    }
+    const combinedController = new AbortController();
+    const listener = () => {
+        combinedController.abort();
+        signals.forEach((s) => s.removeEventListener("abort", listener));
+    };
+    signals.forEach((s) => s.addEventListener("abort", listener));
+    if (signals.some((s) => s.aborted)) {
+        combinedController.abort();
+    }
+    return combinedController.signal;
+}
 //# sourceMappingURL=index.js.map

package/dist/pregel/utils/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/pregel/utils/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,MAAM,UAAU,qBAAqB,CAAC,eAAgC;IACpE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IACrD,MAAM,WAAW,GACf,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjE,IAAI,WAAwC,CAAC;IAC7C,IAAI,WAAW,KAAK,QAAQ,EAAE;QAC5B,WAAW,GAAG,CAAC,CAAC;KACjB;SAAM,IAAI,WAAW,KAAK,QAAQ,EAAE;QACnC,WAAW,GAAG,EAAE,CAAC;KAClB;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,gBAAiC,EACjC,eAAgC;IAEhC,2BAA2B;IAC3B,IAAI,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;QAC5C,MAAM,WAAW,GAAG,qBAAqB,CAAC,eAAe,CAAC,CAAC;QAC3D,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,MAAM,CACpC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,CACrD,CACF,CAAC;KACH;SAAM;QACL,OAAO,eAAe,CAAC;KACxB;AACH,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,aAAa,CAAC,KAAU,EAAE,UAAkB;IAC1D,OAAO,KAAK;QACV,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACrB,uDAAuD;QACvD,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC;QACxB,OAAO,KAAK,KAAK,QAAQ;QACzB,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC;AAC9B,CAAC;AAiCD,MAAM,UAAU,iBAAiB,CAC/B,MAAkC;AAClC,8DAA8D;AAC9D,KAA0B;IAE1B,IAAI,MAAM,KAAK,IAAI,EAAE;QACnB,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;KAChC;SAAM,IAAI,MAAM,EAAE,YAAY,KAAK,SAAS,EAAE;QAC7C,OAAO,EAAE,GAAG,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;KAC3C;SAAM;QACL,OAAO;YACL,GAAG,MAAM;YACT,YAAY,EAAE,EAAE,GAAG,MAAM,CAAC,YAAY,EAAE,GAAG,KAAK,EAAE;SACnD,CAAC;KACH;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAsB,EACtB,QAA6B;IAE7B,MAAM,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;IAExC,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;QACnC,OAAO,iBAAiB,CAAC,MAAM,EAAE;YAC/B,CAAC,yBAAyB,CAAC,EAAE;gBAC3B,GAAG,OAAO;gBACV,CAAC,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,CAAC,EACxC,MAAM,CAAC,YAAY,EAAE,aAAa;aACrC;SACF,CAAC,CAAC;KACJ;SAAM;QACL,OAAO,MAAM,CAAC;KACf;AACH,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/pregel/utils/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,MAAM,UAAU,qBAAqB,CAAC,eAAgC;IACpE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IACrD,MAAM,WAAW,GACf,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACjE,IAAI,WAAwC,CAAC;IAC7C,IAAI,WAAW,KAAK,QAAQ,EAAE;QAC5B,WAAW,GAAG,CAAC,CAAC;KACjB;SAAM,IAAI,WAAW,KAAK,QAAQ,EAAE;QACnC,WAAW,GAAG,EAAE,CAAC;KAClB;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,gBAAiC,EACjC,eAAgC;IAEhC,2BAA2B;IAC3B,IAAI,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;QAC5C,MAAM,WAAW,GAAG,qBAAqB,CAAC,eAAe,CAAC,CAAC;QAC3D,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,MAAM,CACpC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,CACrD,CACF,CAAC;KACH;SAAM;QACL,OAAO,eAAe,CAAC;KACxB;AACH,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,aAAa,CAAC,KAAU,EAAE,UAAkB;IAC1D,OAAO,KAAK;QACV,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACrB,uDAAuD;QACvD,CAAC,CAAC,KAAK,YAAY,IAAI,CAAC;QACxB,OAAO,KAAK,KAAK,QAAQ;QACzB,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC;AAC9B,CAAC;AAiCD,MAAM,UAAU,iBAAiB,CAC/B,MAAkC;AAClC,8DAA8D;AAC9D,KAA0B;IAE1B,IAAI,MAAM,KAAK,IAAI,EAAE;QACnB,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;KAChC;SAAM,IAAI,MAAM,EAAE,YAAY,KAAK,SAAS,EAAE;QAC7C,OAAO,EAAE,GAAG,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;KAC3C;SAAM;QACL,OAAO;YACL,GAAG,MAAM;YACT,YAAY,EAAE,EAAE,GAAG,MAAM,CAAC,YAAY,EAAE,GAAG,KAAK,EAAE;SACnD,CAAC;KACH;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAsB,EACtB,QAA6B;IAE7B,MAAM,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;IAExC,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;QACnC,OAAO,iBAAiB,CAAC,MAAM,EAAE;YAC/B,CAAC,yBAAyB,CAAC,EAAE;gBAC3B,GAAG,OAAO;gBACV,CAAC,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,CAAC,EACxC,MAAM,CAAC,YAAY,EAAE,aAAa;aACrC;SACF,CAAC,CAAC;KACJ;SAAM;QACL,OAAO,MAAM,CAAC;KACf;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,GAAG,OAAsB;IAC3D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;QACxB,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;KACnB;IAED,IAAI,KAAK,IAAI,WAAW,EAAE;QACxB,8DAA8D;QAC9D,OAAQ,WAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;KAC1C;IACD,MAAM,kBAAkB,GAAG,IAAI,eAAe,EAAE,CAAC;IACjD,MAAM,QAAQ,GAAG,GAAG,EAAE;QACpB,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAC3B,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC;IAEF,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;IAE9D,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE;QAClC,kBAAkB,CAAC,KAAK,EAAE,CAAC;KAC5B;IAED,OAAO,kBAAkB,CAAC,MAAM,CAAC;AACnC,CAAC"}

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

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const subgraph_js_1 = require("./subgraph.cjs");
+(0, globals_1.describe)("isPregelLike", () => {
+    (0, globals_1.it)("should return true for objects with lg_is_pregel=true", () => {
+        const mockPregelObj = {
+            lg_is_pregel: true,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        // Cast to any to test just the logic, not the type constraints
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (0, globals_1.expect)((0, subgraph_js_1.isPregelLike)(mockPregelObj)).toBe(true);
+    });
+    (0, globals_1.it)("should return false for objects without lg_is_pregel property", () => {
+        const nonPregelObj = {
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        // Cast to any to test just the logic, not the type constraints
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (0, globals_1.expect)((0, subgraph_js_1.isPregelLike)(nonPregelObj)).toBe(false);
+    });
+    (0, globals_1.it)("should return false for objects with lg_is_pregel=false", () => {
+        const nonPregelObj = {
+            lg_is_pregel: false,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        // Cast to any to test just the logic, not the type constraints
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (0, globals_1.expect)((0, subgraph_js_1.isPregelLike)(nonPregelObj)).toBe(false);
+    });
+});
+(0, globals_1.describe)("findSubgraphPregel", () => {
+    (0, globals_1.it)("should find Pregel object at the top level", () => {
+        const mockPregelObj = {
+            lg_is_pregel: true,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        // Cast to Runnable to test the behavior
+        (0, globals_1.expect)((0, subgraph_js_1.findSubgraphPregel)(mockPregelObj)).toBe(mockPregelObj);
+    });
+    (0, globals_1.it)("should find Pregel object in a RunnableSequence", () => {
+        const mockPregelObj = {
+            lg_is_pregel: true,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        const mockSequence = {
+            steps: [{ someProperty: "value", invoke: () => { } }, mockPregelObj],
+        };
+        (0, globals_1.expect)((0, subgraph_js_1.findSubgraphPregel)(mockSequence)).toBe(mockPregelObj);
+    });
+    (0, globals_1.it)("should find Pregel object in a nested RunnableSequence", () => {
+        const mockPregelObj = {
+            lg_is_pregel: true,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        const innerSequence = {
+            steps: [{ someProperty: "value", invoke: () => { } }, mockPregelObj],
+        };
+        const outerSequence = {
+            steps: [{ someProperty: "otherValue", invoke: () => { } }, innerSequence],
+        };
+        (0, globals_1.expect)((0, subgraph_js_1.findSubgraphPregel)(outerSequence)).toBe(mockPregelObj);
+    });
+    (0, globals_1.it)("should return undefined if no Pregel object is found", () => {
+        const nonPregelRunnable = {
+            someProperty: "value",
+            invoke: () => { },
+        };
+        const sequence = {
+            steps: [{ someProperty: "value1" }, { someProperty: "value2" }],
+        };
+        (0, globals_1.expect)((0, subgraph_js_1.findSubgraphPregel)(nonPregelRunnable)).toBeUndefined();
+        (0, globals_1.expect)((0, subgraph_js_1.findSubgraphPregel)(sequence)).toBeUndefined();
+    });
+});
+//# sourceMappingURL=subgraph.test.js.map

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

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