@langchain/langgraph 0.2.13 → 0.2.15

package/dist/errors.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.InvalidUpdateError = exports.EmptyChannelError = exports.EmptyInputError = exports.isGraphInterrupt = exports.NodeInterrupt = exports.GraphInterrupt = exports.GraphValueError = exports.GraphRecursionError = void 0;
+exports.getSubgraphsSeenSet = exports.MultipleSubgraphsError = exports.InvalidUpdateError = exports.EmptyChannelError = exports.EmptyInputError = exports.isGraphInterrupt = exports.NodeInterrupt = exports.GraphInterrupt = exports.GraphValueError = exports.GraphRecursionError = void 0;
 class GraphRecursionError extends Error {
     constructor(message) {
         super(message);
@@ -92,3 +92,27 @@ class InvalidUpdateError extends Error {
     }
 }
 exports.InvalidUpdateError = InvalidUpdateError;
+class MultipleSubgraphsError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "MultipleSubgraphError";
+    }
+    static get unminifiable_name() {
+        return "MultipleSubgraphError";
+    }
+}
+exports.MultipleSubgraphsError = MultipleSubgraphsError;
+/**
+ * Used for subgraph detection.
+ */
+const getSubgraphsSeenSet = () => {
+    if (
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")] === undefined) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")] = new Set();
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    return globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")];
+};
+exports.getSubgraphsSeenSet = getSubgraphsSeenSet;

package/dist/errors.d.ts

@@ -30,3 +30,11 @@ export declare class InvalidUpdateError extends Error {
     constructor(message?: string);
     static get unminifiable_name(): string;
 }
+export declare class MultipleSubgraphsError extends Error {
+    constructor(message?: string);
+    static get unminifiable_name(): string;
+}
+/**
+ * Used for subgraph detection.
+ */
+export declare const getSubgraphsSeenSet: () => any;

package/dist/errors.js

@@ -81,3 +81,25 @@ export class InvalidUpdateError extends Error {
         return "InvalidUpdateError";
     }
 }
+export class MultipleSubgraphsError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "MultipleSubgraphError";
+    }
+    static get unminifiable_name() {
+        return "MultipleSubgraphError";
+    }
+}
+/**
+ * Used for subgraph detection.
+ */
+export const getSubgraphsSeenSet = () => {
+    if (
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")] === undefined) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")] = new Set();
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    return globalThis[Symbol.for("LG_CHECKPOINT_SEEN_NS_SET")];
+};

package/dist/graph/graph.cjs

@@ -93,6 +93,7 @@ class Graph {
             writable: true,
             value: void 0
         });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         Object.defineProperty(this, "branches", {
             enumerable: true,
             configurable: true,

package/dist/graph/graph.d.ts

@@ -7,21 +7,23 @@ import type { PregelParams } from "../pregel/types.js";
 import { BaseChannel } from "../channels/base.js";
 import { Send } 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> {
+export interface BranchOptions<IO, N extends string, CallOptions extends LangGraphRunnableConfig = LangGraphRunnableConfig> {
     source: N;
-    path: Branch<IO, N>["condition"];
+    path: Branch<IO, N, CallOptions>["condition"];
     pathMap?: Record<string, N | typeof END> | (N | typeof END)[];
 }
-export declare class Branch<IO, N extends string> {
-    condition: (input: IO, config?: RunnableConfig) => string | Send | (string | Send)[] | Promise<string | Send | (string | Send)[]>;
+export declare class Branch<IO, N extends string, CallOptions extends LangGraphRunnableConfig = LangGraphRunnableConfig> {
+    condition: (input: IO, config: CallOptions) => string | Send | (string | Send)[] | Promise<string | Send | (string | Send)[]>;
     ends?: Record<string, N | typeof END>;
-    constructor(options: Omit<BranchOptions<IO, N>, "source">);
-    compile(writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: RunnableConfig) => IO): RunnableCallable<unknown, unknown>;
-    _route(input: IO, config: RunnableConfig, writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: RunnableConfig) => IO): Promise<Runnable | undefined>;
+    constructor(options: Omit<BranchOptions<IO, N, CallOptions>, "source">);
+    compile(writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: CallOptions) => IO): RunnableCallable<unknown, unknown>;
+    _route(input: IO, config: CallOptions, writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: CallOptions) => IO): Promise<Runnable | undefined>;
 }
 export type NodeSpec<RunInput, RunOutput> = {
     runnable: Runnable<RunInput, RunOutput>;
@@ -30,19 +32,19 @@ export type NodeSpec<RunInput, RunOutput> = {
 export type AddNodeOptions = {
     metadata?: Record<string, unknown>;
 };
-export declare class Graph<N extends string = typeof END, RunInput = any, RunOutput = any, NodeSpecType extends NodeSpec<RunInput, RunOutput> = NodeSpec<RunInput, RunOutput>> {
+export declare class Graph<N extends string = typeof END, RunInput = any, RunOutput = any, NodeSpecType extends NodeSpec<RunInput, RunOutput> = NodeSpec<RunInput, RunOutput>, C extends StateDefinition = StateDefinition> {
     nodes: Record<N, NodeSpecType>;
     edges: Set<[N | typeof START, N | typeof END]>;
-    branches: Record<string, Record<string, Branch<RunInput, N>>>;
+    branches: Record<string, Record<string, Branch<RunInput, N, any>>>;
     entryPoint?: string;
     compiled: boolean;
     constructor();
     protected warnIfCompiled(message: string): void;
     get allEdges(): Set<[string, string]>;
-    addNode<K extends string, NodeInput = RunInput>(key: K, action: RunnableLike<NodeInput, RunOutput>, options?: AddNodeOptions): Graph<N | K, RunInput, RunOutput>;
+    addNode<K extends string, NodeInput = RunInput>(key: K, action: RunnableLike<NodeInput, RunOutput extends object ? RunOutput & Record<string, any> : RunOutput, LangGraphRunnableConfig<StateType<C>>>, options?: AddNodeOptions): Graph<N | K, RunInput, RunOutput>;
     addEdge(startKey: N | typeof START, endKey: N | typeof END): this;
-    addConditionalEdges(source: BranchOptions<RunInput, N>): this;
-    addConditionalEdges(source: N, path: Branch<RunInput, N>["condition"], pathMap?: BranchOptions<RunInput, N>["pathMap"]): this;
+    addConditionalEdges(source: BranchOptions<RunInput, N, LangGraphRunnableConfig<StateType<C>>>): this;
+    addConditionalEdges(source: N, path: Branch<RunInput, N, LangGraphRunnableConfig<StateType<C>>>["condition"], pathMap?: BranchOptions<RunInput, N, LangGraphRunnableConfig<StateType<C>>>["pathMap"]): this;
     /**
      * @deprecated use `addEdge(START, key)` instead
      */
@@ -52,7 +54,7 @@ export declare class Graph<N extends string = typeof END, RunInput = any, RunOut
      */
     setFinishPoint(key: N): this;
     compile({ checkpointer, interruptBefore, interruptAfter, }?: {
-        checkpointer?: BaseCheckpointSaver;
+        checkpointer?: BaseCheckpointSaver | false;
         interruptBefore?: N[] | All;
         interruptAfter?: N[] | All;
     }): CompiledGraph<N>;

package/dist/graph/graph.js

@@ -89,6 +89,7 @@ export class Graph {
             writable: true,
             value: void 0
         });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         Object.defineProperty(this, "branches", {
             enumerable: true,
             configurable: true,

package/dist/graph/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.messagesStateReducer = exports.MessageGraph = 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.START = exports.END = 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; } });
@@ -10,6 +10,7 @@ Object.defineProperty(exports, "START", { enumerable: true, get: function () { r
 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; } });
+Object.defineProperty(exports, "CompiledStateGraph", { enumerable: true, get: function () { return state_js_1.CompiledStateGraph; } });
 var message_js_1 = require("./message.cjs");
 Object.defineProperty(exports, "MessageGraph", { enumerable: true, get: function () { return message_js_1.MessageGraph; } });
 Object.defineProperty(exports, "messagesStateReducer", { enumerable: true, get: function () { return message_js_1.messagesStateReducer; } });

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 { type StateGraphArgs, StateGraph, type CompiledStateGraph, } from "./state.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 { StateGraph, } from "./state.js";
+export { StateGraph, CompiledStateGraph, } from "./state.js";
 export { MessageGraph, messagesStateReducer, } from "./message.js";

package/dist/graph/state.cjs

@@ -312,6 +312,11 @@ function _getChannels(schema) {
     }
     return channels;
 }
+/**
+ * Final result from building and compiling a {@link StateGraph}.
+ * Should not be instantiated directly, only using the StateGraph `.compile()`
+ * instance method.
+ */
 class CompiledStateGraph extends graph_js_1.CompiledGraph {
     attachNode(key, node) {
         const stateKeys = Object.keys(this.builder.channels);

package/dist/graph/state.d.ts

@@ -5,6 +5,7 @@ import { END, CompiledGraph, Graph, START, Branch, AddNodeOptions, NodeSpec } fr
 import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType } from "./annotation.js";
 import type { RetryPolicy } from "../pregel/utils/index.js";
 import { ManagedValueSpec } from "../managed/base.js";
+import type { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
 export type ChannelReducers<Channels extends object> = {
     [K in keyof Channels]: SingleReducer<Channels[K], any>;
 };
@@ -94,7 +95,7 @@ export type StateGraphArgsWithInputOutputSchemas<SD extends StateDefinition, O e
  * // }
  * ```
  */
-export declare class StateGraph<SD extends StateDefinition | unknown, S = SD extends StateDefinition ? StateType<SD> : SD, U = SD extends StateDefinition ? UpdateType<SD> : Partial<S>, N extends string = typeof START, I extends StateDefinition = SD extends StateDefinition ? SD : StateDefinition, O extends StateDefinition = SD extends StateDefinition ? SD : StateDefinition, C extends StateDefinition = StateDefinition> extends Graph<N, S, U, StateGraphNodeSpec<S, U>> {
+export declare class StateGraph<SD extends StateDefinition | unknown, S = SD extends StateDefinition ? StateType<SD> : SD, U = SD extends StateDefinition ? UpdateType<SD> : Partial<S>, N extends string = typeof START, I extends StateDefinition = SD extends StateDefinition ? SD : StateDefinition, O extends StateDefinition = SD extends StateDefinition ? SD : StateDefinition, C extends StateDefinition = StateDefinition> extends Graph<N, S, U, StateGraphNodeSpec<S, U>, C> {
     channels: Record<string, BaseChannel | ManagedValueSpec>;
     waitingEdges: Set<[N[], N]>;
     /** @internal */
@@ -113,15 +114,20 @@ export declare class StateGraph<SD extends StateDefinition | unknown, S = SD ext
     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]>;
     _addSchema(stateDefinition: StateDefinition): void;
-    addNode<K extends string, NodeInput = S>(key: K, action: RunnableLike<NodeInput, U extends object ? U & Record<string, any> : U>, options?: StateGraphAddNodeOptions): StateGraph<SD, S, U, N | K, I, O, C>;
+    addNode<K extends string, NodeInput = S>(key: K, action: RunnableLike<NodeInput, U extends object ? U & Record<string, any> : U, LangGraphRunnableConfig<StateType<C>>>, options?: StateGraphAddNodeOptions): StateGraph<SD, S, U, N | K, I, O, C>;
     addEdge(startKey: typeof START | N | N[], endKey: N | typeof END): this;
     compile({ checkpointer, store, interruptBefore, interruptAfter, }?: {
-        checkpointer?: BaseCheckpointSaver;
+        checkpointer?: BaseCheckpointSaver | false;
         store?: BaseStore;
         interruptBefore?: N[] | All;
         interruptAfter?: N[] | All;
     }): CompiledStateGraph<S, U, N, I, O, C>;
 }
+/**
+ * Final result from building and compiling a {@link StateGraph}.
+ * Should not be instantiated directly, only using the StateGraph `.compile()`
+ * instance method.
+ */
 export declare class CompiledStateGraph<S, U, N extends string = typeof START, I extends StateDefinition = StateDefinition, O extends StateDefinition = StateDefinition, C extends StateDefinition = StateDefinition> extends CompiledGraph<N, S, U, StateType<C>> {
     builder: StateGraph<unknown, S, U, N, I, O, C>;
     attachNode(key: typeof START, node?: never): void;

package/dist/graph/state.js

@@ -308,6 +308,11 @@ function _getChannels(schema) {
     }
     return channels;
 }
+/**
+ * Final result from building and compiling a {@link StateGraph}.
+ * Should not be instantiated directly, only using the StateGraph `.compile()`
+ * instance method.
+ */
 export class CompiledStateGraph extends CompiledGraph {
     attachNode(key, node) {
         const stateKeys = Object.keys(this.builder.channels);

package/dist/pregel/debug.cjs

@@ -4,6 +4,7 @@ exports.printStepWrites = exports.printStepTasks = exports.printStepCheckpoint =
 const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 const io_js_1 = require("./io.cjs");
+const subgraph_js_1 = require("./utils/subgraph.cjs");
 const COLORS_MAP = {
     blue: {
         start: "\x1b[34m",
@@ -101,7 +102,7 @@ function* mapDebugTaskResults(step, tasks, streamChannels) {
     }
 }
 exports.mapDebugTaskResults = mapDebugTaskResults;
-function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, tasks, pendingWrites) {
+function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, tasks, pendingWrites, parentConfig) {
     function formatConfig(config) {
         // make sure the config is consistent with Python
         const pyConfig = {};
@@ -123,11 +124,22 @@ function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, t
             pyConfig.tags = config.tags;
         return pyConfig;
     }
-    function getCurrentUTC() {
-        const now = new Date();
-        return new Date(now.getTime() - now.getTimezoneOffset() * 60 * 1000);
+    const parentNs = config.configurable?.checkpoint_ns;
+    const taskStates = {};
+    for (const task of tasks) {
+        if (!(0, subgraph_js_1.findSubgraphPregel)(task.proc))
+            continue;
+        let taskNs = `${task.name}:${task.id}`;
+        if (parentNs)
+            taskNs = `${parentNs}|${taskNs}`;
+        taskStates[task.id] = {
+            configurable: {
+                thread_id: config.configurable?.thread_id,
+                checkpoint_ns: taskNs,
+            },
+        };
     }
-    const ts = getCurrentUTC().toISOString();
+    const ts = new Date().toISOString();
     yield {
         type: "checkpoint",
         timestamp: ts,
@@ -137,7 +149,8 @@ function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, t
             values: (0, io_js_1.readChannels)(channels, streamChannels),
             metadata,
             next: tasks.map((task) => task.name),
-            tasks: tasksWithWrites(tasks, pendingWrites),
+            tasks: tasksWithWrites(tasks, pendingWrites, taskStates),
+            parentConfig: parentConfig ? formatConfig(parentConfig) : undefined,
         },
     };
 }

package/dist/pregel/debug.d.ts

@@ -26,7 +26,7 @@ export declare function mapDebugTaskResults<N extends PropertyKey, C extends Pro
         interrupts: PendingWrite<C>[];
     };
 }, void, unknown>;
-export declare function mapDebugCheckpoint<N extends PropertyKey, C extends PropertyKey>(step: number, config: RunnableConfig, channels: Record<string, BaseChannel>, streamChannels: string | string[], metadata: CheckpointMetadata, tasks: readonly PregelExecutableTask<N, C>[], pendingWrites: CheckpointPendingWrite[]): Generator<{
+export declare function mapDebugCheckpoint<N extends PropertyKey, C extends PropertyKey>(step: number, config: RunnableConfig, channels: Record<string, BaseChannel>, streamChannels: string | string[], metadata: CheckpointMetadata, tasks: readonly PregelExecutableTask<N, C>[], pendingWrites: CheckpointPendingWrite[], parentConfig: RunnableConfig | undefined): Generator<{
     type: string;
     timestamp: string;
     step: number;
@@ -36,6 +36,7 @@ export declare function mapDebugCheckpoint<N extends PropertyKey, C extends Prop
         metadata: CheckpointMetadata;
         next: N[];
         tasks: PregelTaskDescription[];
+        parentConfig: Partial<Record<"configurable" | "timeout" | "signal" | "tags" | "metadata" | "callbacks" | "recursion_limit" | "max_concurrency" | "run_name" | "run_id", unknown>> | undefined;
     };
 }, void, unknown>;
 export declare function tasksWithWrites<N extends PropertyKey, C extends PropertyKey>(tasks: PregelTaskDescription[] | readonly PregelExecutableTask<N, C>[], pendingWrites: CheckpointPendingWrite[], states?: Record<string, RunnableConfig | StateSnapshot>): PregelTaskDescription[];

package/dist/pregel/debug.js

@@ -1,6 +1,7 @@
 import { ERROR, INTERRUPT, TAG_HIDDEN } from "../constants.js";
 import { EmptyChannelError } from "../errors.js";
 import { readChannels } from "./io.js";
+import { findSubgraphPregel } from "./utils/subgraph.js";
 const COLORS_MAP = {
     blue: {
         start: "\x1b[34m",
@@ -95,7 +96,7 @@ export function* mapDebugTaskResults(step, tasks, streamChannels) {
         };
     }
 }
-export function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, tasks, pendingWrites) {
+export function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, tasks, pendingWrites, parentConfig) {
     function formatConfig(config) {
         // make sure the config is consistent with Python
         const pyConfig = {};
@@ -117,11 +118,22 @@ export function* mapDebugCheckpoint(step, config, channels, streamChannels, meta
             pyConfig.tags = config.tags;
         return pyConfig;
     }
-    function getCurrentUTC() {
-        const now = new Date();
-        return new Date(now.getTime() - now.getTimezoneOffset() * 60 * 1000);
+    const parentNs = config.configurable?.checkpoint_ns;
+    const taskStates = {};
+    for (const task of tasks) {
+        if (!findSubgraphPregel(task.proc))
+            continue;
+        let taskNs = `${task.name}:${task.id}`;
+        if (parentNs)
+            taskNs = `${parentNs}|${taskNs}`;
+        taskStates[task.id] = {
+            configurable: {
+                thread_id: config.configurable?.thread_id,
+                checkpoint_ns: taskNs,
+            },
+        };
     }
-    const ts = getCurrentUTC().toISOString();
+    const ts = new Date().toISOString();
     yield {
         type: "checkpoint",
         timestamp: ts,
@@ -131,7 +143,8 @@ export function* mapDebugCheckpoint(step, config, channels, streamChannels, meta
             values: readChannels(channels, streamChannels),
             metadata,
             next: tasks.map((task) => task.name),
-            tasks: tasksWithWrites(tasks, pendingWrites),
+            tasks: tasksWithWrites(tasks, pendingWrites, taskStates),
+            parentConfig: parentConfig ? formatConfig(parentConfig) : undefined,
         },
     };
 }

package/dist/pregel/index.cjs

@@ -18,6 +18,7 @@ const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 const algo_js_1 = require("./algo.cjs");
 const index_js_1 = require("./utils/index.cjs");
+const subgraph_js_1 = require("./utils/subgraph.cjs");
 const loop_js_1 = require("./loop.cjs");
 const retry_js_1 = require("./retry.cjs");
 const base_js_2 = require("../managed/base.cjs");
@@ -81,16 +82,6 @@ class Channel {
     }
 }
 exports.Channel = Channel;
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function isPregel(x) {
-    return ("inputChannels" in x &&
-        x.inputChannels !== undefined &&
-        "outputChannels" &&
-        x.outputChannels !== undefined);
-}
-function isRunnableSequence(x) {
-    return "steps" in x && Array.isArray(x.steps);
-}
 class Pregel extends runnables_1.Runnable {
     static lc_name() {
         return "LangGraph";
@@ -264,18 +255,7 @@ class Pregel extends runnables_1.Runnable {
                     continue;
                 }
             }
-            // find the subgraph if any
-            let graph;
-            const candidates = [node.bound];
-            for (const candidate of candidates) {
-                if (isPregel(candidate)) {
-                    graph = candidate;
-                    break;
-                }
-                else if (isRunnableSequence(candidate)) {
-                    candidates.push(...candidate.steps);
-                }
-            }
+            const graph = (0, subgraph_js_1.findSubgraphPregel)(node.bound);
             // if found, yield recursively
             if (graph !== undefined) {
                 if (name === namespace) {
@@ -609,7 +589,10 @@ class Pregel extends runnables_1.Runnable {
             defaultStreamMode = ["values"];
         }
         let defaultCheckpointer;
-        if (config !== undefined &&
+        if (this.checkpointer === false) {
+            defaultCheckpointer = undefined;
+        }
+        else if (config !== undefined &&
             config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINTER] !== undefined) {
             defaultCheckpointer = config.configurable[constants_js_1.CONFIG_KEY_CHECKPOINTER];
         }
@@ -699,6 +682,7 @@ class Pregel extends runnables_1.Runnable {
             throw new Error(`Passed "recursionLimit" must be at least 1.`);
         }
         if (this.checkpointer !== undefined &&
+            this.checkpointer !== false &&
             inputConfig.configurable === undefined) {
             throw new Error(`Checkpointer requires one or more of the following "configurable" keys: "thread_id", "checkpoint_ns", "checkpoint_id"`);
         }

package/dist/pregel/index.d.ts

@@ -56,7 +56,7 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     interruptBefore?: Array<keyof Nn> | All;
     stepTimeout?: number;
     debug: boolean;
-    checkpointer?: BaseCheckpointSaver;
+    checkpointer?: BaseCheckpointSaver | false;
     retryPolicy?: RetryPolicy;
     config?: LangGraphRunnableConfig;
     store?: BaseStore;

package/dist/pregel/index.js

@@ -12,6 +12,7 @@ import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, ERROR, INTER
 import { GraphRecursionError, GraphValueError, InvalidUpdateError, isGraphInterrupt, } from "../errors.js";
 import { _prepareNextTasks, _localRead, _applyWrites, } from "./algo.js";
 import { _coerceToDict, getNewChannelVersions, patchCheckpointMap, } from "./utils/index.js";
+import { findSubgraphPregel } from "./utils/subgraph.js";
 import { PregelLoop, StreamProtocol } from "./loop.js";
 import { executeTasksWithRetry } from "./retry.js";
 import { ChannelKeyPlaceholder, isConfiguredManagedValue, ManagedValueMapping, NoopManagedValue, } from "../managed/base.js";
@@ -74,16 +75,6 @@ export class Channel {
         return new ChannelWrite(channelWriteEntries);
     }
 }
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function isPregel(x) {
-    return ("inputChannels" in x &&
-        x.inputChannels !== undefined &&
-        "outputChannels" &&
-        x.outputChannels !== undefined);
-}
-function isRunnableSequence(x) {
-    return "steps" in x && Array.isArray(x.steps);
-}
 export class Pregel extends Runnable {
     static lc_name() {
         return "LangGraph";
@@ -257,18 +248,7 @@ export class Pregel extends Runnable {
                     continue;
                 }
             }
-            // find the subgraph if any
-            let graph;
-            const candidates = [node.bound];
-            for (const candidate of candidates) {
-                if (isPregel(candidate)) {
-                    graph = candidate;
-                    break;
-                }
-                else if (isRunnableSequence(candidate)) {
-                    candidates.push(...candidate.steps);
-                }
-            }
+            const graph = findSubgraphPregel(node.bound);
             // if found, yield recursively
             if (graph !== undefined) {
                 if (name === namespace) {
@@ -602,7 +582,10 @@ export class Pregel extends Runnable {
             defaultStreamMode = ["values"];
         }
         let defaultCheckpointer;
-        if (config !== undefined &&
+        if (this.checkpointer === false) {
+            defaultCheckpointer = undefined;
+        }
+        else if (config !== undefined &&
             config.configurable?.[CONFIG_KEY_CHECKPOINTER] !== undefined) {
             defaultCheckpointer = config.configurable[CONFIG_KEY_CHECKPOINTER];
         }
@@ -692,6 +675,7 @@ export class Pregel extends Runnable {
             throw new Error(`Passed "recursionLimit" must be at least 1.`);
         }
         if (this.checkpointer !== undefined &&
+            this.checkpointer !== false &&
             inputConfig.configurable === undefined) {
             throw new Error(`Checkpointer requires one or more of the following "configurable" keys: "thread_id", "checkpoint_ns", "checkpoint_id"`);
         }

package/dist/pregel/io.cjs

@@ -64,7 +64,7 @@ chunk
             }
         }
         else if (Array.isArray(inputChannels)) {
-            throw new Error("Input chunk must be an object when inputChannels is an array");
+            throw new Error(`Input chunk must be an object when "inputChannels" is an array`);
         }
         else {
             yield [inputChannels, chunk];

package/dist/pregel/io.js

@@ -59,7 +59,7 @@ chunk
             }
         }
         else if (Array.isArray(inputChannels)) {
-            throw new Error("Input chunk must be an object when inputChannels is an array");
+            throw new Error(`Input chunk must be an object when "inputChannels" is an array`);
         }
         else {
             yield [inputChannels, chunk];

package/dist/pregel/loop.cjs

@@ -166,6 +166,12 @@ class PregelLoop {
             writable: true,
             value: 0
         });
+        Object.defineProperty(this, "prevCheckpointConfig", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -238,9 +244,11 @@ class PregelLoop {
         this.store = params.store;
         this.stream = params.stream;
         this.checkpointNamespace = params.checkpointNamespace;
+        this.prevCheckpointConfig = params.prevCheckpointConfig;
     }
     static async initialize(params) {
         let { config, stream } = params;
+        const { checkSubgraphs = true } = params;
         if (stream !== undefined &&
             config.configurable?.[constants_js_1.CONFIG_KEY_STREAM] !== undefined) {
             stream = createDuplexStream(stream, config.configurable[constants_js_1.CONFIG_KEY_STREAM]);
@@ -283,6 +291,7 @@ class PregelLoop {
                 ...saved.config.configurable,
             },
         };
+        const prevCheckpointConfig = saved.parentConfig;
         const checkpoint = (0, langgraph_checkpoint_1.copyCheckpoint)(saved.checkpoint);
         const checkpointMetadata = { ...saved.metadata };
         const checkpointPendingWrites = saved.pendingWrites ?? [];
@@ -297,6 +306,19 @@ class PregelLoop {
             // Start the store. This is a batch store, so it will run continuously
             store.start();
         }
+        if (checkSubgraphs && isNested && params.checkpointer !== undefined) {
+            if ((0, errors_js_1.getSubgraphsSeenSet)().has(config.configurable?.checkpoint_ns)) {
+                throw new errors_js_1.MultipleSubgraphsError([
+                    "Detected the same subgraph called multiple times by the same node.",
+                    "This is not allowed if checkpointing is enabled.",
+                    "",
+                    `You can disable checkpointing for a subgraph by compiling it with ".compile({ checkpointer: false });"`,
+                ].join("\n"));
+            }
+            else {
+                (0, errors_js_1.getSubgraphsSeenSet)().add(config.configurable?.checkpoint_ns);
+            }
+        }
         return new PregelLoop({
             input: params.input,
             config,
@@ -304,6 +326,7 @@ class PregelLoop {
             checkpoint,
             checkpointMetadata,
             checkpointConfig,
+            prevCheckpointConfig,
             checkpointNamespace,
             channels,
             managed: params.managed,
@@ -453,7 +476,7 @@ class PregelLoop {
             // Produce debug output
             if (this.checkpointer) {
                 this._emit(await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugCheckpoint)(this.step - 1, // printing checkpoint for previous step
-                this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites), "debug")));
+                this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites, this.prevCheckpointConfig), "debug")));
             }
             if (Object.values(this.tasks).length === 0) {
                 this.status = "done";
@@ -586,6 +609,11 @@ class PregelLoop {
         };
         // Bail if no checkpointer
         if (this.checkpointer !== undefined) {
+            // store the previous checkpoint config for debug events
+            this.prevCheckpointConfig = this.checkpointConfig?.configurable
+                ?.checkpoint_id
+                ? this.checkpointConfig
+                : undefined;
             // create new checkpoint
             this.checkpointMetadata = metadata;
             // child graphs keep at most one checkpoint per parent checkpoint

package/dist/pregel/loop.d.ts

@@ -18,6 +18,7 @@ export type PregelLoopInitializeParams = {
     managed: ManagedValueMapping;
     stream: StreamProtocol;
     store?: BaseStore;
+    checkSubgraphs?: boolean;
 };
 type PregelLoopParams = {
     input?: any;
@@ -40,6 +41,7 @@ type PregelLoopParams = {
     isNested: boolean;
     stream: StreamProtocol;
     store?: AsyncBatchedStore;
+    prevCheckpointConfig: RunnableConfig | undefined;
 };
 export declare class StreamProtocol {
     push: (chunk: StreamChunk) => void;
@@ -67,6 +69,7 @@ export declare class PregelLoop {
     protected nodes: Record<string, PregelNode>;
     protected skipDoneTasks: boolean;
     protected taskWritesLeft: number;
+    protected prevCheckpointConfig: RunnableConfig | undefined;
     status: "pending" | "done" | "interrupt_before" | "interrupt_after" | "out_of_steps";
     tasks: Record<string, PregelExecutableTask<any, any>>;
     stream: StreamProtocol;

package/dist/pregel/loop.js

@@ -4,7 +4,7 @@ import { CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_R
 import { _applyWrites, _prepareNextTasks, increment, shouldInterrupt, } from "./algo.js";
 import { gatherIterator, gatherIteratorSync, prefixGenerator, } from "../utils.js";
 import { mapInput, mapOutputUpdates, mapOutputValues, readChannels, } from "./io.js";
-import { EmptyInputError, GraphInterrupt, isGraphInterrupt, } from "../errors.js";
+import { getSubgraphsSeenSet, EmptyInputError, GraphInterrupt, isGraphInterrupt, MultipleSubgraphsError, } from "../errors.js";
 import { getNewChannelVersions, patchConfigurable } from "./utils/index.js";
 import { mapDebugTasks, mapDebugCheckpoint, mapDebugTaskResults, } from "./debug.js";
 const INPUT_DONE = Symbol.for("INPUT_DONE");
@@ -162,6 +162,12 @@ export class PregelLoop {
             writable: true,
             value: 0
         });
+        Object.defineProperty(this, "prevCheckpointConfig", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "status", {
             enumerable: true,
             configurable: true,
@@ -234,9 +240,11 @@ export class PregelLoop {
         this.store = params.store;
         this.stream = params.stream;
         this.checkpointNamespace = params.checkpointNamespace;
+        this.prevCheckpointConfig = params.prevCheckpointConfig;
     }
     static async initialize(params) {
         let { config, stream } = params;
+        const { checkSubgraphs = true } = params;
         if (stream !== undefined &&
             config.configurable?.[CONFIG_KEY_STREAM] !== undefined) {
             stream = createDuplexStream(stream, config.configurable[CONFIG_KEY_STREAM]);
@@ -279,6 +287,7 @@ export class PregelLoop {
                 ...saved.config.configurable,
             },
         };
+        const prevCheckpointConfig = saved.parentConfig;
         const checkpoint = copyCheckpoint(saved.checkpoint);
         const checkpointMetadata = { ...saved.metadata };
         const checkpointPendingWrites = saved.pendingWrites ?? [];
@@ -293,6 +302,19 @@ export class PregelLoop {
             // Start the store. This is a batch store, so it will run continuously
             store.start();
         }
+        if (checkSubgraphs && isNested && params.checkpointer !== undefined) {
+            if (getSubgraphsSeenSet().has(config.configurable?.checkpoint_ns)) {
+                throw new MultipleSubgraphsError([
+                    "Detected the same subgraph called multiple times by the same node.",
+                    "This is not allowed if checkpointing is enabled.",
+                    "",
+                    `You can disable checkpointing for a subgraph by compiling it with ".compile({ checkpointer: false });"`,
+                ].join("\n"));
+            }
+            else {
+                getSubgraphsSeenSet().add(config.configurable?.checkpoint_ns);
+            }
+        }
         return new PregelLoop({
             input: params.input,
             config,
@@ -300,6 +322,7 @@ export class PregelLoop {
             checkpoint,
             checkpointMetadata,
             checkpointConfig,
+            prevCheckpointConfig,
             checkpointNamespace,
             channels,
             managed: params.managed,
@@ -449,7 +472,7 @@ export class PregelLoop {
             // Produce debug output
             if (this.checkpointer) {
                 this._emit(await gatherIterator(prefixGenerator(mapDebugCheckpoint(this.step - 1, // printing checkpoint for previous step
-                this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites), "debug")));
+                this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites, this.prevCheckpointConfig), "debug")));
             }
             if (Object.values(this.tasks).length === 0) {
                 this.status = "done";
@@ -582,6 +605,11 @@ export class PregelLoop {
         };
         // Bail if no checkpointer
         if (this.checkpointer !== undefined) {
+            // store the previous checkpoint config for debug events
+            this.prevCheckpointConfig = this.checkpointConfig?.configurable
+                ?.checkpoint_id
+                ? this.checkpointConfig
+                : undefined;
             // create new checkpoint
             this.checkpointMetadata = metadata;
             // child graphs keep at most one checkpoint per parent checkpoint

package/dist/pregel/retry.cjs

@@ -132,6 +132,13 @@ pregelTask, retryPolicy) {
                 error.constructor.name;
             console.log(`Retrying task "${pregelTask.name}" after ${interval.toFixed(2)} seconds (attempt ${attempts}) after ${errorName}: ${error}`);
         }
+        finally {
+            // Clear checkpoint_ns seen (for subgraph detection)
+            const checkpointNs = pregelTask.config?.configurable?.checkpoint_ns;
+            if (checkpointNs) {
+                (0, errors_js_1.getSubgraphsSeenSet)().delete(checkpointNs);
+            }
+        }
     }
     return {
         task: pregelTask,

package/dist/pregel/retry.js

@@ -1,4 +1,4 @@
-import { isGraphInterrupt } from "../errors.js";
+import { getSubgraphsSeenSet, isGraphInterrupt } from "../errors.js";
 export const DEFAULT_INITIAL_INTERVAL = 500;
 export const DEFAULT_BACKOFF_FACTOR = 2;
 export const DEFAULT_MAX_INTERVAL = 128000;
@@ -128,6 +128,13 @@ pregelTask, retryPolicy) {
                 error.constructor.name;
             console.log(`Retrying task "${pregelTask.name}" after ${interval.toFixed(2)} seconds (attempt ${attempts}) after ${errorName}: ${error}`);
         }
+        finally {
+            // Clear checkpoint_ns seen (for subgraph detection)
+            const checkpointNs = pregelTask.config?.configurable?.checkpoint_ns;
+            if (checkpointNs) {
+                getSubgraphsSeenSet().delete(checkpointNs);
+            }
+        }
     }
     return {
         task: pregelTask,

package/dist/pregel/runnable_types.d.ts

@@ -1,5 +1,5 @@
 import { RunnableConfig } from "@langchain/core/runnables";
 import { BaseStore } from "@langchain/langgraph-checkpoint";
-export interface LangGraphRunnableConfig extends RunnableConfig {
+export interface LangGraphRunnableConfig<ConfigurableType extends Record<string, any> = Record<string, any>> extends RunnableConfig<ConfigurableType> {
     store?: BaseStore;
 }

package/dist/pregel/types.d.ts

@@ -44,7 +44,7 @@ export interface PregelInterface<Nn extends StrRecord<string, PregelNode>, Cc ex
      * @default false
      */
     debug?: boolean;
-    checkpointer?: BaseCheckpointSaver;
+    checkpointer?: BaseCheckpointSaver | false;
     retryPolicy?: RetryPolicy;
     config?: LangGraphRunnableConfig;
     /**

package/dist/pregel/utils/subgraph.cjs

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findSubgraphPregel = void 0;
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function isRunnableSequence(x) {
+    return "steps" in x && Array.isArray(x.steps);
+}
+function isPregelLike(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+x
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    return ("inputChannels" in x &&
+        x.inputChannels !== undefined &&
+        "outputChannels" &&
+        x.outputChannels !== undefined);
+}
+function findSubgraphPregel(candidate
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    const candidates = [candidate];
+    for (const candidate of candidates) {
+        if (isPregelLike(candidate)) {
+            return candidate;
+        }
+        else if (isRunnableSequence(candidate)) {
+            candidates.push(...candidate.steps);
+        }
+    }
+    return undefined;
+}
+exports.findSubgraphPregel = findSubgraphPregel;

package/dist/pregel/utils/subgraph.d.ts

@@ -0,0 +1,3 @@
+import { Runnable } from "@langchain/core/runnables";
+import type { PregelInterface } from "../types.js";
+export declare function findSubgraphPregel(candidate: Runnable): PregelInterface<any, any> | undefined;

package/dist/pregel/utils/subgraph.js

@@ -0,0 +1,28 @@
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function isRunnableSequence(x) {
+    return "steps" in x && Array.isArray(x.steps);
+}
+function isPregelLike(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+x
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    return ("inputChannels" in x &&
+        x.inputChannels !== undefined &&
+        "outputChannels" &&
+        x.outputChannels !== undefined);
+}
+export function findSubgraphPregel(candidate
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    const candidates = [candidate];
+    for (const candidate of candidates) {
+        if (isPregelLike(candidate)) {
+            return candidate;
+        }
+        else if (isRunnableSequence(candidate)) {
+            candidates.push(...candidate.steps);
+        }
+    }
+    return undefined;
+}

package/dist/web.cjs

@@ -14,12 +14,13 @@ 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.MessagesAnnotation = exports.InMemoryStore = exports.AsyncBatchedStore = exports.BaseStore = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
+exports.MessagesAnnotation = exports.InMemoryStore = exports.AsyncBatchedStore = exports.BaseStore = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.CompiledStateGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
 var index_js_1 = require("./graph/index.cjs");
 Object.defineProperty(exports, "END", { enumerable: true, get: function () { return index_js_1.END; } });
 Object.defineProperty(exports, "Graph", { enumerable: true, get: function () { return index_js_1.Graph; } });
 Object.defineProperty(exports, "START", { enumerable: true, get: function () { return index_js_1.START; } });
 Object.defineProperty(exports, "StateGraph", { enumerable: true, get: function () { return index_js_1.StateGraph; } });
+Object.defineProperty(exports, "CompiledStateGraph", { enumerable: true, get: function () { return index_js_1.CompiledStateGraph; } });
 Object.defineProperty(exports, "MessageGraph", { enumerable: true, get: function () { return index_js_1.MessageGraph; } });
 Object.defineProperty(exports, "messagesStateReducer", { enumerable: true, get: function () { return index_js_1.messagesStateReducer; } });
 Object.defineProperty(exports, "Annotation", { enumerable: true, get: function () { return index_js_1.Annotation; } });

package/dist/web.d.ts

@@ -1,4 +1,4 @@
-export { END, Graph, type StateGraphArgs, START, StateGraph, type CompiledStateGraph, MessageGraph, messagesStateReducer, type Messages, Annotation, type StateType, type UpdateType, type NodeType, type StateDefinition, type SingleReducer, type CompiledGraph, } from "./graph/index.js";
+export { END, Graph, type StateGraphArgs, START, StateGraph, CompiledStateGraph, MessageGraph, messagesStateReducer, type Messages, Annotation, type StateType, type UpdateType, type NodeType, type StateDefinition, type SingleReducer, type CompiledGraph, } from "./graph/index.js";
 export type { StateSnapshot } from "./pregel/types.js";
 export * from "./errors.js";
 export { BaseChannel, type BinaryOperator, BinaryOperatorAggregate, type AnyValue, type WaitForNames, type DynamicBarrierValue, type LastValue, type NamedBarrierValue, type Topic, } from "./channels/index.js";

package/dist/web.js

@@ -1,4 +1,4 @@
-export { END, Graph, START, StateGraph, MessageGraph, messagesStateReducer, Annotation, } from "./graph/index.js";
+export { END, Graph, START, StateGraph, CompiledStateGraph, MessageGraph, messagesStateReducer, Annotation, } from "./graph/index.js";
 export * from "./errors.js";
 export { BaseChannel, BinaryOperatorAggregate, } from "./channels/index.js";
 export { Send } from "./constants.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.13",
+  "version": "0.2.15",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -31,19 +31,19 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/langgraph-checkpoint": "~0.0.9",
+    "@langchain/langgraph-checkpoint": "~0.0.10",
     "double-ended-queue": "^2.1.0-0",
     "uuid": "^10.0.0",
     "zod": "^3.23.8"
   },
   "peerDependencies": {
-    "@langchain/core": ">=0.2.35 <0.3.0 || >=0.3.8 < 0.4.0"
+    "@langchain/core": ">=0.2.36 <0.3.0 || >=0.3.9 < 0.4.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
     "@langchain/anthropic": "^0.3.0",
     "@langchain/community": "^0.3.0",
-    "@langchain/core": "^0.3.6",
+    "@langchain/core": "^0.3.10",
     "@langchain/langgraph-checkpoint-postgres": "workspace:*",
     "@langchain/langgraph-checkpoint-sqlite": "workspace:*",
     "@langchain/openai": "^0.3.0",