@langchain/langgraph 0.2.16 → 0.2.17

package/dist/graph/graph.cjs

@@ -13,6 +13,7 @@ const write_js_1 = require("../pregel/write.cjs");
 const constants_js_1 = require("../constants.cjs");
 const utils_js_1 = require("../utils.cjs");
 const errors_js_1 = require("../errors.cjs");
+const subgraph_js_1 = require("../pregel/utils/subgraph.cjs");
 /** Special reserved node name denoting the start of a graph. */
 exports.START = "__start__";
 /** Special reserved node name denoting the end of a graph. */
@@ -140,11 +141,13 @@ class Graph {
         if (key === exports.END) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
+        const runnable = (0, runnables_1._coerceToRunnable)(
+        // Account for arbitrary state due to Send API
+        action);
         this.nodes[key] = {
-            runnable: (0, runnables_1._coerceToRunnable)(
-            // Account for arbitrary state due to Send API
-            action),
+            runnable,
             metadata: options?.metadata,
+            subgraphs: (0, subgraph_js_1.isPregelLike)(runnable) ? [runnable] : options?.subgraphs,
         };
         return this;
     }
@@ -300,6 +303,7 @@ class CompiledGraph extends index_js_1.Pregel {
             channels: [],
             triggers: [],
             metadata: node.metadata,
+            subgraphs: node.subgraphs,
         })
             .pipe(node.runnable)
             .pipe(new write_js_1.ChannelWrite([{ channel: key, value: write_js_1.PASSTHROUGH }], [constants_js_1.TAG_HIDDEN]));

package/dist/graph/graph.d.ts

@@ -28,9 +28,11 @@ export declare class Branch<IO, N extends string, CallOptions extends LangGraphR
 export type NodeSpec<RunInput, RunOutput> = {
     runnable: Runnable<RunInput, RunOutput>;
     metadata?: Record<string, unknown>;
+    subgraphs?: Pregel<any, any>[];
 };
 export type AddNodeOptions = {
     metadata?: Record<string, unknown>;
+    subgraphs?: Pregel<any, any>[];
 };
 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>;

package/dist/graph/graph.js

@@ -10,6 +10,7 @@ import { ChannelWrite, PASSTHROUGH } from "../pregel/write.js";
 import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
 import { gatherIteratorSync, RunnableCallable } from "../utils.js";
 import { InvalidUpdateError, NodeInterrupt } from "../errors.js";
+import { isPregelLike } from "../pregel/utils/subgraph.js";
 /** Special reserved node name denoting the start of a graph. */
 export const START = "__start__";
 /** Special reserved node name denoting the end of a graph. */
@@ -136,11 +137,13 @@ export class Graph {
         if (key === END) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
+        const runnable = _coerceToRunnable(
+        // Account for arbitrary state due to Send API
+        action);
         this.nodes[key] = {
-            runnable: _coerceToRunnable(
-            // Account for arbitrary state due to Send API
-            action),
+            runnable,
             metadata: options?.metadata,
+            subgraphs: isPregelLike(runnable) ? [runnable] : options?.subgraphs,
         };
         return this;
     }
@@ -295,6 +298,7 @@ export class CompiledGraph extends Pregel {
             channels: [],
             triggers: [],
             metadata: node.metadata,
+            subgraphs: node.subgraphs,
         })
             .pipe(node.runnable)
             .pipe(new ChannelWrite([{ channel: key, value: PASSTHROUGH }], [TAG_HIDDEN]));

package/dist/graph/state.cjs

@@ -14,6 +14,7 @@ const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 const annotation_js_1 = require("./annotation.cjs");
 const base_js_2 = require("../managed/base.cjs");
+const subgraph_js_1 = require("../pregel/utils/subgraph.cjs");
 const ROOT = "__root__";
 /**
  * A graph whose nodes communicate by reading and writing to a shared state.
@@ -216,11 +217,16 @@ class StateGraph extends graph_js_1.Graph {
         if (options?.input !== undefined) {
             this._addSchema(options.input.spec);
         }
+        const runnable = (0, runnables_1._coerceToRunnable)(action);
         const nodeSpec = {
-            runnable: (0, runnables_1._coerceToRunnable)(action),
+            runnable,
             retryPolicy: options?.retryPolicy,
             metadata: options?.metadata,
             input: options?.input?.spec ?? this._schemaDefinition,
+            subgraphs: (0, subgraph_js_1.isPregelLike)(runnable)
+                ? // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    [runnable]
+                : options?.subgraphs,
         };
         this.nodes[key] = nodeSpec;
         return this;
@@ -377,6 +383,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
                 bound: node?.runnable,
                 metadata: node?.metadata,
                 retryPolicy: node?.retryPolicy,
+                subgraphs: node?.subgraphs,
             });
         }
     }

package/dist/graph/state.js

@@ -11,6 +11,7 @@ import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_
 import { InvalidUpdateError } from "../errors.js";
 import { getChannel, } from "./annotation.js";
 import { isConfiguredManagedValue } from "../managed/base.js";
+import { isPregelLike } from "../pregel/utils/subgraph.js";
 const ROOT = "__root__";
 /**
  * A graph whose nodes communicate by reading and writing to a shared state.
@@ -213,11 +214,16 @@ export class StateGraph extends Graph {
         if (options?.input !== undefined) {
             this._addSchema(options.input.spec);
         }
+        const runnable = _coerceToRunnable(action);
         const nodeSpec = {
-            runnable: _coerceToRunnable(action),
+            runnable,
             retryPolicy: options?.retryPolicy,
             metadata: options?.metadata,
             input: options?.input?.spec ?? this._schemaDefinition,
+            subgraphs: isPregelLike(runnable)
+                ? // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    [runnable]
+                : options?.subgraphs,
         };
         this.nodes[key] = nodeSpec;
         return this;
@@ -373,6 +379,7 @@ export class CompiledStateGraph extends CompiledGraph {
                 bound: node?.runnable,
                 metadata: node?.metadata,
                 retryPolicy: node?.retryPolicy,
+                subgraphs: node?.subgraphs,
             });
         }
     }

package/dist/prebuilt/index.d.ts

@@ -2,4 +2,4 @@ export { type AgentExecutorState, createAgentExecutor, } from "./agent_executor.
 export { type FunctionCallingExecutorState, createFunctionCallingExecutor, } from "./chat_agent_executor.js";
 export { type AgentState, createReactAgent } from "./react_agent_executor.js";
 export { type ToolExecutorArgs, type ToolInvocationInterface, ToolExecutor, } from "./tool_executor.js";
-export { ToolNode, toolsCondition } from "./tool_node.js";
+export { ToolNode, toolsCondition, type ToolNodeOptions } from "./tool_node.js";

package/dist/prebuilt/tool_node.d.ts

@@ -1,5 +1,5 @@
 import { BaseMessage } from "@langchain/core/messages";
-import { RunnableToolLike } from "@langchain/core/runnables";
+import { RunnableConfig, RunnableToolLike } from "@langchain/core/runnables";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { RunnableCallable } from "../utils.js";
 import { END } from "../graph/graph.js";
@@ -130,6 +130,6 @@ export declare class ToolNode<T = any> extends RunnableCallable<T, T> {
     tools: (StructuredToolInterface | RunnableToolLike)[];
     handleToolErrors: boolean;
     constructor(tools: (StructuredToolInterface | RunnableToolLike)[], options?: ToolNodeOptions);
-    private run;
+    protected run(input: any, config: RunnableConfig): Promise<T>;
 }
 export declare function toolsCondition(state: BaseMessage[] | typeof MessagesAnnotation.State): "tools" | typeof END;

package/dist/pregel/algo.cjs

@@ -275,6 +275,7 @@ function _prepareSingleTask(taskPath, checkpoint, processes, channels, managed,
                     name: packet.node,
                     input: packet.args,
                     proc: node,
+                    subgraphs: proc.subgraphs,
                     writes,
                     config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
                         metadata,
@@ -371,6 +372,7 @@ function _prepareSingleTask(taskPath, checkpoint, processes, channels, managed,
                         name,
                         input: val,
                         proc: node,
+                        subgraphs: proc.subgraphs,
                         writes,
                         config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
                             metadata,

package/dist/pregel/algo.js

@@ -266,6 +266,7 @@ export function _prepareSingleTask(taskPath, checkpoint, processes, channels, ma
                     name: packet.node,
                     input: packet.args,
                     proc: node,
+                    subgraphs: proc.subgraphs,
                     writes,
                     config: patchConfig(mergeConfigs(config, {
                         metadata,
@@ -362,6 +363,7 @@ export function _prepareSingleTask(taskPath, checkpoint, processes, channels, ma
                         name,
                         input: val,
                         proc: node,
+                        subgraphs: proc.subgraphs,
                         writes,
                         config: patchConfig(mergeConfigs(config, {
                             metadata,

package/dist/pregel/debug.cjs

@@ -127,7 +127,8 @@ function* mapDebugCheckpoint(step, config, channels, streamChannels, metadata, t
     const parentNs = config.configurable?.checkpoint_ns;
     const taskStates = {};
     for (const task of tasks) {
-        if (!(0, subgraph_js_1.findSubgraphPregel)(task.proc))
+        const candidates = task.subgraphs?.length ? task.subgraphs : [task.proc];
+        if (!candidates.find(subgraph_js_1.findSubgraphPregel))
             continue;
         let taskNs = `${task.name}:${task.id}`;
         if (parentNs)

package/dist/pregel/debug.js

@@ -121,7 +121,8 @@ export function* mapDebugCheckpoint(step, config, channels, streamChannels, meta
     const parentNs = config.configurable?.checkpoint_ns;
     const taskStates = {};
     for (const task of tasks) {
-        if (!findSubgraphPregel(task.proc))
+        const candidates = task.subgraphs?.length ? task.subgraphs : [task.proc];
+        if (!candidates.find(findSubgraphPregel))
             continue;
         let taskNs = `${task.name}:${task.id}`;
         if (parentNs)

package/dist/pregel/index.cjs

@@ -255,26 +255,28 @@ class Pregel extends runnables_1.Runnable {
                     continue;
                 }
             }
-            const graph = (0, subgraph_js_1.findSubgraphPregel)(node.bound);
-            // if found, yield recursively
-            if (graph !== undefined) {
-                if (name === namespace) {
-                    yield [name, graph];
-                    return;
-                }
-                if (namespace === undefined) {
-                    yield [name, graph];
-                }
-                if (recurse) {
-                    let newNamespace = namespace;
-                    if (namespace !== undefined) {
-                        newNamespace = namespace.slice(name.length + 1);
+            const candidates = node.subgraphs?.length ? node.subgraphs : [node.bound];
+            for (const candidate of candidates) {
+                const graph = (0, subgraph_js_1.findSubgraphPregel)(candidate);
+                if (graph !== undefined) {
+                    if (name === namespace) {
+                        yield [name, graph];
+                        return;
                     }
-                    for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) {
-                        yield [
-                            `${name}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${subgraphName}`,
-                            subgraph,
-                        ];
+                    if (namespace === undefined) {
+                        yield [name, graph];
+                    }
+                    if (recurse) {
+                        let newNamespace = namespace;
+                        if (namespace !== undefined) {
+                            newNamespace = namespace.slice(name.length + 1);
+                        }
+                        for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) {
+                            yield [
+                                `${name}${constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR}${subgraphName}`,
+                                subgraph,
+                            ];
+                        }
                     }
                 }
             }

package/dist/pregel/index.js

@@ -248,26 +248,28 @@ export class Pregel extends Runnable {
                     continue;
                 }
             }
-            const graph = findSubgraphPregel(node.bound);
-            // if found, yield recursively
-            if (graph !== undefined) {
-                if (name === namespace) {
-                    yield [name, graph];
-                    return;
-                }
-                if (namespace === undefined) {
-                    yield [name, graph];
-                }
-                if (recurse) {
-                    let newNamespace = namespace;
-                    if (namespace !== undefined) {
-                        newNamespace = namespace.slice(name.length + 1);
+            const candidates = node.subgraphs?.length ? node.subgraphs : [node.bound];
+            for (const candidate of candidates) {
+                const graph = findSubgraphPregel(candidate);
+                if (graph !== undefined) {
+                    if (name === namespace) {
+                        yield [name, graph];
+                        return;
                     }
-                    for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) {
-                        yield [
-                            `${name}${CHECKPOINT_NAMESPACE_SEPARATOR}${subgraphName}`,
-                            subgraph,
-                        ];
+                    if (namespace === undefined) {
+                        yield [name, graph];
+                    }
+                    if (recurse) {
+                        let newNamespace = namespace;
+                        if (namespace !== undefined) {
+                            newNamespace = namespace.slice(name.length + 1);
+                        }
+                        for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) {
+                            yield [
+                                `${name}${CHECKPOINT_NAMESPACE_SEPARATOR}${subgraphName}`,
+                                subgraph,
+                            ];
+                        }
                     }
                 }
             }

package/dist/pregel/read.cjs

@@ -63,7 +63,7 @@ const defaultRunnableBound =
 /* #__PURE__ */ new runnables_1.RunnablePassthrough();
 class PregelNode extends runnables_1.RunnableBinding {
     constructor(fields) {
-        const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, tags, } = fields;
+        const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, tags, subgraphs, } = fields;
         const mergedTags = [
             ...(fields.config?.tags ? fields.config.tags : []),
             ...(tags ?? []),
@@ -139,6 +139,12 @@ class PregelNode extends runnables_1.RunnableBinding {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "subgraphs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.channels = channels;
         this.triggers = triggers;
         this.mapper = mapper;
@@ -148,6 +154,7 @@ class PregelNode extends runnables_1.RunnableBinding {
         this.metadata = metadata ?? this.metadata;
         this.tags = mergedTags;
         this.retryPolicy = retryPolicy;
+        this.subgraphs = subgraphs;
     }
     getWriters() {
         const newWriters = [...this.writers];

package/dist/pregel/read.d.ts

@@ -20,6 +20,7 @@ interface PregelNodeArgs<RunInput, RunOutput> extends Partial<RunnableBindingArg
     config?: RunnableConfig;
     metadata?: Record<string, unknown>;
     retryPolicy?: RetryPolicy;
+    subgraphs?: Runnable[];
 }
 export type PregelNodeInputType = any;
 export type PregelNodeOutputType = any;
@@ -34,6 +35,7 @@ export declare class PregelNode<RunInput = PregelNodeInputType, RunOutput = Preg
     metadata: Record<string, unknown>;
     tags: string[];
     retryPolicy?: RetryPolicy;
+    subgraphs?: Runnable[];
     constructor(fields: PregelNodeArgs<RunInput, RunOutput>);
     getWriters(): Array<Runnable>;
     getNode(): Runnable<RunInput, RunOutput> | undefined;

package/dist/pregel/read.js

@@ -59,7 +59,7 @@ const defaultRunnableBound =
 /* #__PURE__ */ new RunnablePassthrough();
 export class PregelNode extends RunnableBinding {
     constructor(fields) {
-        const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, tags, } = fields;
+        const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, tags, subgraphs, } = fields;
         const mergedTags = [
             ...(fields.config?.tags ? fields.config.tags : []),
             ...(tags ?? []),
@@ -135,6 +135,12 @@ export class PregelNode extends RunnableBinding {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "subgraphs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.channels = channels;
         this.triggers = triggers;
         this.mapper = mapper;
@@ -144,6 +150,7 @@ export class PregelNode extends RunnableBinding {
         this.metadata = metadata ?? this.metadata;
         this.tags = mergedTags;
         this.retryPolicy = retryPolicy;
+        this.subgraphs = subgraphs;
     }
     getWriters() {
         const newWriters = [...this.writers];

package/dist/pregel/types.d.ts

@@ -71,6 +71,7 @@ export interface PregelExecutableTask<N extends PropertyKey, C extends PropertyK
     readonly retry_policy?: RetryPolicy;
     readonly id: string;
     readonly path?: [string, ...(string | number)[]];
+    readonly subgraphs?: Runnable[];
 }
 export interface StateSnapshot {
     /**

package/dist/pregel/utils/subgraph.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.findSubgraphPregel = void 0;
+exports.findSubgraphPregel = exports.isPregelLike = void 0;
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function isRunnableSequence(x) {
     return "steps" in x && Array.isArray(x.steps);
@@ -15,6 +15,7 @@ x
         "outputChannels" &&
         x.outputChannels !== undefined);
 }
+exports.isPregelLike = isPregelLike;
 function findSubgraphPregel(candidate
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 ) {

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

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

package/dist/pregel/utils/subgraph.js

@@ -2,7 +2,7 @@
 function isRunnableSequence(x) {
     return "steps" in x && Array.isArray(x.steps);
 }
-function isPregelLike(
+export function isPregelLike(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 x
 // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.16",
+  "version": "0.2.17",
   "description": "LangGraph",
   "type": "module",
   "engines": {