@langchain/langgraph 0.2.41 → 0.2.43-rc.0

package/dist/prebuilt/agent_executor.cjs

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.createAgentExecutor = void 0;
 const tool_executor_js_1 = require("./tool_executor.cjs");
 const state_js_1 = require("../graph/state.cjs");
-const index_js_1 = require("../graph/index.cjs");
+const constants_js_1 = require("../constants.cjs");
 /** @ignore */
 function createAgentExecutor({ agentRunnable, tools, }) {
     let toolExecutor;
@@ -54,7 +54,7 @@ function createAgentExecutor({ agentRunnable, tools, }) {
         .addNode("action", executeTools)
         // Set the entrypoint as `agent`
         // This means that this node is the first one called
-        .addEdge(index_js_1.START, "agent")
+        .addEdge(constants_js_1.START, "agent")
         // We now add a conditional edge
         .addConditionalEdges(
     // First, we define the start node. We use `agent`.
@@ -72,7 +72,7 @@ function createAgentExecutor({ agentRunnable, tools, }) {
         // If `tools`, then we call the tool node.
         continue: "action",
         // Otherwise we finish.
-        end: index_js_1.END,
+        end: constants_js_1.END,
     })
         // We now add a normal edge from `tools` to `agent`.
         // This means that after `tools` is called, `agent` node is called next.

package/dist/prebuilt/agent_executor.d.ts

@@ -18,5 +18,5 @@ export interface AgentExecutorState {
 export declare function createAgentExecutor({ agentRunnable, tools, }: {
     agentRunnable: Runnable;
     tools: Array<Tool> | ToolExecutor;
-}): import("../graph/state.js").CompiledStateGraph<AgentExecutorState, Partial<AgentExecutorState>, "__start__" | "agent" | "action", import("../graph/annotation.js").StateDefinition, import("../graph/annotation.js").StateDefinition, import("../graph/annotation.js").StateDefinition>;
+}): import("../graph/state.js").CompiledStateGraph<AgentExecutorState, Partial<AgentExecutorState>, "__start__" | "agent" | "action", import("../web.js").StateDefinition, import("../web.js").StateDefinition, import("../web.js").StateDefinition>;
 export {};

package/dist/prebuilt/agent_executor.js

@@ -1,6 +1,6 @@
 import { ToolExecutor } from "./tool_executor.js";
 import { StateGraph } from "../graph/state.js";
-import { END, START } from "../graph/index.js";
+import { END, START } from "../constants.js";
 /** @ignore */
 export function createAgentExecutor({ agentRunnable, tools, }) {
     let toolExecutor;

package/dist/prebuilt/chat_agent_executor.cjs

@@ -6,7 +6,7 @@ const messages_1 = require("@langchain/core/messages");
 const runnables_1 = require("@langchain/core/runnables");
 const tool_executor_js_1 = require("./tool_executor.cjs");
 const state_js_1 = require("../graph/state.cjs");
-const index_js_1 = require("../graph/index.cjs");
+const constants_js_1 = require("../constants.cjs");
 /** @deprecated Use {@link createReactAgent} instead with tool calling. */
 function createFunctionCallingExecutor({ model, tools, }) {
     let toolExecutor;
@@ -99,7 +99,7 @@ function createFunctionCallingExecutor({ model, tools, }) {
         .addNode("action", new runnables_1.RunnableLambda({ func: callTool }))
         // Set the entrypoint as `agent`
         // This means that this node is the first one called
-        .addEdge(index_js_1.START, "agent")
+        .addEdge(constants_js_1.START, "agent")
         // We now add a conditional edge
         .addConditionalEdges(
     // First, we define the start node. We use `agent`.
@@ -117,7 +117,7 @@ function createFunctionCallingExecutor({ model, tools, }) {
         // If `tools`, then we call the tool node.
         continue: "action",
         // Otherwise we finish.
-        end: index_js_1.END,
+        end: constants_js_1.END,
     })
         // We now add a normal edge from `tools` to `agent`.
         // This means that after `tools` is called, `agent` node is called next.

package/dist/prebuilt/chat_agent_executor.d.ts

@@ -3,7 +3,7 @@ import { BaseMessage } from "@langchain/core/messages";
 import { RunnableToolLike } from "@langchain/core/runnables";
 import { ToolExecutor } from "./tool_executor.js";
 import { CompiledStateGraph } from "../graph/state.js";
-import { START } from "../graph/index.js";
+import { START } from "../constants.js";
 /** @deprecated Use {@link createReactAgent} instead with tool calling. */
 export type FunctionCallingExecutorState = {
     messages: Array<BaseMessage>;

package/dist/prebuilt/chat_agent_executor.js

@@ -3,7 +3,7 @@ import { FunctionMessage } from "@langchain/core/messages";
 import { RunnableLambda, } from "@langchain/core/runnables";
 import { ToolExecutor } from "./tool_executor.js";
 import { StateGraph, } from "../graph/state.js";
-import { END, START } from "../graph/index.js";
+import { END, START } from "../constants.js";
 /** @deprecated Use {@link createReactAgent} instead with tool calling. */
 export function createFunctionCallingExecutor({ model, tools, }) {
     let toolExecutor;

package/dist/prebuilt/react_agent_executor.cjs

@@ -7,6 +7,7 @@ const index_js_1 = require("../graph/index.cjs");
 const tool_node_js_1 = require("./tool_node.cjs");
 const annotation_js_1 = require("../graph/annotation.cjs");
 const message_js_1 = require("../graph/message.cjs");
+const constants_js_1 = require("../constants.cjs");
 function _convertMessageModifierToStateModifier(messageModifier) {
     // Handle string or SystemMessage
     if (typeof messageModifier === "string" ||
@@ -117,7 +118,7 @@ exports.createReactAgentAnnotation = createReactAgentAnnotation;
  * ```
  */
 function createReactAgent(params) {
-    const { llm, tools, messageModifier, stateModifier, stateSchema, checkpointSaver, interruptBefore, interruptAfter, store, responseFormat, } = params;
+    const { llm, tools, messageModifier, stateModifier, stateSchema, checkpointSaver, checkpointer, interruptBefore, interruptAfter, store, responseFormat, } = params;
     let toolClasses;
     if (!Array.isArray(tools)) {
         toolClasses = tools.tools;
@@ -132,12 +133,17 @@ function createReactAgent(params) {
     // we're passing store here for validation
     const preprocessor = _getModelPreprocessingRunnable(stateModifier, messageModifier);
     const modelRunnable = preprocessor.pipe(modelWithTools);
+    // If any of the tools are configured to return_directly after running,
+    // our graph needs to check if these were called
+    const shouldReturnDirect = new Set(toolClasses
+        .filter((tool) => "returnDirect" in tool && tool.returnDirect)
+        .map((tool) => tool.name));
     const shouldContinue = (state) => {
         const { messages } = state;
         const lastMessage = messages[messages.length - 1];
         if ((0, messages_1.isAIMessage)(lastMessage) &&
             (!lastMessage.tool_calls || lastMessage.tool_calls.length === 0)) {
-            return responseFormat != null ? "generate_structured_response" : index_js_1.END;
+            return responseFormat != null ? "generate_structured_response" : constants_js_1.END;
         }
         else {
             return "continue";
@@ -171,26 +177,45 @@ function createReactAgent(params) {
     const workflow = new index_js_1.StateGraph(stateSchema ?? (0, exports.createReactAgentAnnotation)())
         .addNode("agent", callModel)
         .addNode("tools", new tool_node_js_1.ToolNode(toolClasses))
-        .addEdge(index_js_1.START, "agent")
-        .addEdge("tools", "agent");
+        .addEdge(constants_js_1.START, "agent");
     if (responseFormat !== undefined) {
         workflow
             .addNode("generate_structured_response", generateStructuredResponse)
-            .addEdge("generate_structured_response", index_js_1.END)
+            .addEdge("generate_structured_response", constants_js_1.END)
             .addConditionalEdges("agent", shouldContinue, {
             continue: "tools",
-            [index_js_1.END]: index_js_1.END,
+            [constants_js_1.END]: constants_js_1.END,
             generate_structured_response: "generate_structured_response",
         });
     }
     else {
         workflow.addConditionalEdges("agent", shouldContinue, {
             continue: "tools",
-            [index_js_1.END]: index_js_1.END,
+            [constants_js_1.END]: constants_js_1.END,
         });
     }
+    const routeToolResponses = (state) => {
+        // Check the last consecutive tool calls
+        for (let i = state.messages.length - 1; i >= 0; i -= 1) {
+            const message = state.messages[i];
+            if (!(0, messages_1.isToolMessage)(message)) {
+                break;
+            }
+            // Check if this tool is configured to return directly
+            if (message.name !== undefined && shouldReturnDirect.has(message.name)) {
+                return constants_js_1.END;
+            }
+        }
+        return "agent";
+    };
+    if (shouldReturnDirect.size > 0) {
+        workflow.addConditionalEdges("tools", routeToolResponses, ["agent", constants_js_1.END]);
+    }
+    else {
+        workflow.addEdge("tools", "agent");
+    }
     return workflow.compile({
-        checkpointer: checkpointSaver,
+        checkpointer: checkpointer ?? checkpointSaver,
         interruptBefore,
         interruptAfter,
         store,

package/dist/prebuilt/react_agent_executor.d.ts

@@ -4,11 +4,12 @@ import { Runnable, RunnableToolLike } from "@langchain/core/runnables";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { All, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
 import { z } from "zod";
-import { START, CompiledStateGraph, AnnotationRoot } from "../graph/index.js";
+import { CompiledStateGraph, AnnotationRoot } from "../graph/index.js";
 import { MessagesAnnotation } from "../graph/messages_annotation.js";
 import { ToolNode } from "./tool_node.js";
 import { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
 import { Messages } from "../graph/message.js";
+import { START } from "../constants.js";
 export interface AgentState<StructuredResponseType extends Record<string, any> = Record<string, any>> {
     messages: BaseMessage[];
     structuredResponse: StructuredResponseType;
@@ -91,6 +92,8 @@ export type CreateReactAgentParams<A extends AnnotationRoot<any> = AnnotationRoo
     stateSchema?: A;
     /** An optional checkpoint saver to persist the agent's state. */
     checkpointSaver?: BaseCheckpointSaver;
+    /** An optional checkpoint saver to persist the agent's state. Alias of "checkpointSaver". */
+    checkpointer?: BaseCheckpointSaver;
     /** An optional list of node names to interrupt before running. */
     interruptBefore?: N[] | All;
     /** An optional list of node names to interrupt after running. */

package/dist/prebuilt/react_agent_executor.js

@@ -1,9 +1,10 @@
-import { isAIMessage, isBaseMessage, SystemMessage, } from "@langchain/core/messages";
+import { isAIMessage, isBaseMessage, isToolMessage, SystemMessage, } from "@langchain/core/messages";
 import { Runnable, RunnableLambda, } from "@langchain/core/runnables";
-import { END, START, StateGraph, } from "../graph/index.js";
+import { StateGraph, } from "../graph/index.js";
 import { ToolNode } from "./tool_node.js";
 import { Annotation } from "../graph/annotation.js";
 import { messagesStateReducer } from "../graph/message.js";
+import { END, START } from "../constants.js";
 function _convertMessageModifierToStateModifier(messageModifier) {
     // Handle string or SystemMessage
     if (typeof messageModifier === "string" ||
@@ -113,7 +114,7 @@ export const createReactAgentAnnotation = () => Annotation.Root({
  * ```
  */
 export function createReactAgent(params) {
-    const { llm, tools, messageModifier, stateModifier, stateSchema, checkpointSaver, interruptBefore, interruptAfter, store, responseFormat, } = params;
+    const { llm, tools, messageModifier, stateModifier, stateSchema, checkpointSaver, checkpointer, interruptBefore, interruptAfter, store, responseFormat, } = params;
     let toolClasses;
     if (!Array.isArray(tools)) {
         toolClasses = tools.tools;
@@ -128,6 +129,11 @@ export function createReactAgent(params) {
     // we're passing store here for validation
     const preprocessor = _getModelPreprocessingRunnable(stateModifier, messageModifier);
     const modelRunnable = preprocessor.pipe(modelWithTools);
+    // If any of the tools are configured to return_directly after running,
+    // our graph needs to check if these were called
+    const shouldReturnDirect = new Set(toolClasses
+        .filter((tool) => "returnDirect" in tool && tool.returnDirect)
+        .map((tool) => tool.name));
     const shouldContinue = (state) => {
         const { messages } = state;
         const lastMessage = messages[messages.length - 1];
@@ -167,8 +173,7 @@ export function createReactAgent(params) {
     const workflow = new StateGraph(stateSchema ?? createReactAgentAnnotation())
         .addNode("agent", callModel)
         .addNode("tools", new ToolNode(toolClasses))
-        .addEdge(START, "agent")
-        .addEdge("tools", "agent");
+        .addEdge(START, "agent");
     if (responseFormat !== undefined) {
         workflow
             .addNode("generate_structured_response", generateStructuredResponse)
@@ -185,8 +190,28 @@ export function createReactAgent(params) {
             [END]: END,
         });
     }
+    const routeToolResponses = (state) => {
+        // Check the last consecutive tool calls
+        for (let i = state.messages.length - 1; i >= 0; i -= 1) {
+            const message = state.messages[i];
+            if (!isToolMessage(message)) {
+                break;
+            }
+            // Check if this tool is configured to return directly
+            if (message.name !== undefined && shouldReturnDirect.has(message.name)) {
+                return END;
+            }
+        }
+        return "agent";
+    };
+    if (shouldReturnDirect.size > 0) {
+        workflow.addConditionalEdges("tools", routeToolResponses, ["agent", END]);
+    }
+    else {
+        workflow.addEdge("tools", "agent");
+    }
     return workflow.compile({
-        checkpointer: checkpointSaver,
+        checkpointer: checkpointer ?? checkpointSaver,
         interruptBefore,
         interruptAfter,
         store,

package/dist/prebuilt/tool_node.cjs

@@ -3,7 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.toolsCondition = exports.ToolNode = void 0;
 const messages_1 = require("@langchain/core/messages");
 const utils_js_1 = require("../utils.cjs");
-const graph_js_1 = require("../graph/graph.cjs");
 const errors_js_1 = require("../errors.cjs");
 const constants_js_1 = require("../constants.cjs");
 /**
@@ -218,7 +217,7 @@ function toolsCondition(state) {
         return "tools";
     }
     else {
-        return graph_js_1.END;
+        return constants_js_1.END;
     }
 }
 exports.toolsCondition = toolsCondition;

package/dist/prebuilt/tool_node.d.ts

@@ -2,8 +2,8 @@ import { BaseMessage } from "@langchain/core/messages";
 import { RunnableConfig, RunnableToolLike } from "@langchain/core/runnables";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { RunnableCallable } from "../utils.js";
-import { END } from "../graph/graph.js";
 import { MessagesAnnotation } from "../graph/messages_annotation.js";
+import { END } from "../constants.js";
 export type ToolNodeOptions = {
     name?: string;
     tags?: string[];

package/dist/prebuilt/tool_node.js

@@ -1,8 +1,7 @@
 import { ToolMessage, isBaseMessage, } from "@langchain/core/messages";
 import { RunnableCallable } from "../utils.js";
-import { END } from "../graph/graph.js";
 import { isGraphInterrupt } from "../errors.js";
-import { isCommand } from "../constants.js";
+import { END, isCommand } from "../constants.js";
 /**
  * A node that runs the tools requested in the last AIMessage. It can be used
  * either in StateGraph with a "messages" key or in MessageGraph. If multiple

package/dist/pregel/algo.cjs

@@ -7,8 +7,10 @@ const langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
 const base_js_1 = require("../channels/base.cjs");
 const io_js_1 = require("./io.cjs");
 const constants_js_1 = require("../constants.cjs");
+const types_js_1 = require("./types.cjs");
 const errors_js_1 = require("../errors.cjs");
 const index_js_1 = require("./utils/index.cjs");
+const call_js_1 = require("./call.cjs");
 const increment = (current) => {
     return current !== undefined ? current + 1 : 1;
 };
@@ -79,7 +81,7 @@ commit, processes, managed,
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 writes) {
     for (const [chan, value] of writes) {
-        if (chan === constants_js_1.TASKS) {
+        if (chan === constants_js_1.TASKS && value != null) {
             if (!(0, constants_js_1._isSend)(value)) {
                 throw new errors_js_1.InvalidUpdateError(`Invalid packet type, expected SendProtocol, got ${JSON.stringify(value)}`);
             }
@@ -93,7 +95,14 @@ writes) {
     commit(writes);
 }
 exports._localWrite = _localWrite;
-const IGNORE = new Set([constants_js_1.PUSH, constants_js_1.RESUME, constants_js_1.INTERRUPT]);
+const IGNORE = new Set([
+    constants_js_1.NO_WRITES,
+    constants_js_1.PUSH,
+    constants_js_1.RESUME,
+    constants_js_1.INTERRUPT,
+    constants_js_1.RETURN,
+    constants_js_1.ERROR,
+]);
 function _applyWrites(checkpoint, channels, tasks, 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 getNextVersion) {
@@ -260,7 +269,84 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
     const { step, checkpointer, manager } = extra;
     const configurable = config.configurable ?? {};
     const parentNamespace = configurable.checkpoint_ns ?? "";
-    if (taskPath[0] === constants_js_1.PUSH) {
+    if (taskPath[0] === constants_js_1.PUSH && (0, types_js_1.isCall)(taskPath[taskPath.length - 1])) {
+        const call = taskPath[taskPath.length - 1];
+        const proc = (0, call_js_1.getRunnableForFunc)(call.name, call.func);
+        const triggers = [constants_js_1.PUSH];
+        const checkpointNamespace = parentNamespace === ""
+            ? call.name
+            : `${parentNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${call.name}`;
+        const id = (0, langgraph_checkpoint_1.uuid5)(JSON.stringify([
+            checkpointNamespace,
+            step.toString(),
+            call.name,
+            constants_js_1.PUSH,
+            taskPath[1],
+            taskPath[2],
+        ]), checkpoint.id);
+        const taskCheckpointNamespace = `${checkpointNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_END}${id}`;
+        const metadata = {
+            langgraph_step: step,
+            langgraph_node: call.name,
+            langgraph_triggers: triggers,
+            langgraph_path: taskPath.slice(0, 3),
+            langgraph_checkpoint_ns: taskCheckpointNamespace,
+        };
+        if (forExecution) {
+            const writes = [];
+            const task = {
+                name: call.name,
+                input: call.input,
+                proc,
+                writes,
+                config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
+                    metadata,
+                    store: extra.store ?? config.store,
+                }), {
+                    runName: call.name,
+                    callbacks: manager?.getChild(`graph:step:${step}`),
+                    configurable: {
+                        [constants_js_1.CONFIG_KEY_TASK_ID]: id,
+                        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                        [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, managed, writes_),
+                        [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                            name: call.name,
+                            writes: writes,
+                            triggers,
+                            path: taskPath.slice(0, 3),
+                        }, select_, fresh_),
+                        [constants_js_1.CONFIG_KEY_CHECKPOINTER]: checkpointer ?? configurable[constants_js_1.CONFIG_KEY_CHECKPOINTER],
+                        [constants_js_1.CONFIG_KEY_CHECKPOINT_MAP]: {
+                            ...configurable[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP],
+                            [parentNamespace]: checkpoint.id,
+                        },
+                        [constants_js_1.CONFIG_KEY_SCRATCHPAD]: _scratchpad([
+                            ...(pendingWrites || []),
+                            ...(configurable[constants_js_1.CONFIG_KEY_SCRATCHPAD]?.resume || []).map((v) => [id, constants_js_1.RESUME, v]),
+                        ], id),
+                        [constants_js_1.CONFIG_KEY_PREVIOUS_STATE]: checkpoint.channel_values[constants_js_1.PREVIOUS],
+                        checkpoint_id: undefined,
+                        checkpoint_ns: taskCheckpointNamespace,
+                    },
+                }),
+                triggers,
+                retry_policy: call.retry,
+                id,
+                path: taskPath.slice(0, 3),
+                writers: [],
+            };
+            return task;
+        }
+        else {
+            return {
+                id,
+                name: call.name,
+                interrupts: [],
+                path: taskPath.slice(0, 3),
+            };
+        }
+    }
+    else if (taskPath[0] === constants_js_1.PUSH) {
         const index = typeof taskPath[1] === "number"
             ? taskPath[1]
             : parseInt(taskPath[1], 10);
@@ -332,11 +418,15 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
                                 ...configurable[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP],
                                 [parentNamespace]: checkpoint.id,
                             },
-                            [constants_js_1.CONFIG_KEY_WRITES]: [
+                            [constants_js_1.CONFIG_KEY_SCRATCHPAD]: _scratchpad([
                                 ...(pendingWrites || []),
-                                ...(configurable[constants_js_1.CONFIG_KEY_WRITES] || []),
-                            ].filter((w) => w[0] === constants_js_1.NULL_TASK_ID || w[0] === taskId),
-                            [constants_js_1.CONFIG_KEY_SCRATCHPAD]: {},
+                                ...(configurable[constants_js_1.CONFIG_KEY_SCRATCHPAD]?.resume || []).map((v) => [
+                                    taskId,
+                                    constants_js_1.RESUME,
+                                    v,
+                                ]),
+                            ], taskId),
+                            [constants_js_1.CONFIG_KEY_PREVIOUS_STATE]: checkpoint.channel_values[constants_js_1.PREVIOUS],
                             checkpoint_id: undefined,
                             checkpoint_ns: taskCheckpointNamespace,
                         },
@@ -350,7 +440,12 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
             }
         }
         else {
-            return { id: taskId, name: packet.node, interrupts: [], path: taskPath };
+            return {
+                id: taskId,
+                name: packet.node,
+                interrupts: [],
+                path: taskPath,
+            };
         }
     }
     else if (taskPath[0] === constants_js_1.PULL) {
@@ -438,11 +533,15 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
                                     ...configurable[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP],
                                     [parentNamespace]: checkpoint.id,
                                 },
-                                [constants_js_1.CONFIG_KEY_WRITES]: [
+                                [constants_js_1.CONFIG_KEY_SCRATCHPAD]: _scratchpad([
                                     ...(pendingWrites || []),
-                                    ...(configurable[constants_js_1.CONFIG_KEY_WRITES] || []),
-                                ].filter((w) => w[0] === constants_js_1.NULL_TASK_ID || w[0] === taskId),
-                                [constants_js_1.CONFIG_KEY_SCRATCHPAD]: {},
+                                    ...(configurable[constants_js_1.CONFIG_KEY_SCRATCHPAD]?.resume || []).map((v) => [
+                                        taskId,
+                                        constants_js_1.RESUME,
+                                        v,
+                                    ]),
+                                ], taskId),
+                                [constants_js_1.CONFIG_KEY_PREVIOUS_STATE]: checkpoint.channel_values[constants_js_1.PREVIOUS],
                                 checkpoint_id: undefined,
                                 checkpoint_ns: taskCheckpointNamespace,
                             },
@@ -533,3 +632,13 @@ function _procInput(step, proc, managed, channels, forExecution) {
     }
     return val;
 }
+function _scratchpad(pendingWrites, taskId) {
+    return {
+        callCounter: 0,
+        interruptCounter: -1,
+        resume: pendingWrites
+            .filter(([writeTaskId, chan]) => writeTaskId === taskId && chan === constants_js_1.RESUME)
+            .flatMap(([_writeTaskId, _chan, resume]) => resume),
+        nullResume: pendingWrites.find(([writeTaskId, chan]) => writeTaskId === constants_js_1.NULL_TASK_ID && chan === constants_js_1.RESUME)?.[2],
+    };
+}

package/dist/pregel/algo.d.ts

@@ -1,10 +1,11 @@
 import { RunnableConfig } from "@langchain/core/runnables";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, type PendingWriteValue, BaseStore } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, type PendingWriteValue, BaseStore, CheckpointPendingWrite } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelNode } from "./read.js";
-import { PregelExecutableTask, PregelTaskDescription } from "./types.js";
+import { PregelExecutableTask, PregelTaskDescription, SimpleTaskPath, TaskPath } from "./types.js";
 import { ManagedValueMapping } from "../managed/base.js";
+import { IterableReadableWritableStream } from "./stream.js";
 /**
  * Construct a type with a set of properties K of type T
  */
@@ -15,7 +16,7 @@ export type WritesProtocol<C = string> = {
     name: string;
     writes: PendingWrite<C>[];
     triggers: string[];
-    path?: [string, ...(string | number)[]];
+    path?: TaskPath;
 };
 export declare const increment: (current?: number) => number;
 export declare function shouldInterrupt<N extends PropertyKey, C extends PropertyKey>(checkpoint: Checkpoint, interruptNodes: All | N[], tasks: PregelExecutableTask<N, C>[]): boolean;
@@ -28,6 +29,7 @@ export type NextTaskExtraFields = {
     checkpointer?: BaseCheckpointSaver;
     manager?: CallbackManagerForChainRun;
     store?: BaseStore;
+    stream?: IterableReadableWritableStream;
 };
 export type NextTaskExtraFieldsWithStore = NextTaskExtraFields & {
     store?: BaseStore;
@@ -37,6 +39,6 @@ export type NextTaskExtraFieldsWithoutStore = NextTaskExtraFields & {
 };
 export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFieldsWithoutStore): Record<string, PregelTaskDescription>;
 export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFieldsWithStore): Record<string, PregelExecutableTask<keyof Nn, keyof Cc>>;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, string | number], checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription | undefined;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, ...(string | number)[]], checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, ...(string | number)[]], checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFieldsWithStore): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: SimpleTaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFieldsWithStore): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;