@langchain/langgraph 0.2.41 → 0.2.43-rc.0

package/dist/graph/graph.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CompiledGraph = exports.Graph = exports.Branch = exports.END = exports.START = void 0;
+exports.CompiledGraph = exports.Graph = exports.Branch = void 0;
 /* eslint-disable @typescript-eslint/no-use-before-define */
 const runnables_1 = require("@langchain/core/runnables");
 const graph_1 = require("@langchain/core/runnables/graph");
@@ -14,10 +14,6 @@ const constants_js_1 = require("../constants.cjs");
 const utils_js_1 = require("../utils.cjs");
 const errors_js_1 = require("../errors.cjs");
 const subgraph_js_1 = require("../pregel/utils/subgraph.cjs");
-/** Special reserved node name denoting the start of a graph. */
-exports.START = "__start__";
-/** Special reserved node name denoting the end of a graph. */
-exports.END = "__end__";
 class Branch {
     constructor(options) {
         Object.defineProperty(this, "condition", {
@@ -83,7 +79,7 @@ class Branch {
         if (destinations.some((dest) => !dest)) {
             throw new Error("Branch condition returned unknown or null destination");
         }
-        if (destinations.filter(constants_js_1._isSend).some((packet) => packet.node === exports.END)) {
+        if (destinations.filter(constants_js_1._isSend).some((packet) => packet.node === constants_js_1.END)) {
             throw new errors_js_1.InvalidUpdateError("Cannot send a packet to the END node");
         }
         const writeResult = await writer(destinations, config);
@@ -149,7 +145,7 @@ class Graph {
         if (key in this.nodes) {
             throw new Error(`Node \`${key}\` already present.`);
         }
-        if (key === exports.END) {
+        if (key === constants_js_1.END) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
         const runnable = (0, runnables_1._coerceToRunnable)(
@@ -165,10 +161,10 @@ class Graph {
     }
     addEdge(startKey, endKey) {
         this.warnIfCompiled(`Adding an edge to a graph that has already been compiled. This will not be reflected in the compiled graph.`);
-        if (startKey === exports.END) {
+        if (startKey === constants_js_1.END) {
             throw new Error("END cannot be a start node");
         }
-        if (endKey === exports.START) {
+        if (endKey === constants_js_1.START) {
             throw new Error("START cannot be an end node");
         }
         if (Array.from(this.edges).some(([start]) => start === startKey) &&
@@ -215,14 +211,14 @@ class Graph {
      */
     setEntryPoint(key) {
         this.warnIfCompiled("Setting the entry point of a graph that has already been compiled. This will not be reflected in the compiled graph.");
-        return this.addEdge(exports.START, key);
+        return this.addEdge(constants_js_1.START, key);
     }
     /**
      * @deprecated use `addEdge(key, END)` instead
      */
     setFinishPoint(key) {
         this.warnIfCompiled("Setting a finish point of a graph that has already been compiled. This will not be reflected in the compiled graph.");
-        return this.addEdge(key, exports.END);
+        return this.addEdge(key, constants_js_1.END);
     }
     compile({ checkpointer, interruptBefore, interruptAfter, } = {}) {
         // validate the graph
@@ -239,11 +235,11 @@ class Graph {
             autoValidate: false,
             nodes: {},
             channels: {
-                [exports.START]: new ephemeral_value_js_1.EphemeralValue(),
-                [exports.END]: new ephemeral_value_js_1.EphemeralValue(),
+                [constants_js_1.START]: new ephemeral_value_js_1.EphemeralValue(),
+                [constants_js_1.END]: new ephemeral_value_js_1.EphemeralValue(),
             },
-            inputChannels: exports.START,
-            outputChannels: exports.END,
+            inputChannels: constants_js_1.START,
+            outputChannels: constants_js_1.END,
             streamChannels: [],
             streamMode: "values",
         });
@@ -268,7 +264,7 @@ class Graph {
             allSources.add(start);
         }
         for (const source of allSources) {
-            if (source !== exports.START && !(source in this.nodes)) {
+            if (source !== constants_js_1.START && !(source in this.nodes)) {
                 throw new Error(`Found edge starting at unknown node \`${source}\``);
             }
         }
@@ -282,7 +278,7 @@ class Graph {
                     }
                 }
                 else {
-                    allTargets.add(exports.END);
+                    allTargets.add(constants_js_1.END);
                     for (const node of Object.keys(this.nodes)) {
                         if (node !== start) {
                             allTargets.add(node);
@@ -311,7 +307,7 @@ class Graph {
             }
         }
         for (const target of allTargets) {
-            if (target !== exports.END && !(target in this.nodes)) {
+            if (target !== constants_js_1.END && !(target in this.nodes)) {
                 throw new Error(`Found edge ending at unknown node \`${target}\``);
             }
         }
@@ -352,11 +348,11 @@ class CompiledGraph extends index_js_1.Pregel {
         this.streamChannels.push(key);
     }
     attachEdge(start, end) {
-        if (end === exports.END) {
-            if (start === exports.START) {
+        if (end === constants_js_1.END) {
+            if (start === constants_js_1.START) {
                 throw new Error("Cannot have an edge from START to END");
             }
-            this.nodes[start].writers.push(new write_js_1.ChannelWrite([{ channel: exports.END, value: write_js_1.PASSTHROUGH }], [constants_js_1.TAG_HIDDEN]));
+            this.nodes[start].writers.push(new write_js_1.ChannelWrite([{ channel: constants_js_1.END, value: write_js_1.PASSTHROUGH }], [constants_js_1.TAG_HIDDEN]));
         }
         else {
             this.nodes[end].triggers.push(start);
@@ -365,8 +361,8 @@ class CompiledGraph extends index_js_1.Pregel {
     }
     attachBranch(start, name, branch) {
         // add hidden start node
-        if (start === exports.START && this.nodes[exports.START]) {
-            this.nodes[exports.START] = index_js_1.Channel.subscribeTo(exports.START, { tags: [constants_js_1.TAG_HIDDEN] });
+        if (start === constants_js_1.START && this.nodes[constants_js_1.START]) {
+            this.nodes[constants_js_1.START] = index_js_1.Channel.subscribeTo(constants_js_1.START, { tags: [constants_js_1.TAG_HIDDEN] });
         }
         // attach branch writer
         this.nodes[start].pipe(branch.run((dests) => {
@@ -375,7 +371,7 @@ class CompiledGraph extends index_js_1.Pregel {
                     return dest;
                 }
                 return {
-                    channel: dest === exports.END ? exports.END : `branch:${start}:${name}:${dest}`,
+                    channel: dest === constants_js_1.END ? constants_js_1.END : `branch:${start}:${name}:${dest}`,
                     value: write_js_1.PASSTHROUGH,
                 };
             });
@@ -386,7 +382,7 @@ class CompiledGraph extends index_js_1.Pregel {
             ? Object.values(branch.ends)
             : Object.keys(this.nodes);
         for (const end of ends) {
-            if (end !== exports.END) {
+            if (end !== constants_js_1.END) {
                 const channelName = `branch:${start}:${name}:${end}`;
                 this.channels[channelName] =
                     new ephemeral_value_js_1.EphemeralValue();
@@ -402,9 +398,9 @@ class CompiledGraph extends index_js_1.Pregel {
         const xray = config?.xray;
         const graph = new graph_1.Graph();
         const startNodes = {
-            [exports.START]: graph.addNode({
+            [constants_js_1.START]: graph.addNode({
                 schema: zod_1.z.any(),
-            }, exports.START),
+            }, constants_js_1.START),
         };
         const endNodes = {};
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -415,8 +411,8 @@ class CompiledGraph extends index_js_1.Pregel {
             (x) => isCompiledGraph(x[1])));
         }
         function addEdge(start, end, label, conditional = false) {
-            if (end === exports.END && endNodes[exports.END] === undefined) {
-                endNodes[exports.END] = graph.addNode({ schema: zod_1.z.any() }, exports.END);
+            if (end === constants_js_1.END && endNodes[constants_js_1.END] === undefined) {
+                endNodes[constants_js_1.END] = graph.addNode({ schema: zod_1.z.any() }, constants_js_1.END);
             }
             return graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, conditional);
         }
@@ -526,7 +522,7 @@ class CompiledGraph extends index_js_1.Pregel {
                 ...Object.fromEntries(Object.keys(this.builder.nodes)
                     .filter((k) => k !== start)
                     .map((k) => [_escapeMermaidKeywords(k), _escapeMermaidKeywords(k)])),
-                [exports.END]: exports.END,
+                [constants_js_1.END]: constants_js_1.END,
             };
             for (const branch of Object.values(branches)) {
                 let ends;
@@ -559,9 +555,9 @@ class CompiledGraph extends index_js_1.Pregel {
         const xray = config?.xray;
         const graph = new graph_1.Graph();
         const startNodes = {
-            [exports.START]: graph.addNode({
+            [constants_js_1.START]: graph.addNode({
                 schema: zod_1.z.any(),
-            }, exports.START),
+            }, constants_js_1.START),
         };
         const endNodes = {};
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -572,8 +568,8 @@ class CompiledGraph extends index_js_1.Pregel {
             (x) => isCompiledGraph(x[1])));
         }
         function addEdge(start, end, label, conditional = false) {
-            if (end === exports.END && endNodes[exports.END] === undefined) {
-                endNodes[exports.END] = graph.addNode({ schema: zod_1.z.any() }, exports.END);
+            if (end === constants_js_1.END && endNodes[constants_js_1.END] === undefined) {
+                endNodes[constants_js_1.END] = graph.addNode({ schema: zod_1.z.any() }, constants_js_1.END);
             }
             return graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, conditional);
         }
@@ -683,7 +679,7 @@ class CompiledGraph extends index_js_1.Pregel {
                 ...Object.fromEntries(Object.keys(this.builder.nodes)
                     .filter((k) => k !== start)
                     .map((k) => [_escapeMermaidKeywords(k), _escapeMermaidKeywords(k)])),
-                [exports.END]: exports.END,
+                [constants_js_1.END]: constants_js_1.END,
             };
             for (const branch of Object.values(branches)) {
                 let ends;

package/dist/graph/graph.d.ts

@@ -5,14 +5,10 @@ import { PregelNode } from "../pregel/read.js";
 import { Pregel } from "../pregel/index.js";
 import type { PregelParams } from "../pregel/types.js";
 import { BaseChannel } from "../channels/base.js";
-import { Send } from "../constants.js";
+import { END, Send, START } from "../constants.js";
 import { RunnableCallable } from "../utils.js";
 import { StateDefinition, StateType } from "./annotation.js";
 import type { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
-/** Special reserved node name denoting the start of a graph. */
-export declare const START = "__start__";
-/** Special reserved node name denoting the end of a graph. */
-export declare const END = "__end__";
 export interface BranchOptions<IO, N extends string, CallOptions extends LangGraphRunnableConfig = LangGraphRunnableConfig> {
     source: N;
     path: RunnableLike<IO, BranchPathReturnValue, CallOptions>;

package/dist/graph/graph.js

@@ -7,14 +7,10 @@ import { PregelNode } from "../pregel/read.js";
 import { Channel, Pregel } from "../pregel/index.js";
 import { EphemeralValue } from "../channels/ephemeral_value.js";
 import { ChannelWrite, PASSTHROUGH } from "../pregel/write.js";
-import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
+import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, END, START, TAG_HIDDEN, } from "../constants.js";
 import { gatherIterator, gatherIteratorSync, RunnableCallable, } from "../utils.js";
 import { InvalidUpdateError, NodeInterrupt, UnreachableNodeError, } from "../errors.js";
 import { isPregelLike } from "../pregel/utils/subgraph.js";
-/** Special reserved node name denoting the start of a graph. */
-export const START = "__start__";
-/** Special reserved node name denoting the end of a graph. */
-export const END = "__end__";
 export class Branch {
     constructor(options) {
         Object.defineProperty(this, "condition", {

package/dist/graph/index.cjs

@@ -1,12 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.messagesStateReducer = exports.MessageGraph = exports.CompiledStateGraph = exports.StateGraph = exports.Graph = exports.START = exports.END = exports.AnnotationRoot = exports.Annotation = void 0;
+exports.messagesStateReducer = exports.MessageGraph = exports.CompiledStateGraph = exports.StateGraph = exports.Graph = exports.AnnotationRoot = exports.Annotation = void 0;
 var annotation_js_1 = require("./annotation.cjs");
 Object.defineProperty(exports, "Annotation", { enumerable: true, get: function () { return annotation_js_1.Annotation; } });
 Object.defineProperty(exports, "AnnotationRoot", { enumerable: true, get: function () { return annotation_js_1.AnnotationRoot; } });
 var graph_js_1 = require("./graph.cjs");
-Object.defineProperty(exports, "END", { enumerable: true, get: function () { return graph_js_1.END; } });
-Object.defineProperty(exports, "START", { enumerable: true, get: function () { return graph_js_1.START; } });
 Object.defineProperty(exports, "Graph", { enumerable: true, get: function () { return graph_js_1.Graph; } });
 var state_js_1 = require("./state.cjs");
 Object.defineProperty(exports, "StateGraph", { enumerable: true, get: function () { return state_js_1.StateGraph; } });

package/dist/graph/index.d.ts

@@ -1,4 +1,4 @@
 export { Annotation, type StateType, type UpdateType, type NodeType, AnnotationRoot, type StateDefinition, type SingleReducer, } from "./annotation.js";
-export { END, START, Graph, type CompiledGraph } from "./graph.js";
+export { Graph, type CompiledGraph } from "./graph.js";
 export { type StateGraphArgs, StateGraph, CompiledStateGraph, } from "./state.js";
 export { MessageGraph, messagesStateReducer, type Messages, } from "./message.js";

package/dist/graph/index.js

@@ -1,4 +1,4 @@
 export { Annotation, AnnotationRoot, } from "./annotation.js";
-export { END, START, Graph } from "./graph.js";
+export { Graph } from "./graph.js";
 export { StateGraph, CompiledStateGraph, } from "./state.js";
 export { MessageGraph, messagesStateReducer, } from "./message.js";

package/dist/graph/message.d.ts

@@ -6,7 +6,7 @@ export type Messages = Array<BaseMessage | BaseMessageLike> | BaseMessage | Base
  * Can handle standard messages and special modifiers like {@link RemoveMessage}
  * instances.
  */
-export declare function messagesStateReducer(left: BaseMessage[], right: Messages): BaseMessage[];
+export declare function messagesStateReducer(left: Messages, right: Messages): BaseMessage[];
 /** @ignore */
 export declare class MessageGraph extends StateGraph<BaseMessage[], BaseMessage[], Messages> {
     constructor();

package/dist/graph/state.cjs

@@ -211,7 +211,7 @@ class StateGraph extends graph_js_1.Graph {
         if (key in this.nodes) {
             throw new Error(`Node \`${key}\` already present.`);
         }
-        if (key === graph_js_1.END || key === graph_js_1.START) {
+        if (key === constants_js_1.END || key === constants_js_1.START) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
         if (options?.input !== undefined) {
@@ -254,14 +254,14 @@ class StateGraph extends graph_js_1.Graph {
                 "not be reflected in the compiled graph.");
         }
         for (const start of startKey) {
-            if (start === graph_js_1.END) {
+            if (start === constants_js_1.END) {
                 throw new Error("END cannot be a start node");
             }
             if (!Object.keys(this.nodes).some((node) => node === start)) {
                 throw new Error(`Need to add a node named "${start}" first`);
             }
         }
-        if (endKey === graph_js_1.END) {
+        if (endKey === constants_js_1.END) {
             throw new Error("END cannot be an end node");
         }
         if (!Object.keys(this.nodes).some((node) => node === endKey)) {
@@ -291,20 +291,20 @@ class StateGraph extends graph_js_1.Graph {
             nodes: {},
             channels: {
                 ...this.channels,
-                [graph_js_1.START]: new ephemeral_value_js_1.EphemeralValue(),
+                [constants_js_1.START]: new ephemeral_value_js_1.EphemeralValue(),
             },
-            inputChannels: graph_js_1.START,
+            inputChannels: constants_js_1.START,
             outputChannels,
             streamChannels,
             streamMode: "updates",
             store,
         });
         // attach nodes, edges and branches
-        compiled.attachNode(graph_js_1.START);
+        compiled.attachNode(constants_js_1.START);
         for (const [key, node] of Object.entries(this.nodes)) {
             compiled.attachNode(key, node);
         }
-        compiled.attachBranch(graph_js_1.START, constants_js_1.SELF, _getControlBranch(), {
+        compiled.attachBranch(constants_js_1.START, constants_js_1.SELF, _getControlBranch(), {
             withReader: false,
         });
         for (const [key] of Object.entries(this.nodes)) {
@@ -348,7 +348,7 @@ function _getChannels(schema) {
 class CompiledStateGraph extends graph_js_1.CompiledGraph {
     attachNode(key, node) {
         let outputKeys;
-        if (key === graph_js_1.START) {
+        if (key === constants_js_1.START) {
             // Get input schema keys excluding managed values
             outputKeys = Object.entries(this.builder._schemaDefinitions.get(this.builder._inputDefinition))
                 .filter(([_, v]) => !(0, base_js_2.isConfiguredManagedValue)(v))
@@ -443,11 +443,11 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             },
         ];
         // add node and output channel
-        if (key === graph_js_1.START) {
+        if (key === constants_js_1.START) {
             this.nodes[key] = new read_js_1.PregelNode({
                 tags: [constants_js_1.TAG_HIDDEN],
-                triggers: [graph_js_1.START],
-                channels: [graph_js_1.START],
+                triggers: [constants_js_1.START],
+                channels: [constants_js_1.START],
                 writers: [new write_js_1.ChannelWrite(stateWriteEntries, [constants_js_1.TAG_HIDDEN])],
             });
         }
@@ -479,7 +479,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
         }
     }
     attachEdge(start, end) {
-        if (end === graph_js_1.END) {
+        if (end === constants_js_1.END) {
             return;
         }
         if (Array.isArray(start)) {
@@ -494,15 +494,15 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
                 this.nodes[s].writers.push(new write_js_1.ChannelWrite([{ channel: channelName, value: s }], [constants_js_1.TAG_HIDDEN]));
             }
         }
-        else if (start === graph_js_1.START) {
-            const channelName = `${graph_js_1.START}:${end}`;
+        else if (start === constants_js_1.START) {
+            const channelName = `${constants_js_1.START}:${end}`;
             // register channel
             this.channels[channelName] =
                 new ephemeral_value_js_1.EphemeralValue();
             // subscribe to channel
             this.nodes[end].triggers.push(channelName);
             // publish to channel
-            this.nodes[graph_js_1.START].writers.push(new write_js_1.ChannelWrite([{ channel: channelName, value: graph_js_1.START }], [constants_js_1.TAG_HIDDEN]));
+            this.nodes[constants_js_1.START].writers.push(new write_js_1.ChannelWrite([{ channel: channelName, value: constants_js_1.START }], [constants_js_1.TAG_HIDDEN]));
         }
         else {
             this.nodes[end].triggers.push(start);
@@ -510,7 +510,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
     }
     attachBranch(start, name, branch, options = { withReader: true }) {
         const branchWriter = async (packets, config) => {
-            const filteredPackets = packets.filter((p) => p !== graph_js_1.END);
+            const filteredPackets = packets.filter((p) => p !== constants_js_1.END);
             if (!filteredPackets.length) {
                 return;
             }
@@ -536,7 +536,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             ? Object.values(branch.ends)
             : Object.keys(this.builder.nodes);
         for (const end of ends) {
-            if (end === graph_js_1.END) {
+            if (end === constants_js_1.END) {
                 continue;
             }
             const channelName = `branch:${start}:${name}:${end}`;

package/dist/graph/state.d.ts

@@ -1,7 +1,8 @@
 import { RunnableLike } from "@langchain/core/runnables";
 import { All, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
-import { END, CompiledGraph, Graph, START, Branch, AddNodeOptions, NodeSpec } from "./graph.js";
+import { CompiledGraph, Graph, Branch, AddNodeOptions, NodeSpec } from "./graph.js";
+import { END, START } from "../constants.js";
 import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType } from "./annotation.js";
 import type { RetryPolicy } from "../pregel/utils/index.js";
 import { ManagedValueSpec } from "../managed/base.js";

package/dist/graph/state.js

@@ -1,13 +1,13 @@
 /* eslint-disable @typescript-eslint/no-use-before-define */
 import { _coerceToRunnable, Runnable, } from "@langchain/core/runnables";
 import { isBaseChannel } from "../channels/base.js";
-import { END, CompiledGraph, Graph, START, Branch, } from "./graph.js";
+import { CompiledGraph, Graph, Branch, } from "./graph.js";
 import { ChannelWrite, PASSTHROUGH, } from "../pregel/write.js";
 import { ChannelRead, PregelNode } from "../pregel/read.js";
 import { NamedBarrierValue } from "../channels/named_barrier_value.js";
 import { EphemeralValue } from "../channels/ephemeral_value.js";
 import { RunnableCallable } from "../utils.js";
-import { isCommand, _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, Command, SELF, TAG_HIDDEN, } from "../constants.js";
+import { isCommand, _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, Command, END, SELF, START, TAG_HIDDEN, } from "../constants.js";
 import { InvalidUpdateError, ParentCommand } from "../errors.js";
 import { getChannel, } from "./annotation.js";
 import { isConfiguredManagedValue } from "../managed/base.js";

package/dist/index.cjs

@@ -15,7 +15,15 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.getPreviousState = exports.getWriter = exports.getStore = exports.interrupt = void 0;
 const async_local_storage_js_1 = require("./setup/async_local_storage.cjs");
 // Initialize global async local storage instance for tracing
 (0, async_local_storage_js_1.initializeAsyncLocalStorageSingleton)();
 __exportStar(require("./web.cjs"), exports);
+var interrupt_js_1 = require("./interrupt.cjs");
+Object.defineProperty(exports, "interrupt", { enumerable: true, get: function () { return interrupt_js_1.interrupt; } });
+var config_js_1 = require("./pregel/utils/config.cjs");
+Object.defineProperty(exports, "getStore", { enumerable: true, get: function () { return config_js_1.getStore; } });
+Object.defineProperty(exports, "getWriter", { enumerable: true, get: function () { return config_js_1.getWriter; } });
+var index_js_1 = require("./func/index.cjs");
+Object.defineProperty(exports, "getPreviousState", { enumerable: true, get: function () { return index_js_1.getPreviousState; } });

package/dist/index.d.ts

@@ -1 +1,4 @@
 export * from "./web.js";
+export { interrupt } from "./interrupt.js";
+export { getStore, getWriter } from "./pregel/utils/config.js";
+export { getPreviousState } from "./func/index.js";

package/dist/index.js

@@ -3,3 +3,6 @@ import { initializeAsyncLocalStorageSingleton } from "./setup/async_local_storag
 // Initialize global async local storage instance for tracing
 initializeAsyncLocalStorageSingleton();
 export * from "./web.js";
+export { interrupt } from "./interrupt.js";
+export { getStore, getWriter } from "./pregel/utils/config.js";
+export { getPreviousState } from "./func/index.js";

package/dist/interrupt.cjs

@@ -43,50 +43,37 @@ const constants_js_1 = require("./constants.cjs");
  * @throws {Error} If called outside the context of a graph
  * @throws {GraphInterrupt} When no resume value is available
  */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
 function interrupt(value) {
     const config = singletons_1.AsyncLocalStorageProviderSingleton.getRunnableConfig();
     if (!config) {
         throw new Error("Called interrupt() outside the context of a graph.");
     }
-    // Track interrupt index
-    const scratchpad = config.configurable?.[constants_js_1.CONFIG_KEY_SCRATCHPAD];
-    if (scratchpad.interruptCounter === undefined) {
-        scratchpad.interruptCounter = 0;
-    }
-    else {
-        scratchpad.interruptCounter += 1;
+    const conf = config.configurable;
+    if (!conf) {
+        throw new Error("No configurable found in config");
     }
+    // Track interrupt index
+    const scratchpad = conf[constants_js_1.CONFIG_KEY_SCRATCHPAD];
+    scratchpad.interruptCounter += 1;
     const idx = scratchpad.interruptCounter;
     // Find previous resume values
-    const taskId = config.configurable?.[constants_js_1.CONFIG_KEY_TASK_ID];
-    const writes = config.configurable?.[constants_js_1.CONFIG_KEY_WRITES] ?? [];
-    if (!scratchpad.resume) {
-        const newResume = (writes.find((w) => w[0] === taskId && w[1] === constants_js_1.RESUME)?.[2] || []);
-        scratchpad.resume = Array.isArray(newResume) ? newResume : [newResume];
-    }
-    if (scratchpad.resume) {
-        if (idx < scratchpad.resume.length) {
-            return scratchpad.resume[idx];
-        }
+    if (scratchpad.resume.length > 0 && idx < scratchpad.resume.length) {
+        return scratchpad.resume[idx];
     }
     // Find current resume value
-    if (!scratchpad.usedNullResume) {
-        scratchpad.usedNullResume = true;
-        const sortedWrites = [...writes].sort((a, b) => b[0].localeCompare(a[0]) // Sort in reverse order
-        );
-        for (const [tid, c, v] of sortedWrites) {
-            if (tid === constants_js_1.NULL_TASK_ID && c === constants_js_1.RESUME) {
-                if (scratchpad.resume.length !== idx) {
-                    throw new Error(`Resume length mismatch: ${scratchpad.resume.length} !== ${idx}`);
-                }
-                scratchpad.resume.push(v);
-                const send = config.configurable?.[constants_js_1.CONFIG_KEY_SEND];
-                if (send) {
-                    send([[constants_js_1.RESUME, scratchpad.resume]]);
-                }
-                return v;
-            }
+    if (scratchpad.nullResume !== undefined) {
+        if (scratchpad.resume.length !== idx) {
+            throw new Error(`Resume length mismatch: ${scratchpad.resume.length} !== ${idx}`);
+        }
+        const v = scratchpad.nullResume;
+        delete scratchpad.nullResume;
+        scratchpad.resume.push(v);
+        const send = conf[constants_js_1.CONFIG_KEY_SEND];
+        if (send) {
+            send([[constants_js_1.RESUME, scratchpad.resume]]);
         }
+        return v;
     }
     // No resume value found
     throw new errors_js_1.GraphInterrupt([
@@ -94,7 +81,7 @@ function interrupt(value) {
             value,
             when: "during",
             resumable: true,
-            ns: config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]?.split(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR),
+            ns: conf[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]?.split(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR),
         },
     ]);
 }

package/dist/interrupt.d.ts

@@ -37,4 +37,4 @@
  * @throws {Error} If called outside the context of a graph
  * @throws {GraphInterrupt} When no resume value is available
  */
-export declare function interrupt<I = unknown, R = unknown>(value: I): R;
+export declare function interrupt<I = unknown, R = any>(value: I): R;

package/dist/interrupt.js

@@ -1,6 +1,6 @@
 import { AsyncLocalStorageProviderSingleton } from "@langchain/core/singletons";
 import { GraphInterrupt } from "./errors.js";
-import { CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_SCRATCHPAD, CONFIG_KEY_TASK_ID, CONFIG_KEY_WRITES, CONFIG_KEY_SEND, CHECKPOINT_NAMESPACE_SEPARATOR, NULL_TASK_ID, RESUME, } from "./constants.js";
+import { CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_SCRATCHPAD, CONFIG_KEY_SEND, CHECKPOINT_NAMESPACE_SEPARATOR, RESUME, } from "./constants.js";
 /**
  * Interrupts the execution of a graph node.
  * This function can be used to pause execution of a node, and return the value of the `resume`
@@ -40,50 +40,37 @@ import { CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_SCRATCHPAD, CONFIG_KEY_TASK_ID, CO
  * @throws {Error} If called outside the context of a graph
  * @throws {GraphInterrupt} When no resume value is available
  */
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
 export function interrupt(value) {
     const config = AsyncLocalStorageProviderSingleton.getRunnableConfig();
     if (!config) {
         throw new Error("Called interrupt() outside the context of a graph.");
     }
-    // Track interrupt index
-    const scratchpad = config.configurable?.[CONFIG_KEY_SCRATCHPAD];
-    if (scratchpad.interruptCounter === undefined) {
-        scratchpad.interruptCounter = 0;
-    }
-    else {
-        scratchpad.interruptCounter += 1;
+    const conf = config.configurable;
+    if (!conf) {
+        throw new Error("No configurable found in config");
     }
+    // Track interrupt index
+    const scratchpad = conf[CONFIG_KEY_SCRATCHPAD];
+    scratchpad.interruptCounter += 1;
     const idx = scratchpad.interruptCounter;
     // Find previous resume values
-    const taskId = config.configurable?.[CONFIG_KEY_TASK_ID];
-    const writes = config.configurable?.[CONFIG_KEY_WRITES] ?? [];
-    if (!scratchpad.resume) {
-        const newResume = (writes.find((w) => w[0] === taskId && w[1] === RESUME)?.[2] || []);
-        scratchpad.resume = Array.isArray(newResume) ? newResume : [newResume];
-    }
-    if (scratchpad.resume) {
-        if (idx < scratchpad.resume.length) {
-            return scratchpad.resume[idx];
-        }
+    if (scratchpad.resume.length > 0 && idx < scratchpad.resume.length) {
+        return scratchpad.resume[idx];
     }
     // Find current resume value
-    if (!scratchpad.usedNullResume) {
-        scratchpad.usedNullResume = true;
-        const sortedWrites = [...writes].sort((a, b) => b[0].localeCompare(a[0]) // Sort in reverse order
-        );
-        for (const [tid, c, v] of sortedWrites) {
-            if (tid === NULL_TASK_ID && c === RESUME) {
-                if (scratchpad.resume.length !== idx) {
-                    throw new Error(`Resume length mismatch: ${scratchpad.resume.length} !== ${idx}`);
-                }
-                scratchpad.resume.push(v);
-                const send = config.configurable?.[CONFIG_KEY_SEND];
-                if (send) {
-                    send([[RESUME, scratchpad.resume]]);
-                }
-                return v;
-            }
+    if (scratchpad.nullResume !== undefined) {
+        if (scratchpad.resume.length !== idx) {
+            throw new Error(`Resume length mismatch: ${scratchpad.resume.length} !== ${idx}`);
+        }
+        const v = scratchpad.nullResume;
+        delete scratchpad.nullResume;
+        scratchpad.resume.push(v);
+        const send = conf[CONFIG_KEY_SEND];
+        if (send) {
+            send([[RESUME, scratchpad.resume]]);
         }
+        return v;
     }
     // No resume value found
     throw new GraphInterrupt([
@@ -91,7 +78,7 @@ export function interrupt(value) {
             value,
             when: "during",
             resumable: true,
-            ns: config.configurable?.[CONFIG_KEY_CHECKPOINT_NS]?.split(CHECKPOINT_NAMESPACE_SEPARATOR),
+            ns: conf[CONFIG_KEY_CHECKPOINT_NS]?.split(CHECKPOINT_NAMESPACE_SEPARATOR),
         },
     ]);
 }