@langchain/langgraph 0.2.62 → 0.2.63

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

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

@@ -0,0 +1,81 @@
+import { describe, expect, it } from "@jest/globals";
+import { isPregelLike, findSubgraphPregel } from "./subgraph.js";
+describe("isPregelLike", () => {
+    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
+        expect(isPregelLike(mockPregelObj)).toBe(true);
+    });
+    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
+        expect(isPregelLike(nonPregelObj)).toBe(false);
+    });
+    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
+        expect(isPregelLike(nonPregelObj)).toBe(false);
+    });
+});
+describe("findSubgraphPregel", () => {
+    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
+        expect(findSubgraphPregel(mockPregelObj)).toBe(mockPregelObj);
+    });
+    it("should find Pregel object in a RunnableSequence", () => {
+        const mockPregelObj = {
+            lg_is_pregel: true,
+            invoke: () => { },
+            someOtherProp: "value",
+        };
+        const mockSequence = {
+            steps: [{ someProperty: "value", invoke: () => { } }, mockPregelObj],
+        };
+        expect(findSubgraphPregel(mockSequence)).toBe(mockPregelObj);
+    });
+    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],
+        };
+        expect(findSubgraphPregel(outerSequence)).toBe(mockPregelObj);
+    });
+    it("should return undefined if no Pregel object is found", () => {
+        const nonPregelRunnable = {
+            someProperty: "value",
+            invoke: () => { },
+        };
+        const sequence = {
+            steps: [{ someProperty: "value1" }, { someProperty: "value2" }],
+        };
+        expect(findSubgraphPregel(nonPregelRunnable)).toBeUndefined();
+        expect(findSubgraphPregel(sequence)).toBeUndefined();
+    });
+});
+//# sourceMappingURL=subgraph.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"subgraph.test.js","sourceRoot":"","sources":["../../../src/pregel/utils/subgraph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAErD,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEjE,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,IAAI;YAClB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,+DAA+D;QAC/D,8DAA8D;QAC9D,MAAM,CAAC,YAAY,CAAC,aAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,+DAA+D;QAC/D,8DAA8D;QAC9D,MAAM,CAAC,YAAY,CAAC,YAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,YAAY,GAAG;YACnB,YAAY,EAAE,KAAK;YACnB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,+DAA+D;QAC/D,8DAA8D;QAC9D,MAAM,CAAC,YAAY,CAAC,YAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,IAAI;YAClB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,wCAAwC;QACxC,MAAM,CAAC,kBAAkB,CAAC,aAAoC,CAAC,CAAC,CAAC,IAAI,CACnE,aAAa,CACd,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,IAAI;YAClB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,KAAK,EAAE,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAAE,aAAa,CAAC;SACpE,CAAC;QAEF,MAAM,CAAC,kBAAkB,CAAC,YAAmC,CAAC,CAAC,CAAC,IAAI,CAClE,aAAa,CACd,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,IAAI;YAClB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;YAChB,aAAa,EAAE,OAAO;SACvB,CAAC;QAEF,MAAM,aAAa,GAAG;YACpB,KAAK,EAAE,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAAE,aAAa,CAAC;SACpE,CAAC;QAEF,MAAM,aAAa,GAAG;YACpB,KAAK,EAAE,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAAE,aAAa,CAAC;SACzE,CAAC;QAEF,MAAM,CAAC,kBAAkB,CAAC,aAAoC,CAAC,CAAC,CAAC,IAAI,CACnE,aAAa,CACd,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,iBAAiB,GAAG;YACxB,YAAY,EAAE,OAAO;YACrB,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;SACjB,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,KAAK,EAAE,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;SAChE,CAAC;QAEF,MAAM,CACJ,kBAAkB,CAAC,iBAAwC,CAAC,CAC7D,CAAC,aAAa,EAAE,CAAC;QAClB,MAAM,CAAC,kBAAkB,CAAC,QAA+B,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC9E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/validate.test.cjs

@@ -0,0 +1,220 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const validate_js_1 = require("./validate.cjs");
+const read_js_1 = require("./read.cjs");
+const constants_js_1 = require("../constants.cjs");
+const last_value_js_1 = require("../channels/last_value.cjs");
+// Common test setup
+const setupValidGraph = () => {
+    // Create test channels
+    const inputChannel = new last_value_js_1.LastValue();
+    const outputChannel = new last_value_js_1.LastValue();
+    // Create test nodes
+    const testNode = new read_js_1.PregelNode({
+        channels: {},
+        triggers: ["input"],
+    });
+    return {
+        nodes: { testNode },
+        channels: {
+            input: inputChannel,
+            output: outputChannel,
+        },
+        inputChannels: "input",
+        outputChannels: "output",
+    };
+};
+(0, globals_1.describe)("GraphValidationError", () => {
+    (0, globals_1.it)("should be properly constructed with the right name", () => {
+        const error = new validate_js_1.GraphValidationError("Test error message");
+        (0, globals_1.expect)(error).toBeInstanceOf(Error);
+        (0, globals_1.expect)(error.name).toBe("GraphValidationError");
+        (0, globals_1.expect)(error.message).toBe("Test error message");
+    });
+});
+(0, globals_1.describe)("validateGraph", () => {
+    (0, globals_1.it)("should validate a correct graph without errors", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should not throw
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+        })).not.toThrow();
+    });
+    (0, globals_1.it)("should throw when channels are not provided", () => {
+        const { nodes, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should throw with specific message
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels: null,
+            inputChannels,
+            outputChannels,
+        })).toThrow("Channels not provided");
+    });
+    (0, globals_1.it)("should throw when a node is named INTERRUPT", () => {
+        const { channels, inputChannels, outputChannels } = setupValidGraph();
+        // Create a node with the reserved name
+        const badNode = new read_js_1.PregelNode({
+            channels: {},
+            triggers: ["input"],
+        });
+        // Create nodes object with the reserved name
+        const nodes = { [constants_js_1.INTERRUPT]: badNode };
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+        })).toThrow(`"Node name ${constants_js_1.INTERRUPT} is reserved"`);
+    });
+    (0, globals_1.it)("should throw when a node is not a PregelNode instance", () => {
+        const { channels, inputChannels, outputChannels } = setupValidGraph();
+        // Create an invalid node (not a PregelNode)
+        const badNode = {
+            triggers: ["input"],
+            func: () => "not a pregel node",
+        };
+        // Create nodes object with invalid node
+        const nodes = {
+            badNode: badNode,
+        };
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+        })).toThrow("Invalid node type object, expected PregelNode");
+    });
+    (0, globals_1.it)("should throw when a subscribed channel is not in channels", () => {
+        const { channels, inputChannels, outputChannels } = setupValidGraph();
+        // Create a node that subscribes to a non-existent channel
+        const badNode = new read_js_1.PregelNode({
+            channels: {},
+            triggers: ["nonexistent"],
+        });
+        const nodes = { badNode };
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+        })).toThrow("Subcribed channel 'nonexistent' not in channels");
+    });
+    (0, globals_1.it)("should throw when a singular input channel is not subscribed by any node", () => {
+        const { nodes, channels } = setupValidGraph();
+        // Act & Assert: Should throw specific error for an unused input
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels: "output",
+            outputChannels: "output",
+        })).toThrow("Input channel output is not subscribed to by any node");
+    });
+    (0, globals_1.it)("should throw when none of the array input channels are subscribed by any node", () => {
+        const { nodes, channels } = setupValidGraph();
+        // Act & Assert: Should throw specific error for unused inputs
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels: [
+                "output",
+                "nonexistent",
+            ],
+            outputChannels: "output",
+        })).toThrow("None of the input channels output,nonexistent are subscribed to by any node");
+    });
+    (0, globals_1.it)("should throw when an output channel is not in channels", () => {
+        const { nodes, channels, inputChannels } = setupValidGraph();
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels: "nonexistent",
+        })).toThrow("Output channel 'nonexistent' not in channels");
+    });
+    (0, globals_1.it)("should throw when a stream channel is not in channels", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+            streamChannels: "nonexistent",
+        })).toThrow("Output channel 'nonexistent' not in channels");
+    });
+    (0, globals_1.it)("should throw when an interruptAfterNode is not in nodes", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+            interruptAfterNodes: [
+                "nonexistentNode",
+            ],
+        })).toThrow("Node nonexistentNode not in nodes");
+    });
+    (0, globals_1.it)("should throw when an interruptBeforeNode is not in nodes", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should throw specific error
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+            interruptBeforeNodes: [
+                "nonexistentNode",
+            ],
+        })).toThrow("Node nonexistentNode not in nodes");
+    });
+    (0, globals_1.it)("should accept '*' as a valid value for interruptAfterNodes", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should not throw
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+            interruptAfterNodes: "*",
+        })).not.toThrow();
+    });
+    (0, globals_1.it)("should accept '*' as a valid value for interruptBeforeNodes", () => {
+        const { nodes, channels, inputChannels, outputChannels } = setupValidGraph();
+        // Act & Assert: Should not throw
+        (0, globals_1.expect)(() => (0, validate_js_1.validateGraph)({
+            nodes,
+            channels,
+            inputChannels,
+            outputChannels,
+            interruptBeforeNodes: "*",
+        })).not.toThrow();
+    });
+});
+(0, globals_1.describe)("validateKeys", () => {
+    (0, globals_1.it)("should validate keys that exist in channels", () => {
+        const { channels } = setupValidGraph();
+        // Act & Assert: Should not throw for valid keys
+        (0, globals_1.expect)(() => (0, validate_js_1.validateKeys)("input", channels)).not.toThrow();
+        (0, globals_1.expect)(() => (0, validate_js_1.validateKeys)(["input", "output"], channels)).not.toThrow();
+    });
+    (0, globals_1.it)("should throw when a single key doesn't exist in channels", () => {
+        const { channels } = setupValidGraph();
+        // Act & Assert: Should throw with specific message
+        (0, globals_1.expect)(() => (0, validate_js_1.validateKeys)("nonexistent", channels)).toThrow("Key nonexistent not found in channels");
+    });
+    (0, globals_1.it)("should throw when any key in an array doesn't exist in channels", () => {
+        const { channels } = setupValidGraph();
+        // Act & Assert: Should throw with specific message
+        (0, globals_1.expect)(() => (0, validate_js_1.validateKeys)(["input", "nonexistent"], channels)).toThrow("Key nonexistent not found in channels");
+    });
+});
+//# sourceMappingURL=validate.test.js.map

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

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