@langchain/langgraph 0.2.7 → 0.2.9

package/dist/constants.cjs

@@ -1,18 +1,24 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.TASK_NAMESPACE = exports.TASKS = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.CONFIG_KEY_STORE = exports.INTERRUPT = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = void 0;
+exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_END = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.TASK_NAMESPACE = exports.PULL = exports.PUSH = exports.TASKS = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.INTERRUPT = exports.CONFIG_KEY_CHECKPOINT_MAP = exports.CONFIG_KEY_STREAM = exports.CONFIG_KEY_TASK_ID = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_STORE = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = void 0;
 exports.INPUT = "__input__";
 exports.ERROR = "__error__";
 exports.CONFIG_KEY_SEND = "__pregel_send";
 exports.CONFIG_KEY_READ = "__pregel_read";
 exports.CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
+exports.CONFIG_KEY_STORE = "__pregel_store";
 exports.CONFIG_KEY_RESUMING = "__pregel_resuming";
+exports.CONFIG_KEY_TASK_ID = "__pregel_task_id";
+exports.CONFIG_KEY_STREAM = "__pregel_stream";
+// this one is part of public API
+exports.CONFIG_KEY_CHECKPOINT_MAP = "checkpoint_map";
 exports.INTERRUPT = "__interrupt__";
-exports.CONFIG_KEY_STORE = "__pregel_store";
 exports.RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 exports.RECURSION_LIMIT_DEFAULT = 25;
 exports.TAG_HIDDEN = "langsmith:hidden";
 exports.TASKS = "__pregel_tasks";
+exports.PUSH = "__pregel_push";
+exports.PULL = "__pregel_pull";
 exports.TASK_NAMESPACE = "6ba7b831-9dad-11d1-80b4-00c04fd430c8";
 exports.RESERVED = [
     exports.INTERRUPT,
@@ -22,9 +28,13 @@ exports.RESERVED = [
     exports.CONFIG_KEY_READ,
     exports.CONFIG_KEY_CHECKPOINTER,
     exports.CONFIG_KEY_RESUMING,
+    exports.CONFIG_KEY_TASK_ID,
+    exports.CONFIG_KEY_STREAM,
+    exports.CONFIG_KEY_CHECKPOINT_MAP,
     exports.INPUT,
 ];
 exports.CHECKPOINT_NAMESPACE_SEPARATOR = "|";
+exports.CHECKPOINT_NAMESPACE_END = ":";
 function _isSendInterface(x) {
     const operation = x;
     return typeof operation.node === "string" && operation.args !== undefined;

package/dist/constants.d.ts

@@ -3,16 +3,22 @@ export declare const ERROR = "__error__";
 export declare const CONFIG_KEY_SEND = "__pregel_send";
 export declare const CONFIG_KEY_READ = "__pregel_read";
 export declare const CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
+export declare const CONFIG_KEY_STORE = "__pregel_store";
 export declare const CONFIG_KEY_RESUMING = "__pregel_resuming";
+export declare const CONFIG_KEY_TASK_ID = "__pregel_task_id";
+export declare const CONFIG_KEY_STREAM = "__pregel_stream";
+export declare const CONFIG_KEY_CHECKPOINT_MAP = "checkpoint_map";
 export declare const INTERRUPT = "__interrupt__";
-export declare const CONFIG_KEY_STORE = "__pregel_store";
 export declare const RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 export declare const RECURSION_LIMIT_DEFAULT = 25;
 export declare const TAG_HIDDEN = "langsmith:hidden";
 export declare const TASKS = "__pregel_tasks";
+export declare const PUSH = "__pregel_push";
+export declare const PULL = "__pregel_pull";
 export declare const TASK_NAMESPACE = "6ba7b831-9dad-11d1-80b4-00c04fd430c8";
 export declare const RESERVED: string[];
 export declare const CHECKPOINT_NAMESPACE_SEPARATOR = "|";
+export declare const CHECKPOINT_NAMESPACE_END = ":";
 export interface SendInterface {
     node: string;
     args: any;

package/dist/constants.js

@@ -3,13 +3,19 @@ export const ERROR = "__error__";
 export const CONFIG_KEY_SEND = "__pregel_send";
 export const CONFIG_KEY_READ = "__pregel_read";
 export const CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
+export const CONFIG_KEY_STORE = "__pregel_store";
 export const CONFIG_KEY_RESUMING = "__pregel_resuming";
+export const CONFIG_KEY_TASK_ID = "__pregel_task_id";
+export const CONFIG_KEY_STREAM = "__pregel_stream";
+// this one is part of public API
+export const CONFIG_KEY_CHECKPOINT_MAP = "checkpoint_map";
 export const INTERRUPT = "__interrupt__";
-export const CONFIG_KEY_STORE = "__pregel_store";
 export const RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 export const RECURSION_LIMIT_DEFAULT = 25;
 export const TAG_HIDDEN = "langsmith:hidden";
 export const TASKS = "__pregel_tasks";
+export const PUSH = "__pregel_push";
+export const PULL = "__pregel_pull";
 export const TASK_NAMESPACE = "6ba7b831-9dad-11d1-80b4-00c04fd430c8";
 export const RESERVED = [
     INTERRUPT,
@@ -19,9 +25,13 @@ export const RESERVED = [
     CONFIG_KEY_READ,
     CONFIG_KEY_CHECKPOINTER,
     CONFIG_KEY_RESUMING,
+    CONFIG_KEY_TASK_ID,
+    CONFIG_KEY_STREAM,
+    CONFIG_KEY_CHECKPOINT_MAP,
     INPUT,
 ];
 export const CHECKPOINT_NAMESPACE_SEPARATOR = "|";
+export const CHECKPOINT_NAMESPACE_END = ":";
 export function _isSendInterface(x) {
     const operation = x;
     return typeof operation.node === "string" && operation.args !== undefined;

package/dist/errors.cjs

@@ -55,10 +55,11 @@ class NodeInterrupt extends GraphInterrupt {
 }
 exports.NodeInterrupt = NodeInterrupt;
 function isGraphInterrupt(e) {
-    return [
-        GraphInterrupt.unminifiable_name,
-        NodeInterrupt.unminifiable_name,
-    ].includes(e.name);
+    return (e !== undefined &&
+        [
+            GraphInterrupt.unminifiable_name,
+            NodeInterrupt.unminifiable_name,
+        ].includes(e.name));
 }
 exports.isGraphInterrupt = isGraphInterrupt;
 class EmptyInputError extends Error {

package/dist/errors.d.ts

@@ -17,7 +17,7 @@ export declare class NodeInterrupt extends GraphInterrupt {
     constructor(message: string);
     static get unminifiable_name(): string;
 }
-export declare function isGraphInterrupt(e: GraphInterrupt | Error): e is GraphInterrupt;
+export declare function isGraphInterrupt(e?: GraphInterrupt | Error): e is GraphInterrupt;
 export declare class EmptyInputError extends Error {
     constructor(message?: string);
     static get unminifiable_name(): string;

package/dist/errors.js

@@ -48,10 +48,11 @@ export class NodeInterrupt extends GraphInterrupt {
     }
 }
 export function isGraphInterrupt(e) {
-    return [
-        GraphInterrupt.unminifiable_name,
-        NodeInterrupt.unminifiable_name,
-    ].includes(e.name);
+    return (e !== undefined &&
+        [
+            GraphInterrupt.unminifiable_name,
+            NodeInterrupt.unminifiable_name,
+        ].includes(e.name));
 }
 export class EmptyInputError extends Error {
     constructor(message) {

package/dist/graph/graph.cjs

@@ -123,8 +123,13 @@ class Graph {
         return this.edges;
     }
     addNode(key, action, options) {
-        if (key.includes(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR)) {
-            throw new Error(`"${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}" is a reserved character and is not allowed in node names.`);
+        for (const reservedChar of [
+            constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR,
+            constants_js_1.CHECKPOINT_NAMESPACE_END,
+        ]) {
+            if (key.includes(reservedChar)) {
+                throw new Error(`"${reservedChar}" is a reserved character and is not allowed in node names.`);
+            }
         }
         this.warnIfCompiled(`Adding a node to a graph that has already been compiled. This will not be reflected in the compiled graph.`);
         if (key in this.nodes) {

package/dist/graph/graph.js

@@ -6,7 +6,7 @@ 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_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
+import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
 import { RunnableCallable } from "../utils.js";
 import { InvalidUpdateError, NodeInterrupt } from "../errors.js";
 /** Special reserved node name denoting the start of a graph. */
@@ -119,8 +119,13 @@ export class Graph {
         return this.edges;
     }
     addNode(key, action, options) {
-        if (key.includes(CHECKPOINT_NAMESPACE_SEPARATOR)) {
-            throw new Error(`"${CHECKPOINT_NAMESPACE_SEPARATOR}" is a reserved character and is not allowed in node names.`);
+        for (const reservedChar of [
+            CHECKPOINT_NAMESPACE_SEPARATOR,
+            CHECKPOINT_NAMESPACE_END,
+        ]) {
+            if (key.includes(reservedChar)) {
+                throw new Error(`"${reservedChar}" is a reserved character and is not allowed in node names.`);
+            }
         }
         this.warnIfCompiled(`Adding a node to a graph that has already been compiled. This will not be reflected in the compiled graph.`);
         if (key in this.nodes) {

package/dist/graph/message.cjs

@@ -19,11 +19,13 @@ function messagesStateReducer(left, right) {
     for (const m of leftMessages) {
         if (m.id === null || m.id === undefined) {
             m.id = (0, uuid_1.v4)();
+            m.lc_kwargs.id = m.id;
         }
     }
     for (const m of rightMessages) {
         if (m.id === null || m.id === undefined) {
             m.id = (0, uuid_1.v4)();
+            m.lc_kwargs.id = m.id;
         }
     }
     // merge

package/dist/graph/message.js

@@ -16,11 +16,13 @@ export function messagesStateReducer(left, right) {
     for (const m of leftMessages) {
         if (m.id === null || m.id === undefined) {
             m.id = v4();
+            m.lc_kwargs.id = m.id;
         }
     }
     for (const m of rightMessages) {
         if (m.id === null || m.id === undefined) {
             m.id = v4();
+            m.lc_kwargs.id = m.id;
         }
     }
     // merge

package/dist/graph/state.cjs

@@ -124,7 +124,7 @@ class StateGraph extends graph_js_1.Graph {
             writable: true,
             value: new Map()
         });
-        /** @internal */
+        /** @internal Used only for typing. */
         Object.defineProperty(this, "_configSchema", {
             enumerable: true,
             configurable: true,
@@ -198,8 +198,13 @@ class StateGraph extends graph_js_1.Graph {
         if (key in this.channels) {
             throw new Error(`${key} is already being used as a state attribute (a.k.a. a channel), cannot also be used as a node name.`);
         }
-        if (key.includes(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR)) {
-            throw new Error(`"${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}" is a reserved character and is not allowed in node names.`);
+        for (const reservedChar of [
+            constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR,
+            constants_js_1.CHECKPOINT_NAMESPACE_END,
+        ]) {
+            if (key.includes(reservedChar)) {
+                throw new Error(`"${reservedChar}" is a reserved character and is not allowed in node names.`);
+            }
         }
         this.warnIfCompiled(`Adding a node to a graph that has already been compiled. This will not be reflected in the compiled graph.`);
         if (key in this.nodes) {
@@ -259,9 +264,6 @@ class StateGraph extends graph_js_1.Graph {
         // create empty compiled graph
         const compiled = new CompiledStateGraph({
             builder: this,
-            configKeys: this._configSchema
-                ? Object.keys(this._configSchema)
-                : undefined,
             checkpointer,
             interruptAfter,
             interruptBefore,
@@ -388,7 +390,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             }
         }
         else if (start === graph_js_1.START) {
-            const channelName = `start:${end}`;
+            const channelName = `${graph_js_1.START}:${end}`;
             // register channel
             this.channels[channelName] =
                 new ephemeral_value_js_1.EphemeralValue();

package/dist/graph/state.d.ts

@@ -3,7 +3,7 @@ import { All, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { END, CompiledGraph, Graph, START, Branch, AddNodeOptions, NodeSpec } from "./graph.js";
 import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType } from "./annotation.js";
-import type { RetryPolicy } from "../pregel/utils.js";
+import type { RetryPolicy } from "../pregel/utils/index.js";
 import { BaseStore } from "../store/base.js";
 import { ManagedValueSpec } from "../managed/base.js";
 export type ChannelReducers<Channels extends object> = {
@@ -109,7 +109,7 @@ export declare class StateGraph<SD extends StateDefinition | unknown, S = SD ext
      * @internal
      */
     _schemaDefinitions: Map<any, any>;
-    /** @internal */
+    /** @internal Used only for typing. */
     _configSchema: C | undefined;
     constructor(fields: SD extends StateDefinition ? SD | AnnotationRoot<SD> | StateGraphArgs<S> | StateGraphArgsWithStateSchema<SD, I, O> | StateGraphArgsWithInputOutputSchemas<SD, O> : StateGraphArgs<S>, configSchema?: AnnotationRoot<C>);
     get allEdges(): Set<[string, string]>;

package/dist/graph/state.js

@@ -7,7 +7,7 @@ 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 { _isSend, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
+import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
 import { InvalidUpdateError } from "../errors.js";
 import { getChannel, } from "./annotation.js";
 import { isConfiguredManagedValue } from "../managed/base.js";
@@ -121,7 +121,7 @@ export class StateGraph extends Graph {
             writable: true,
             value: new Map()
         });
-        /** @internal */
+        /** @internal Used only for typing. */
         Object.defineProperty(this, "_configSchema", {
             enumerable: true,
             configurable: true,
@@ -195,8 +195,13 @@ export class StateGraph extends Graph {
         if (key in this.channels) {
             throw new Error(`${key} is already being used as a state attribute (a.k.a. a channel), cannot also be used as a node name.`);
         }
-        if (key.includes(CHECKPOINT_NAMESPACE_SEPARATOR)) {
-            throw new Error(`"${CHECKPOINT_NAMESPACE_SEPARATOR}" is a reserved character and is not allowed in node names.`);
+        for (const reservedChar of [
+            CHECKPOINT_NAMESPACE_SEPARATOR,
+            CHECKPOINT_NAMESPACE_END,
+        ]) {
+            if (key.includes(reservedChar)) {
+                throw new Error(`"${reservedChar}" is a reserved character and is not allowed in node names.`);
+            }
         }
         this.warnIfCompiled(`Adding a node to a graph that has already been compiled. This will not be reflected in the compiled graph.`);
         if (key in this.nodes) {
@@ -256,9 +261,6 @@ export class StateGraph extends Graph {
         // create empty compiled graph
         const compiled = new CompiledStateGraph({
             builder: this,
-            configKeys: this._configSchema
-                ? Object.keys(this._configSchema)
-                : undefined,
             checkpointer,
             interruptAfter,
             interruptBefore,
@@ -384,7 +386,7 @@ export class CompiledStateGraph extends CompiledGraph {
             }
         }
         else if (start === START) {
-            const channelName = `start:${end}`;
+            const channelName = `${START}:${end}`;
             // register channel
             this.channels[channelName] =
                 new EphemeralValue();

package/dist/managed/shared_value.cjs

@@ -44,7 +44,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
             this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
         }
         else {
-            throw new Error(`Scope ${this.scope} for shared state key not in config.configurable`);
+            throw new Error(`Required scope "${this.scope}" for shared state key was not passed in "config.configurable".`);
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/managed/shared_value.js

@@ -41,7 +41,7 @@ export class SharedValue extends WritableManagedValue {
             this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
         }
         else {
-            throw new Error(`Scope ${this.scope} for shared state key not in config.configurable`);
+            throw new Error(`Required scope "${this.scope}" for shared state key was not passed in "config.configurable".`);
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/pregel/algo.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._prepareNextTasks = exports._applyWrites = exports._localWrite = exports._localRead = exports.shouldInterrupt = exports.increment = void 0;
+exports._prepareSingleTask = exports._prepareNextTasks = exports._applyWrites = exports._localWrite = exports._localRead = exports.shouldInterrupt = exports.increment = void 0;
 /* eslint-disable no-param-reassign */
 const runnables_1 = require("@langchain/core/runnables");
 const langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
@@ -8,7 +8,7 @@ const base_js_1 = require("../channels/base.cjs");
 const io_js_1 = require("./io.cjs");
 const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
-const utils_js_1 = require("./utils.cjs");
+const index_js_1 = require("./utils/index.cjs");
 const increment = (current) => {
     return current !== undefined ? current + 1 : 1;
 };
@@ -204,48 +204,84 @@ getNextVersion) {
 }
 exports._applyWrites = _applyWrites;
 function _prepareNextTasks(checkpoint, processes, channels, managed, config, forExecution, extra) {
-    const parentNamespace = config.configurable?.checkpoint_ns ?? "";
-    const tasks = [];
-    const taskDescriptions = [];
-    const { step, isResuming = false, checkpointer, manager } = extra;
-    for (const packet of checkpoint.pending_sends) {
+    const tasks = {};
+    // Consume pending packets
+    for (let i = 0; i < checkpoint.pending_sends.length; i += 1) {
+        const task = _prepareSingleTask([constants_js_1.PUSH, i], checkpoint, processes, channels, managed, config, forExecution, extra);
+        if (task !== undefined) {
+            tasks[task.id] = task;
+        }
+    }
+    // Check if any processes should be run in next step
+    // If so, prepare the values to be passed to them
+    for (const name of Object.keys(processes)) {
+        const task = _prepareSingleTask([constants_js_1.PULL, name], checkpoint, processes, channels, managed, config, forExecution, extra);
+        if (task !== undefined) {
+            tasks[task.id] = task;
+        }
+    }
+    return tasks;
+}
+exports._prepareNextTasks = _prepareNextTasks;
+function _prepareSingleTask(taskPath, checkpoint, processes, channels, managed, config, forExecution, extra) {
+    const { step, checkpointer, manager } = extra;
+    const configurable = config.configurable ?? {};
+    const parentNamespace = configurable.checkpoint_ns ?? "";
+    if (taskPath[0] === constants_js_1.PUSH) {
+        const index = typeof taskPath[1] === "number" ? taskPath[1] : parseInt(taskPath[1], 10);
+        if (index >= checkpoint.pending_sends.length) {
+            return undefined;
+        }
+        const packet = checkpoint.pending_sends[index];
         if (!(0, constants_js_1._isSendInterface)(packet)) {
             console.warn(`Ignoring invalid packet ${JSON.stringify(packet)} in pending sends.`);
-            continue;
+            return undefined;
         }
         if (!(packet.node in processes)) {
             console.warn(`Ignoring unknown node name ${packet.node} in pending sends.`);
-            continue;
+            return undefined;
         }
-        const triggers = [constants_js_1.TASKS];
-        const metadata = (0, utils_js_1._getIdMetadata)({
-            langgraph_step: step,
-            langgraph_node: packet.node,
-            langgraph_triggers: triggers,
-            langgraph_task_idx: forExecution ? tasks.length : taskDescriptions.length,
-        });
+        const triggers = [constants_js_1.PUSH];
         const checkpointNamespace = parentNamespace === ""
             ? packet.node
             : `${parentNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${packet.node}`;
-        const taskId = (0, langgraph_checkpoint_1.uuid5)(JSON.stringify([checkpointNamespace, metadata]), checkpoint.id);
+        const taskId = (0, langgraph_checkpoint_1.uuid5)(JSON.stringify([
+            checkpointNamespace,
+            step.toString(),
+            packet.node,
+            constants_js_1.PUSH,
+            index.toString(),
+        ]), checkpoint.id);
+        const taskCheckpointNamespace = `${checkpointNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_END}${taskId}`;
+        let metadata = {
+            langgraph_step: step,
+            langgraph_node: packet.node,
+            langgraph_triggers: triggers,
+            langgraph_path: taskPath,
+            langgraph_checkpoint_ns: taskCheckpointNamespace,
+        };
         if (forExecution) {
             const proc = processes[packet.node];
             const node = proc.getNode();
             if (node !== undefined) {
-                const writes = [];
                 managed.replaceRuntimePlaceholders(step, packet.args);
-                tasks.push({
+                if (proc.metadata !== undefined) {
+                    metadata = { ...metadata, ...proc.metadata };
+                }
+                const writes = [];
+                return {
                     name: packet.node,
                     input: packet.args,
                     proc: node,
                     writes,
-                    triggers,
-                    config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, processes[packet.node].config, {
+                    config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
                         metadata,
+                        tags: proc.tags,
                     }), {
                         runName: packet.node,
                         callbacks: manager?.getChild(`graph:step:${step}`),
                         configurable: {
+                            [constants_js_1.CONFIG_KEY_TASK_ID]: taskId,
                             // eslint-disable-next-line @typescript-eslint/no-explicit-any
                             [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, channels, managed, writes_),
                             [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
@@ -253,24 +289,36 @@ function _prepareNextTasks(checkpoint, processes, channels, managed, config, for
                                 writes: writes,
                                 triggers,
                             }, 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,
+                            },
+                            checkpoint_id: undefined,
+                            checkpoint_ns: taskCheckpointNamespace,
                         },
                     }),
-                    id: taskId,
+                    triggers,
                     retry_policy: proc.retryPolicy,
-                });
+                    id: taskId,
+                    path: taskPath,
+                };
             }
         }
         else {
-            taskDescriptions.push({ id: taskId, name: packet.node, interrupts: [] });
+            return { id: taskId, name: packet.node, interrupts: [], path: taskPath };
         }
     }
-    // Check if any processes should be run in next step
-    // If so, prepare the values to be passed to them
-    const nullVersion = (0, utils_js_1.getNullChannelVersion)(checkpoint.channel_versions);
-    if (nullVersion === undefined) {
-        return forExecution ? tasks : taskDescriptions;
-    }
-    for (const [name, proc] of Object.entries(processes)) {
+    else if (taskPath[0] === constants_js_1.PULL) {
+        const name = taskPath[1].toString();
+        const proc = processes[name];
+        if (proc === undefined) {
+            return undefined;
+        }
+        const nullVersion = (0, index_js_1.getNullChannelVersion)(checkpoint.channel_versions);
+        if (nullVersion === undefined) {
+            return undefined;
+        }
         const seen = checkpoint.versions_seen[name] ?? {};
         const triggers = proc.triggers
             .filter((chan) => {
@@ -288,60 +336,77 @@ function _prepareNextTasks(checkpoint, processes, channels, managed, config, for
         if (triggers.length > 0) {
             const val = _procInput(step, proc, managed, channels, forExecution);
             if (val === undefined) {
-                continue;
+                return undefined;
             }
-            const metadata = (0, utils_js_1._getIdMetadata)({
-                langgraph_step: step,
-                langgraph_node: name,
-                langgraph_triggers: triggers,
-                langgraph_task_idx: forExecution
-                    ? tasks.length
-                    : taskDescriptions.length,
-            });
             const checkpointNamespace = parentNamespace === ""
                 ? name
                 : `${parentNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${name}`;
-            const taskId = (0, langgraph_checkpoint_1.uuid5)(JSON.stringify([checkpointNamespace, metadata]), checkpoint.id);
+            const taskId = (0, langgraph_checkpoint_1.uuid5)(JSON.stringify([
+                checkpointNamespace,
+                step.toString(),
+                name,
+                constants_js_1.PULL,
+                triggers,
+            ]), checkpoint.id);
+            const taskCheckpointNamespace = `${checkpointNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_END}${taskId}`;
+            let metadata = {
+                langgraph_step: step,
+                langgraph_node: name,
+                langgraph_triggers: triggers,
+                langgraph_path: taskPath,
+                langgraph_checkpoint_ns: taskCheckpointNamespace,
+            };
             if (forExecution) {
                 const node = proc.getNode();
                 if (node !== undefined) {
+                    if (proc.metadata !== undefined) {
+                        metadata = { ...metadata, ...proc.metadata };
+                    }
                     const writes = [];
-                    tasks.push({
+                    const taskCheckpointNamespace = `${checkpointNamespace}${constants_js_1.CHECKPOINT_NAMESPACE_END}${taskId}`;
+                    return {
                         name,
                         input: val,
                         proc: node,
                         writes,
-                        triggers,
-                        config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, proc.config, { metadata }), {
+                        config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, { metadata, tags: proc.tags }), {
                             runName: name,
                             callbacks: manager?.getChild(`graph:step:${step}`),
                             configurable: {
+                                [constants_js_1.CONFIG_KEY_TASK_ID]: taskId,
                                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                                [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, channels, managed, writes_),
+                                [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => {
+                                    writes.push(...items);
+                                }, processes, channels, managed, writes_),
                                 [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
                                     name,
                                     writes: writes,
                                     triggers,
                                 }, select_, fresh_),
-                                [constants_js_1.CONFIG_KEY_CHECKPOINTER]: checkpointer,
-                                [constants_js_1.CONFIG_KEY_RESUMING]: isResuming,
-                                checkpoint_id: checkpoint.id,
-                                checkpoint_ns: checkpointNamespace,
+                                [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,
+                                },
+                                checkpoint_id: undefined,
+                                checkpoint_ns: taskCheckpointNamespace,
                             },
                         }),
-                        id: taskId,
+                        triggers,
                         retry_policy: proc.retryPolicy,
-                    });
+                        id: taskId,
+                        path: taskPath,
+                    };
                 }
             }
             else {
-                taskDescriptions.push({ id: taskId, name, interrupts: [] });
+                return { id: taskId, name, interrupts: [], path: taskPath };
             }
         }
     }
-    return forExecution ? tasks : taskDescriptions;
+    return undefined;
 }
-exports._prepareNextTasks = _prepareNextTasks;
+exports._prepareSingleTask = _prepareSingleTask;
 function _procInput(step, proc, managed, channels, forExecution) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let val;

package/dist/pregel/algo.d.ts

@@ -27,5 +27,8 @@ export type NextTaskExtraFields = {
     checkpointer?: BaseCheckpointSaver;
     manager?: CallbackManagerForChainRun;
 };
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription[];
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc>[];
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): Record<string, PregelTaskDescription>;
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): 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, 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, 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, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFields): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;