@langchain/langgraph 0.2.73 → 0.3.0

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

@@ -1,75 +1,75 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const globals_1 = require("@jest/globals");
+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, globals_1.describe)("mapCommand", () => {
-    (0, globals_1.it)("should handle Command with goto (string)", () => {
+(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, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode",
+                "branch:to:nextNode",
                 "__start__",
             ],
         ]);
     });
-    (0, globals_1.it)("should handle Command with goto (Send object)", () => {
+    (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, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
         ]);
     });
-    (0, globals_1.it)("should handle Command with goto (array of strings and Send objects)", () => {
+    (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, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode1",
+                "branch:to:nextNode1",
                 "__start__",
             ],
             ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode2",
+                "branch:to:nextNode2",
                 "__start__",
             ],
         ]);
     });
-    (0, globals_1.it)("should throw error for invalid goto value", () => {
+    (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, globals_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("In Command.send, expected Send or string, got object");
+        (0, vitest_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("In Command.send, expected Send or string, got object");
     });
-    (0, globals_1.it)("should handle Command with resume (single value)", () => {
+    (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, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],
         ]);
     });
-    (0, globals_1.it)("should handle Command with resume (object of task IDs)", () => {
+    (0, vitest_1.it)("should handle Command with resume (object of task IDs)", () => {
         // Using a valid UUID-like structure
         const cmd = new constants_js_1.Command({
             resume: {
@@ -79,12 +79,12 @@ const errors_js_1 = require("../errors.cjs");
         });
         const pendingWrites = [];
         const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             ["123e4567-e89b-12d3-a456-426614174000", "__resume__", ["resumeValue1"]],
             ["123e4567-e89b-12d3-a456-426614174001", "__resume__", ["resumeValue2"]],
         ]);
     });
-    (0, globals_1.it)("should handle Command with update (object)", () => {
+    (0, vitest_1.it)("should handle Command with update (object)", () => {
         const cmd = new constants_js_1.Command({
             update: {
                 channel1: "value1",
@@ -93,12 +93,12 @@ const errors_js_1 = require("../errors.cjs");
         });
         const pendingWrites = [];
         const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
             ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
         ]);
     });
-    (0, globals_1.it)("should handle Command with update (array of tuples)", () => {
+    (0, vitest_1.it)("should handle Command with update (array of tuples)", () => {
         const cmd = new constants_js_1.Command({
             update: [
                 ["channel1", "value1"],
@@ -107,29 +107,29 @@ const errors_js_1 = require("../errors.cjs");
         });
         const pendingWrites = [];
         const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             ["00000000-0000-0000-0000-000000000000", "channel1", "value1"],
             ["00000000-0000-0000-0000-000000000000", "channel2", "value2"],
         ]);
     });
-    (0, globals_1.it)("should throw error for invalid update type", () => {
+    (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, globals_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.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, globals_1.it)("should throw error for parent graph reference when none exists", () => {
+    (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, globals_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow(errors_js_1.InvalidUpdateError);
-        (0, globals_1.expect)(() => Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites))).toThrow("There is no parent graph.");
+        (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, globals_1.it)("should handle multiple command attributes together", () => {
+    (0, vitest_1.it)("should handle multiple command attributes together", () => {
         const cmd = new constants_js_1.Command({
             goto: "nextNode",
             resume: "resumeValue",
@@ -137,10 +137,10 @@ const errors_js_1 = require("../errors.cjs");
         });
         const pendingWrites = [];
         const result = Array.from((0, io_js_1.mapCommand)(cmd, pendingWrites));
-        (0, globals_1.expect)(result).toEqual([
+        (0, vitest_1.expect)(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode",
+                "branch:to:nextNode",
                 "__start__",
             ],
             ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],

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

@@ -1,4 +1,4 @@
-import { describe, expect, it } from "@jest/globals";
+import { describe, expect, it } from "vitest";
 import { Command, Send } from "../constants.js";
 import { mapCommand } from "./io.js";
 import { InvalidUpdateError } from "../errors.js";
@@ -12,7 +12,7 @@ describe("mapCommand", () => {
         expect(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode",
+                "branch:to:nextNode",
                 "__start__",
             ],
         ]);
@@ -38,13 +38,13 @@ describe("mapCommand", () => {
         expect(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode1",
+                "branch:to:nextNode1",
                 "__start__",
             ],
             ["00000000-0000-0000-0000-000000000000", "__pregel_tasks", send],
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode2",
+                "branch:to:nextNode2",
                 "__start__",
             ],
         ]);
@@ -138,7 +138,7 @@ describe("mapCommand", () => {
         expect(result).toEqual([
             [
                 "00000000-0000-0000-0000-000000000000",
-                "branch:__start__:__self__:nextNode",
+                "branch:to:nextNode",
                 "__start__",
             ],
             ["00000000-0000-0000-0000-000000000000", "__resume__", "resumeValue"],

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

@@ -1 +1 @@
-{"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,eAAe,CAAC;AACrD,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,oCAAoC;gBACpC,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,qCAAqC;gBACrC,WAAW;aACZ;YACD,CAAC,sCAAsC,EAAE,gBAAgB,EAAE,IAAI,CAAC;YAChE;gBACE,sCAAsC;gBACtC,qCAAqC;gBACrC,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,wDAAwD,EAAE,GAAG,EAAE;QAChE,oCAAoC;QACpC,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,sCAAsC,EAAE,cAAc;gBACtD,sCAAsC,EAAE,cAAc;aACvD;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,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;YACxE,CAAC,sCAAsC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;SACzE,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,oCAAoC;gBACpC,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"}
+{"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,wDAAwD,EAAE,GAAG,EAAE;QAChE,oCAAoC;QACpC,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,MAAM,EAAE;gBACN,sCAAsC,EAAE,cAAc;gBACtD,sCAAsC,EAAE,cAAc;aACvD;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,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;YACxE,CAAC,sCAAsC,EAAE,YAAY,EAAE,CAAC,cAAc,CAAC,CAAC;SACzE,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.cjs

@@ -26,6 +26,44 @@ function createDuplexStream(...streams) {
         modes: new Set(streams.flatMap((s) => Array.from(s.modes))),
     });
 }
+class AsyncBatchedCache extends langgraph_checkpoint_1.BaseCache {
+    constructor(cache) {
+        super();
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queue", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: Promise.resolve()
+        });
+        this.cache = cache;
+    }
+    async get(keys) {
+        return this.enqueueOperation("get", keys);
+    }
+    async set(pairs) {
+        return this.enqueueOperation("set", pairs);
+    }
+    async clear(namespaces) {
+        return this.enqueueOperation("clear", namespaces);
+    }
+    async stop() {
+        await this.queue;
+    }
+    enqueueOperation(type, ...args) {
+        const newPromise = this.queue.then(() => {
+            // @ts-expect-error Tuple type warning
+            return this.cache[type](...args);
+        });
+        this.queue = newPromise.then(() => void 0, () => void 0);
+        return newPromise;
+    }
+}
 class PregelLoop {
     get isResuming() {
         const hasChannelVersions = Object.keys(this.checkpoint.channel_versions).length !== 0;
@@ -207,6 +245,12 @@ class PregelLoop {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "manager", {
             enumerable: true,
             configurable: true,
@@ -237,6 +281,12 @@ class PregelLoop {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "triggerToNodes", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.input = params.input;
         this.checkpointer = params.checkpointer;
         // TODO: if managed values no longer needs graph we can replace with
@@ -264,12 +314,14 @@ class PregelLoop {
         this.nodes = params.nodes;
         this.skipDoneTasks = params.skipDoneTasks;
         this.store = params.store;
+        this.cache = params.cache ? new AsyncBatchedCache(params.cache) : undefined;
         this.stream = params.stream;
         this.checkpointNamespace = params.checkpointNamespace;
         this.prevCheckpointConfig = params.prevCheckpointConfig;
         this.interruptAfter = params.interruptAfter;
         this.interruptBefore = params.interruptBefore;
         this.debug = params.debug;
+        this.triggerToNodes = params.triggerToNodes;
     }
     static async initialize(params) {
         let { config, stream } = params;
@@ -367,9 +419,11 @@ class PregelLoop {
             nodes: params.nodes,
             stream,
             store,
+            cache: params.cache,
             interruptAfter: params.interruptAfter,
             interruptBefore: params.interruptBefore,
             debug: params.debug,
+            triggerToNodes: params.triggerToNodes,
         });
     }
     _checkpointerPutAfterPrevious(input) {
@@ -423,6 +477,25 @@ class PregelLoop {
         if (this.tasks) {
             this._outputWrites(taskId, writesCopy);
         }
+        if (!writes.length || !this.cache || !this.tasks) {
+            return;
+        }
+        // only cache tasks with a cache key
+        const task = this.tasks[taskId];
+        if (task == null || task.cache_key == null) {
+            return;
+        }
+        // only cache successful tasks
+        if (writes[0][0] === constants_js_1.ERROR || writes[0][0] === constants_js_1.INTERRUPT) {
+            return;
+        }
+        void this.cache.set([
+            {
+                key: [task.cache_key.ns, task.cache_key.key],
+                value: task.writes,
+                ttl: task.cache_key.ttl,
+            },
+        ]);
     }
     _outputWrites(taskId, writes, cached = false) {
         const task = this.tasks[taskId];
@@ -441,6 +514,34 @@ class PregelLoop {
             }
         }
     }
+    async _matchCachedWrites() {
+        if (!this.cache)
+            return [];
+        const matched = [];
+        const serializeKey = ([ns, key]) => {
+            return `ns:${ns.join(",")}|key:${key}`;
+        };
+        const keys = [];
+        const keyMap = {};
+        for (const task of Object.values(this.tasks)) {
+            if (task.cache_key != null && !task.writes.length) {
+                keys.push([task.cache_key.ns, task.cache_key.key]);
+                keyMap[serializeKey([task.cache_key.ns, task.cache_key.key])] = task;
+            }
+        }
+        if (keys.length === 0)
+            return [];
+        const cache = await this.cache.get(keys);
+        for (const { key, value } of cache) {
+            const task = keyMap[serializeKey(key)];
+            if (task != null) {
+                // update the task with the cached writes
+                task.writes.push(...value);
+                matched.push({ task, result: value });
+            }
+        }
+        return matched;
+    }
     /**
      * Execute a single iteration of the Pregel loop.
      * Returns true if more iterations are needed.
@@ -465,7 +566,7 @@ class PregelLoop {
             // finish superstep
             const writes = Object.values(this.tasks).flatMap((t) => t.writes);
             // All tasks have finished
-            const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion);
+            const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
             for (const [key, values] of Object.entries(managedValueWrites)) {
                 await this.updateManagedValues(key, values);
             }
@@ -554,25 +655,22 @@ class PregelLoop {
             if (this.tasks !== undefined &&
                 this.checkpointPendingWrites.length > 0 &&
                 Object.values(this.tasks).some((task) => task.writes.length > 0)) {
-                const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion);
+                const managedValueWrites = (0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
                 for (const [key, values] of Object.entries(managedValueWrites)) {
                     await this.updateManagedValues(key, values);
                 }
                 this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
             }
             // Emit INTERRUPT event
+            const interrupts = { [constants_js_1.INTERRUPT]: error.interrupts };
             this._emit([
-                [
-                    "updates",
-                    {
-                        [constants_js_1.INTERRUPT]: error.interrupts,
-                    },
-                ],
+                ["updates", interrupts],
+                ["values", interrupts],
             ]);
         }
         return suppress;
     }
-    acceptPush(task, writeIdx, call) {
+    async acceptPush(task, writeIdx, call) {
         if (this.interruptAfter?.length > 0 &&
             (0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptAfter, [task])) {
             this.toInterrupt.push(task);
@@ -585,22 +683,24 @@ class PregelLoop {
             store: this.store,
             stream: this.stream,
         });
-        if (pushed) {
-            if (this.interruptBefore?.length > 0 &&
-                (0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptBefore, [pushed])) {
-                this.toInterrupt.push(pushed);
-                return;
-            }
-            this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugTasks)(this.step, [pushed]), "debug")));
-            if (this.debug) {
-                (0, debug_js_1.printStepTasks)(this.step, [pushed]);
-            }
-            this.tasks[pushed.id] = pushed;
-            if (this.skipDoneTasks) {
-                this._matchWrites({ [pushed.id]: pushed });
-            }
-            return pushed;
+        if (!pushed)
+            return;
+        if (this.interruptBefore?.length > 0 &&
+            (0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptBefore, [pushed])) {
+            this.toInterrupt.push(pushed);
+            return;
         }
+        this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugTasks)(this.step, [pushed]), "debug")));
+        if (this.debug)
+            (0, debug_js_1.printStepTasks)(this.step, [pushed]);
+        this.tasks[pushed.id] = pushed;
+        if (this.skipDoneTasks)
+            this._matchWrites({ [pushed.id]: pushed });
+        const tasks = await this._matchCachedWrites();
+        for (const { task } of tasks) {
+            this._outputWrites(task.id, task.writes, true);
+        }
+        return pushed;
     }
     _suppressInterrupt(e) {
         return (0, errors_js_1.isGraphInterrupt)(e) && !this.isNested;
@@ -649,7 +749,7 @@ class PregelLoop {
                     writes: nullWrites,
                     triggers: [],
                 },
-            ], this.checkpointerGetNextVersion);
+            ], this.checkpointerGetNextVersion, this.triggerToNodes);
         }
         const isCommandUpdateOrGoto = (0, constants_js_1.isCommand)(this.input) && nullWrites.length > 0;
         if (this.isResuming || isCommandUpdateOrGoto) {
@@ -687,7 +787,7 @@ class PregelLoop {
                         writes: inputWrites,
                         triggers: [],
                     },
-                ]), this.checkpointerGetNextVersion);
+                ]), this.checkpointerGetNextVersion, this.triggerToNodes);
                 // save input checkpoint
                 await this._putCheckpoint({
                     source: "input",

package/dist/pregel/loop.d.ts

@@ -1,6 +1,6 @@
 import type { RunnableConfig } from "@langchain/core/runnables";
 import type { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All, BaseStore, AsyncBatchedStore } from "@langchain/langgraph-checkpoint";
+import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All, BaseStore, AsyncBatchedStore, BaseCache, CacheFullKey, CacheNamespace } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { Call, PregelExecutableTask, StreamMode } from "./types.js";
 import { Command } from "../constants.js";
@@ -19,10 +19,12 @@ export type PregelLoopInitializeParams = {
     managed: ManagedValueMapping;
     stream: IterableReadableWritableStream;
     store?: BaseStore;
+    cache?: BaseCache<PendingWrite<string>[]>;
     interruptAfter: string[] | All;
     interruptBefore: string[] | All;
     manager?: CallbackManagerForChainRun;
     debug: boolean;
+    triggerToNodes: Record<string, string[]>;
 };
 type PregelLoopParams = {
     input?: any | Command;
@@ -46,11 +48,30 @@ type PregelLoopParams = {
     manager?: CallbackManagerForChainRun;
     stream: IterableReadableWritableStream;
     store?: AsyncBatchedStore;
+    cache?: BaseCache<PendingWrite<string>[]>;
     prevCheckpointConfig: RunnableConfig | undefined;
     interruptAfter: string[] | All;
     interruptBefore: string[] | All;
     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;
@@ -79,11 +100,13 @@ export declare class PregelLoop {
     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>;
@@ -101,6 +124,10 @@ export declare class PregelLoop {
      */
     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.
@@ -110,7 +137,7 @@ export declare class PregelLoop {
         inputKeys?: string | string[];
     }): Promise<boolean>;
     finishAndHandleError(error?: Error): Promise<boolean>;
-    acceptPush(task: PregelExecutableTask<string, string>, writeIdx: number, call?: Call): PregelExecutableTask<string, string> | void;
+    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;