@langchain/langgraph 0.4.9 → 1.0.0-alpha.1

package/dist/pregel/io.mapCommand.test.cjs

@@ -1,150 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const vitest_1 = require("vitest");
-const constants_js_1 = require("../constants.cjs");
-const io_js_1 = require("./io.cjs");
-const errors_js_1 = require("../errors.cjs");
-(0, vitest_1.describe)("mapCommand", () => {
-    (0, vitest_1.it)("should handle Command with goto (string)", () => {
-        const cmd = new constants_js_1.Command({
-            goto: "nextNode",
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode",
-                "__start__",
-            ],
-        ]);
-    });
-    (0, vitest_1.it)("should handle Command with goto (Send object)", () => {
-        const send = new constants_js_1.Send("targetNode", { arg1: "value1" });
-        const cmd = new constants_js_1.Command({
-            goto: send,
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
-        ]);
-    });
-    (0, vitest_1.it)("should handle Command with goto (array of strings and Send objects)", () => {
-        const send = new constants_js_1.Send("targetNode", { arg1: "value1" });
-        const cmd = new constants_js_1.Command({
-            goto: ["nextNode1", send, "nextNode2"],
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode1",
-                "__start__",
-            ],
-            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode2",
-                "__start__",
-            ],
-        ]);
-    });
-    (0, vitest_1.it)("should throw error for invalid goto value", () => {
-        const cmd = new constants_js_1.Command({
-            // @ts-expect-error Testing invalid input
-            goto: { invalidType: true },
-        });
-        const pendingWrites = [];
-        (0, vitest_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("In Command.send, expected Send or string, got object");
-    });
-    (0, vitest_1.it)("should handle Command with resume (single value)", () => {
-        const cmd = new constants_js_1.Command({
-            resume: "resumeValue",
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
-        ]);
-    });
-    (0, vitest_1.it)("should handle Command with resume (object of interrupt IDs)", () => {
-        const cmd = new constants_js_1.Command({
-            resume: {
-                "123e4567e89b12d3a456426614174000": "resumeValue1",
-                "123e4567e89b12d3a456426614174001": "resumeValue2",
-            },
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            ["123e4567e89b12d3a456426614174000", "__resume__", ["resumeValue1"]],
-            ["123e4567e89b12d3a456426614174001", "__resume__", ["resumeValue2"]],
-        ]);
-    });
-    (0, vitest_1.it)("should handle Command with update (object)", () => {
-        const cmd = new constants_js_1.Command({
-            update: {
-                channel1: "value1",
-                channel2: "value2",
-            },
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
-        ]);
-    });
-    (0, vitest_1.it)("should handle Command with update (array of tuples)", () => {
-        const cmd = new constants_js_1.Command({
-            update: [
-                ["channel1", "value1"],
-                ["channel2", "value2"],
-            ],
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
-        ]);
-    });
-    (0, vitest_1.it)("should throw error for invalid update type", () => {
-        const cmd = new constants_js_1.Command({
-            // @ts-expect-error Testing invalid input
-            update: "invalidUpdateType",
-        });
-        const pendingWrites = [];
-        (0, vitest_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("Expected cmd.update to be a dict mapping channel names to update values");
-    });
-    (0, vitest_1.it)("should throw error for parent graph reference when none exists", () => {
-        const cmd = new constants_js_1.Command({
-            graph: constants_js_1.Command.PARENT,
-            goto: "nextNode",
-        });
-        const pendingWrites = [];
-        (0, vitest_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow(errors_js_1.InvalidUpdateError);
-        (0, vitest_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("There is no parent graph.");
-    });
-    (0, vitest_1.it)("should handle multiple command attributes together", () => {
-        const cmd = new constants_js_1.Command({
-            goto: "nextNode",
-            resume: "resumeValue",
-            update: { channel1: "value1" },
-        });
-        const pendingWrites = [];
-        const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, vitest_1.expect)(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode",
-                "__start__",
-            ],
-            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-        ]);
-    });
-});
-//# sourceMappingURL=io.mapCommand.test.js.map

package/dist/pregel/io.mapCommand.test.d.ts

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

package/dist/pregel/io.mapCommand.test.js

@@ -1,148 +0,0 @@
-import { describe, expect, it } from "vitest";
-import { Command, Send } from "../constants.js";
-import { mapCommand } from "./io.js";
-import { InvalidUpdateError } from "../errors.js";
-describe("mapCommand", () => {
-    it("should handle Command with goto (string)", () => {
-        const cmd = new Command({
-            goto: "nextNode",
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode",
-                "__start__",
-            ],
-        ]);
-    });
-    it("should handle Command with goto (Send object)", () => {
-        const send = new Send("targetNode", { arg1: "value1" });
-        const cmd = new Command({
-            goto: send,
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
-        ]);
-    });
-    it("should handle Command with goto (array of strings and Send objects)", () => {
-        const send = new Send("targetNode", { arg1: "value1" });
-        const cmd = new Command({
-            goto: ["nextNode1", send, "nextNode2"],
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode1",
-                "__start__",
-            ],
-            ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode2",
-                "__start__",
-            ],
-        ]);
-    });
-    it("should throw error for invalid goto value", () => {
-        const cmd = new Command({
-            // @ts-expect-error Testing invalid input
-            goto: { invalidType: true },
-        });
-        const pendingWrites = [];
-        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("In Command.send, expected Send or string, got object");
-    });
-    it("should handle Command with resume (single value)", () => {
-        const cmd = new Command({
-            resume: "resumeValue",
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
-        ]);
-    });
-    it("should handle Command with resume (object of interrupt IDs)", () => {
-        const cmd = new Command({
-            resume: {
-                "123e4567e89b12d3a456426614174000": "resumeValue1",
-                "123e4567e89b12d3a456426614174001": "resumeValue2",
-            },
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            ["123e4567e89b12d3a456426614174000", "__resume__", ["resumeValue1"]],
-            ["123e4567e89b12d3a456426614174001", "__resume__", ["resumeValue2"]],
-        ]);
-    });
-    it("should handle Command with update (object)", () => {
-        const cmd = new Command({
-            update: {
-                channel1: "value1",
-                channel2: "value2",
-            },
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
-        ]);
-    });
-    it("should handle Command with update (array of tuples)", () => {
-        const cmd = new Command({
-            update: [
-                ["channel1", "value1"],
-                ["channel2", "value2"],
-            ],
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-            ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
-        ]);
-    });
-    it("should throw error for invalid update type", () => {
-        const cmd = new Command({
-            // @ts-expect-error Testing invalid input
-            update: "invalidUpdateType",
-        });
-        const pendingWrites = [];
-        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("Expected cmd.update to be a dict mapping channel names to update values");
-    });
-    it("should throw error for parent graph reference when none exists", () => {
-        const cmd = new Command({
-            graph: Command.PARENT,
-            goto: "nextNode",
-        });
-        const pendingWrites = [];
-        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow(InvalidUpdateError);
-        expect(() => Array.from(mapCommand(cmd, pendingWrites))).toThrow("There is no parent graph.");
-    });
-    it("should handle multiple command attributes together", () => {
-        const cmd = new Command({
-            goto: "nextNode",
-            resume: "resumeValue",
-            update: { channel1: "value1" },
-        });
-        const pendingWrites = [];
-        const result = Array.from(mapCommand(cmd, pendingWrites));
-        expect(result).toEqual([
-            [
-                "00000000-0000-0000-0000-000000000000",
-                "branch:to:nextNode",
-                "__start__",
-            ],
-            ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
-            ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
-        ]);
-    });
-});
-//# sourceMappingURL=io.mapCommand.test.js.map

package/dist/pregel/io.mapCommand.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"io.mapCommand.test.js","sourceRoot":"","sources":["../../src/pregel/io.mapCommand.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,oBAAoB;gBACpB,WAAW;aACZ;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,gBAAgB,EAAE,IAAI,CAAC;SACjE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,EAAE,WAAW,CAAC;SACvC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,qBAAqB;gBACrB,WAAW;aACZ;YACD,CAAC,sCAAsC,EAAE,gBAAgB,EAAE,IAAI,CAAC;YAChE;gBACE,sCAAsC;gBACtC,qBAAqB;gBACrB,WAAW;aACZ;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,yCAAyC;YACzC,IAAI,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;SAC5B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,sDAAsD,CACvD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,YAAY,EAAE,aAAa,CAAC;SACtE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,kCAAkC,EAAE,cAAc;gBAClD,kCAAkC,EAAE,cAAc;aACnD;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,kCAAkC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;YACpE,CAAC,kCAAkC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;SACrE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,QAAQ,EAAE,QAAQ;aACnB;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;YAC9D,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,CAAC,UAAU,EAAE,QAAQ,CAAC;gBACtB,CAAC,UAAU,EAAE,QAAQ,CAAC;aACvB;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;YAC9D,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,yCAAyC;YACzC,MAAM,EAAE,mBAAmB;SAC5B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,yEAAyE,CAC1E,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,KAAK,EAAE,OAAO,CAAC,MAAM;YACrB,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,kBAAkB,CACnB,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,2BAA2B,CAC5B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,IAAI,EAAE,UAAU;YAChB,MAAM,EAAE,aAAa;YACrB,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE;SAC/B,CAAC,CAAC;QAEH,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,sCAAsC;gBACtC,oBAAoB;gBACpB,WAAW;aACZ;YACD,CAAC,sCAAsC,EAAE,YAAY,EAAE,aAAa,CAAC;YACrE,CAAC,sCAAsC,EAAE,UAAU,EAAE,QAAQ,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/pregel/loop.d.ts

@@ -1,147 +0,0 @@
-import type { RunnableConfig } from "@langchain/core/runnables";
-import type { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All, BaseStore, AsyncBatchedStore, BaseCache, CacheFullKey, CacheNamespace } from "@langchain/langgraph-checkpoint";
-import { BaseChannel } from "../channels/base.js";
-import type { Call, Durability, PregelExecutableTask, StreamMode } from "./types.js";
-import { Command } from "../constants.js";
-import { PregelNode } from "./read.js";
-import { LangGraphRunnableConfig } from "./runnable_types.js";
-import { IterableReadableWritableStream } from "./stream.js";
-export type PregelLoopInitializeParams = {
-    input?: any | Command;
-    config: RunnableConfig;
-    checkpointer?: BaseCheckpointSaver;
-    outputKeys: string | string[];
-    streamKeys: string | string[];
-    nodes: Record<string, PregelNode>;
-    channelSpecs: Record<string, BaseChannel>;
-    stream: IterableReadableWritableStream;
-    store?: BaseStore;
-    cache?: BaseCache<PendingWrite<string>[]>;
-    interruptAfter: string[] | All;
-    interruptBefore: string[] | All;
-    durability: Durability;
-    manager?: CallbackManagerForChainRun;
-    debug: boolean;
-    triggerToNodes: Record<string, string[]>;
-};
-type PregelLoopParams = {
-    input?: any | Command;
-    config: RunnableConfig;
-    checkpointer?: BaseCheckpointSaver;
-    checkpoint: Checkpoint;
-    checkpointMetadata: CheckpointMetadata;
-    checkpointPreviousVersions: Record<string, string | number>;
-    checkpointPendingWrites: CheckpointPendingWrite[];
-    checkpointConfig: RunnableConfig;
-    channels: Record<string, BaseChannel>;
-    step: number;
-    stop: number;
-    outputKeys: string | string[];
-    streamKeys: string | string[];
-    nodes: Record<string, PregelNode>;
-    checkpointNamespace: string[];
-    skipDoneTasks: boolean;
-    isNested: boolean;
-    manager?: CallbackManagerForChainRun;
-    stream: IterableReadableWritableStream;
-    store?: AsyncBatchedStore;
-    cache?: BaseCache<PendingWrite<string>[]>;
-    prevCheckpointConfig: RunnableConfig | undefined;
-    interruptAfter: string[] | All;
-    interruptBefore: string[] | All;
-    durability: Durability;
-    debug: boolean;
-    triggerToNodes: Record<string, string[]>;
-};
-declare class AsyncBatchedCache extends BaseCache<PendingWrite<string>[]> {
-    protected cache: BaseCache<PendingWrite<string>[]>;
-    private queue;
-    constructor(cache: BaseCache<unknown>);
-    get(keys: CacheFullKey[]): Promise<{
-        key: CacheFullKey;
-        value: PendingWrite<string>[];
-    }[]>;
-    set(pairs: {
-        key: CacheFullKey;
-        value: PendingWrite<string>[];
-        ttl?: number;
-    }[]): Promise<void>;
-    clear(namespaces: CacheNamespace[]): Promise<void>;
-    stop(): Promise<void>;
-    private enqueueOperation;
-}
-export declare class PregelLoop {
-    protected input?: any | Command;
-    output: any;
-    config: LangGraphRunnableConfig;
-    protected checkpointer?: BaseCheckpointSaver;
-    protected checkpointerGetNextVersion: (current: number | undefined) => number;
-    channels: Record<string, BaseChannel>;
-    protected checkpoint: Checkpoint;
-    protected checkpointIdSaved: string | undefined;
-    protected checkpointConfig: RunnableConfig;
-    checkpointMetadata: CheckpointMetadata;
-    protected checkpointNamespace: string[];
-    protected checkpointPendingWrites: CheckpointPendingWrite[];
-    protected checkpointPreviousVersions: Record<string, string | number>;
-    step: number;
-    protected stop: number;
-    protected durability: Durability;
-    protected outputKeys: string | string[];
-    protected streamKeys: string | string[];
-    protected nodes: Record<string, PregelNode>;
-    protected skipDoneTasks: boolean;
-    protected prevCheckpointConfig: RunnableConfig | undefined;
-    status: "pending" | "done" | "interrupt_before" | "interrupt_after" | "out_of_steps";
-    tasks: Record<string, PregelExecutableTask<any, any>>;
-    stream: IterableReadableWritableStream;
-    checkpointerPromises: Promise<unknown>[];
-    isNested: boolean;
-    protected _checkpointerChainedPromise: Promise<unknown>;
-    store?: AsyncBatchedStore;
-    cache?: AsyncBatchedCache;
-    manager?: CallbackManagerForChainRun;
-    interruptAfter: string[] | All;
-    interruptBefore: string[] | All;
-    toInterrupt: PregelExecutableTask<string, string>[];
-    debug: boolean;
-    triggerToNodes: Record<string, string[]>;
-    get isResuming(): any;
-    constructor(params: PregelLoopParams);
-    static initialize(params: PregelLoopInitializeParams): Promise<PregelLoop>;
-    protected _checkpointerPutAfterPrevious(input: {
-        config: RunnableConfig;
-        checkpoint: Checkpoint;
-        metadata: CheckpointMetadata;
-        newVersions: Record<string, string | number>;
-    }): void;
-    /**
-     * Put writes for a task, to be read by the next tick.
-     * @param taskId
-     * @param writes
-     */
-    putWrites(taskId: string, writes: PendingWrite<string>[]): void;
-    _outputWrites(taskId: string, writes: [string, unknown][], cached?: boolean): void;
-    _matchCachedWrites(): Promise<{
-        task: PregelExecutableTask<string, string>;
-        result: unknown;
-    }[]>;
-    /**
-     * Execute a single iteration of the Pregel loop.
-     * Returns true if more iterations are needed.
-     * @param params
-     */
-    tick(params: {
-        inputKeys?: string | string[];
-    }): Promise<boolean>;
-    finishAndHandleError(error?: Error): Promise<boolean>;
-    acceptPush(task: PregelExecutableTask<string, string>, writeIdx: number, call?: Call): Promise<PregelExecutableTask<string, string> | void>;
-    protected _suppressInterrupt(e?: Error): boolean;
-    protected _first(inputKeys: string | string[]): Promise<void>;
-    protected _emit(values: [StreamMode, unknown][]): void;
-    protected _putCheckpoint(inputMetadata: Omit<CheckpointMetadata, "step" | "parents">): void;
-    protected _flushPendingWrites(): void;
-    protected _matchWrites(tasks: Record<string, PregelExecutableTask<string, string>>): void;
-}
-export {};

package/dist/pregel/messages.d.ts

@@ -1,30 +0,0 @@
-import { BaseCallbackHandler, HandleLLMNewTokenCallbackFields, NewTokenIndices } from "@langchain/core/callbacks/base";
-import { BaseMessage } from "@langchain/core/messages";
-import { Serialized } from "@langchain/core/load/serializable";
-import { LLMResult } from "@langchain/core/outputs";
-import { ChainValues } from "@langchain/core/utils/types";
-import { StreamChunk } from "./stream.js";
-type Meta = [string[], Record<string, any>];
-/**
- * A callback handler that implements stream_mode=messages.
- * Collects messages from (1) chat model stream events and (2) node outputs.
- */
-export declare class StreamMessagesHandler extends BaseCallbackHandler {
-    name: string;
-    streamFn: (streamChunk: StreamChunk) => void;
-    metadatas: Record<string, Meta | undefined>;
-    seen: Record<string, BaseMessage>;
-    emittedChatModelRunIds: Record<string, boolean>;
-    stableMessageIdMap: Record<string, string>;
-    lc_prefer_streaming: boolean;
-    constructor(streamFn: (streamChunk: StreamChunk) => void);
-    _emit(meta: Meta, message: BaseMessage, runId: string | undefined, dedupe?: boolean): void;
-    handleChatModelStart(_llm: Serialized, _messages: BaseMessage[][], runId: string, _parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, name?: string): void;
-    handleLLMNewToken(token: string, _idx: NewTokenIndices, runId: string, _parentRunId?: string, _tags?: string[], fields?: HandleLLMNewTokenCallbackFields): void;
-    handleLLMEnd(output: LLMResult, runId: string): void;
-    handleLLMError(_err: any, runId: string): void;
-    handleChainStart(_chain: Serialized, inputs: ChainValues, runId: string, _parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, name?: string): void;
-    handleChainEnd(outputs: ChainValues, runId: string): void;
-    handleChainError(_err: any, runId: string): void;
-}
-export {};