@langchain/langgraph 0.2.5 → 0.2.6

package/dist/graph/graph.cjs

@@ -40,7 +40,20 @@ class Branch {
     }
     compile(writer, reader) {
         return write_js_1.ChannelWrite.registerWriter(new utils_js_1.RunnableCallable({
-            func: (input, config) => this._route(input, config, writer, reader),
+            func: async (input, config) => {
+                try {
+                    return await this._route(input, config, writer, reader);
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                }
+                catch (e) {
+                    // Detect & warn if NodeInterrupt is thrown in a conditional edge
+                    if (e.name === errors_js_1.NodeInterrupt.unminifiable_name) {
+                        console.warn("[WARN]: 'NodeInterrupt' thrown in conditional edge. This is likely a bug in your graph implementation.\n" +
+                            "NodeInterrupt should only be thrown inside a node, not in edge conditions.");
+                    }
+                    throw e;
+                }
+            },
         }));
     }
     async _route(input, config, writer, reader) {

package/dist/graph/graph.js

@@ -8,7 +8,7 @@ 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 { RunnableCallable } from "../utils.js";
-import { InvalidUpdateError } from "../errors.js";
+import { InvalidUpdateError, NodeInterrupt } from "../errors.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. */
@@ -37,7 +37,20 @@ export class Branch {
     }
     compile(writer, reader) {
         return ChannelWrite.registerWriter(new RunnableCallable({
-            func: (input, config) => this._route(input, config, writer, reader),
+            func: async (input, config) => {
+                try {
+                    return await this._route(input, config, writer, reader);
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                }
+                catch (e) {
+                    // Detect & warn if NodeInterrupt is thrown in a conditional edge
+                    if (e.name === NodeInterrupt.unminifiable_name) {
+                        console.warn("[WARN]: 'NodeInterrupt' thrown in conditional edge. This is likely a bug in your graph implementation.\n" +
+                            "NodeInterrupt should only be thrown inside a node, not in edge conditions.");
+                    }
+                    throw e;
+                }
+            },
         }));
     }
     async _route(input, config, writer, reader) {

package/dist/managed/base.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isConfiguredManagedValue = exports.isManagedValue = exports.ManagedValueMapping = exports.ChannelKeyPlaceholder = exports.WritableManagedValue = exports.ManagedValue = void 0;
+exports.NoopManagedValue = exports.isConfiguredManagedValue = exports.isManagedValue = exports.ManagedValueMapping = exports.ChannelKeyPlaceholder = exports.WritableManagedValue = exports.ManagedValue = void 0;
 const constants_js_1 = require("../constants.cjs");
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 class ManagedValue {
@@ -161,3 +161,16 @@ function isConfiguredManagedValue(value) {
     return false;
 }
 exports.isConfiguredManagedValue = isConfiguredManagedValue;
+/**
+ * No-op class used when getting state values, as managed values should never be returned
+ * in get state calls.
+ */
+class NoopManagedValue extends ManagedValue {
+    call() { }
+    static async initialize(config, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _args) {
+        return Promise.resolve(new NoopManagedValue(config));
+    }
+}
+exports.NoopManagedValue = NoopManagedValue;

package/dist/managed/base.d.ts

@@ -28,3 +28,11 @@ export declare class ManagedValueMapping extends Map<string, ManagedValue<any>>
 }
 export declare function isManagedValue(value: unknown): value is typeof ManagedValue;
 export declare function isConfiguredManagedValue(value: unknown): value is ConfiguredManagedValue;
+/**
+ * No-op class used when getting state values, as managed values should never be returned
+ * in get state calls.
+ */
+export declare class NoopManagedValue extends ManagedValue {
+    call(): void;
+    static initialize(config: RunnableConfig, _args?: any): Promise<ManagedValue>;
+}

package/dist/managed/base.js

@@ -153,3 +153,15 @@ export function isConfiguredManagedValue(value) {
     }
     return false;
 }
+/**
+ * No-op class used when getting state values, as managed values should never be returned
+ * in get state calls.
+ */
+export class NoopManagedValue extends ManagedValue {
+    call() { }
+    static async initialize(config, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _args) {
+        return Promise.resolve(new NoopManagedValue(config));
+    }
+}

package/dist/pregel/index.cjs

@@ -243,7 +243,8 @@ class Pregel extends runnables_1.Runnable {
         const saved = await this.checkpointer.getTuple(config);
         const checkpoint = saved ? saved.checkpoint : (0, langgraph_checkpoint_1.emptyCheckpoint)();
         const channels = (0, base_js_1.emptyChannels)(this.channels, checkpoint);
-        const { managed } = await this.prepareSpecs(config);
+        // Pass `skipManaged: true` as managed values should not be returned in get state calls.
+        const { managed } = await this.prepareSpecs(config, { skipManaged: true });
         const nextTasks = (0, algo_js_1._prepareNextTasks)(checkpoint, this.nodes, channels, managed, saved !== undefined ? saved.config : config, false, { step: saved ? (saved.metadata?.step ?? -1) + 1 : -1 });
         return {
             values: (0, io_js_1.readChannels)(channels, this.streamChannelsAsIs),
@@ -262,7 +263,8 @@ class Pregel extends runnables_1.Runnable {
         if (!this.checkpointer) {
             throw new errors_js_1.GraphValueError("No checkpointer set");
         }
-        const { managed } = await this.prepareSpecs(config);
+        // Pass `skipManaged: true` as managed values should not be returned in get state calls.
+        const { managed } = await this.prepareSpecs(config, { skipManaged: true });
         for await (const saved of this.checkpointer.list(config, options)) {
             const channels = (0, base_js_1.emptyChannels)(this.channels, saved.checkpoint);
             const nextTasks = (0, algo_js_1._prepareNextTasks)(saved.checkpoint, this.nodes, channels, managed, saved.config, false, { step: -1 });
@@ -467,7 +469,7 @@ class Pregel extends runnables_1.Runnable {
     async stream(input, options) {
         return super.stream(input, options);
     }
-    async prepareSpecs(config) {
+    async prepareSpecs(config, options) {
         const configForManaged = (0, utils_js_2.patchConfigurable)(config, {
             [constants_js_1.CONFIG_KEY_STORE]: this.store,
         });
@@ -477,6 +479,12 @@ class Pregel extends runnables_1.Runnable {
             if ((0, base_js_1.isBaseChannel)(spec)) {
                 channelSpecs[name] = spec;
             }
+            else if (options?.skipManaged) {
+                managedSpecs[name] = {
+                    cls: base_js_2.NoopManagedValue,
+                    params: { config: {} },
+                };
+            }
             else {
                 managedSpecs[name] = spec;
             }

package/dist/pregel/index.d.ts

@@ -105,7 +105,9 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
      * @param options.debug Whether to print debug information during execution.
      */
     stream(input: PregelInputType, options?: Partial<PregelOptions<Nn, Cc>>): Promise<IterableReadableStream<PregelOutputType>>;
-    protected prepareSpecs(config: RunnableConfig): Promise<{
+    protected prepareSpecs(config: RunnableConfig, options?: {
+        skipManaged?: boolean;
+    }): Promise<{
         channelSpecs: Record<string, BaseChannel<unknown, unknown, unknown>>;
         managed: ManagedValueMapping;
     }>;

package/dist/pregel/index.js

@@ -13,7 +13,7 @@ import { _prepareNextTasks, _localRead, _applyWrites, } from "./algo.js";
 import { _coerceToDict, getNewChannelVersions } from "./utils.js";
 import { PregelLoop } from "./loop.js";
 import { executeTasksWithRetry } from "./retry.js";
-import { ChannelKeyPlaceholder, isConfiguredManagedValue, ManagedValueMapping, } from "../managed/base.js";
+import { ChannelKeyPlaceholder, isConfiguredManagedValue, ManagedValueMapping, NoopManagedValue, } from "../managed/base.js";
 import { patchConfigurable } from "../utils.js";
 function isString(value) {
     return typeof value === "string";
@@ -239,7 +239,8 @@ export class Pregel extends Runnable {
         const saved = await this.checkpointer.getTuple(config);
         const checkpoint = saved ? saved.checkpoint : emptyCheckpoint();
         const channels = emptyChannels(this.channels, checkpoint);
-        const { managed } = await this.prepareSpecs(config);
+        // Pass `skipManaged: true` as managed values should not be returned in get state calls.
+        const { managed } = await this.prepareSpecs(config, { skipManaged: true });
         const nextTasks = _prepareNextTasks(checkpoint, this.nodes, channels, managed, saved !== undefined ? saved.config : config, false, { step: saved ? (saved.metadata?.step ?? -1) + 1 : -1 });
         return {
             values: readChannels(channels, this.streamChannelsAsIs),
@@ -258,7 +259,8 @@ export class Pregel extends Runnable {
         if (!this.checkpointer) {
             throw new GraphValueError("No checkpointer set");
         }
-        const { managed } = await this.prepareSpecs(config);
+        // Pass `skipManaged: true` as managed values should not be returned in get state calls.
+        const { managed } = await this.prepareSpecs(config, { skipManaged: true });
         for await (const saved of this.checkpointer.list(config, options)) {
             const channels = emptyChannels(this.channels, saved.checkpoint);
             const nextTasks = _prepareNextTasks(saved.checkpoint, this.nodes, channels, managed, saved.config, false, { step: -1 });
@@ -463,7 +465,7 @@ export class Pregel extends Runnable {
     async stream(input, options) {
         return super.stream(input, options);
     }
-    async prepareSpecs(config) {
+    async prepareSpecs(config, options) {
         const configForManaged = patchConfigurable(config, {
             [CONFIG_KEY_STORE]: this.store,
         });
@@ -473,6 +475,12 @@ export class Pregel extends Runnable {
             if (isBaseChannel(spec)) {
                 channelSpecs[name] = spec;
             }
+            else if (options?.skipManaged) {
+                managedSpecs[name] = {
+                    cls: NoopManagedValue,
+                    params: { config: {} },
+                };
+            }
             else {
                 managedSpecs[name] = spec;
             }

package/package.json

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