@langchain/langgraph 0.2.8 → 0.2.10-rc.0

package/dist/constants.cjs

@@ -1,18 +1,23 @@
 "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_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_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 +27,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

@@ -4,15 +4,20 @@ 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_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

@@ -4,12 +4,17 @@ 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_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 +24,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

@@ -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) {
@@ -385,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

@@ -1,10 +1,9 @@
 import { RunnableLike } from "@langchain/core/runnables";
-import { All, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, BaseStore } 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 { BaseStore } from "../store/base.js";
+import type { RetryPolicy } from "../pregel/utils/index.js";
 import { ManagedValueSpec } from "../managed/base.js";
 export type ChannelReducers<Channels extends object> = {
     [K in keyof Channels]: SingleReducer<Channels[K], any>;

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";
@@ -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) {
@@ -381,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

@@ -1,8 +1,8 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SharedValue = void 0;
 const base_js_1 = require("./base.cjs");
-const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 class SharedValue extends base_js_1.WritableManagedValue {
     constructor(config, params) {
@@ -32,7 +32,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
             value: {}
         });
         this.scope = params.scope;
-        this.store = config.configurable?.[constants_js_1.CONFIG_KEY_STORE] || null;
+        this.store = config.store || null;
         if (!this.store) {
             this.ns = null;
         }
@@ -41,13 +41,12 @@ class SharedValue extends base_js_1.WritableManagedValue {
             const scopedValueString = typeof scopeValue === "string"
                 ? scopeValue
                 : JSON.stringify(scopeValue);
-            this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
+            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
     static async initialize(config, args) {
         const instance = new this(config, args);
         await instance.loadStore();
@@ -73,7 +72,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
                     if (k in this.value) {
                         delete this.value[k];
                         if (this.ns) {
-                            writes.push([this.ns, k, null]);
+                            writes.push({ namespace: this.ns, id: k, value: null });
                         }
                     }
                 }
@@ -83,7 +82,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
                 else {
                     this.value[k] = v;
                     if (this.ns) {
-                        writes.push([this.ns, k, v]);
+                        writes.push({ namespace: this.ns, id: k, value: v });
                     }
                 }
             }
@@ -95,13 +94,16 @@ class SharedValue extends base_js_1.WritableManagedValue {
             this.processUpdate(values);
         }
         else {
-            await this.store.put(this.processUpdate(values));
+            await this.store.batch(this.processUpdate(values));
         }
     }
     async loadStore() {
         if (this.store && this.ns) {
-            const saved = await this.store.list([this.ns]);
-            this.value = saved[this.ns] || {};
+            const saved = await this.store.search(this.ns);
+            this.value = saved.reduce((acc, item) => {
+                acc[item.id] = item.value;
+                return acc;
+            }, {});
         }
         return false;
     }

package/dist/managed/shared_value.d.ts

@@ -1,8 +1,9 @@
 import { RunnableConfig } from "@langchain/core/runnables";
-import { BaseStore, type Values } from "../store/base.js";
+import { BaseStore } from "@langchain/langgraph-checkpoint";
 import { ConfiguredManagedValue, ManagedValue, ManagedValueParams, WritableManagedValue } from "./base.js";
-type Value = Record<string, Values>;
-type Update = Record<string, Values | null>;
+import { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
+type Value = Record<string, Record<string, any>>;
+type Update = Record<string, Record<string, any> | null>;
 export interface SharedValueParams extends ManagedValueParams {
     scope: string;
     key: string;
@@ -10,9 +11,9 @@ export interface SharedValueParams extends ManagedValueParams {
 export declare class SharedValue extends WritableManagedValue<Value, Update> {
     scope: string;
     store: BaseStore | null;
-    ns: string | null;
+    ns: ["scoped", string, string, any] | null;
     value: Value;
-    constructor(config: RunnableConfig, params: SharedValueParams);
+    constructor(config: LangGraphRunnableConfig, params: SharedValueParams);
     static initialize<Value = any>(config: RunnableConfig, args: SharedValueParams): Promise<ManagedValue<Value>>;
     static on(scope: string): ConfiguredManagedValue<Value>;
     call(_step: number): Value;

package/dist/managed/shared_value.js

@@ -1,5 +1,5 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { ChannelKeyPlaceholder, WritableManagedValue, } from "./base.js";
-import { CONFIG_KEY_STORE } from "../constants.js";
 import { InvalidUpdateError } from "../errors.js";
 export class SharedValue extends WritableManagedValue {
     constructor(config, params) {
@@ -29,7 +29,7 @@ export class SharedValue extends WritableManagedValue {
             value: {}
         });
         this.scope = params.scope;
-        this.store = config.configurable?.[CONFIG_KEY_STORE] || null;
+        this.store = config.store || null;
         if (!this.store) {
             this.ns = null;
         }
@@ -38,13 +38,12 @@ export class SharedValue extends WritableManagedValue {
             const scopedValueString = typeof scopeValue === "string"
                 ? scopeValue
                 : JSON.stringify(scopeValue);
-            this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
+            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
     static async initialize(config, args) {
         const instance = new this(config, args);
         await instance.loadStore();
@@ -70,7 +69,7 @@ export class SharedValue extends WritableManagedValue {
                     if (k in this.value) {
                         delete this.value[k];
                         if (this.ns) {
-                            writes.push([this.ns, k, null]);
+                            writes.push({ namespace: this.ns, id: k, value: null });
                         }
                     }
                 }
@@ -80,7 +79,7 @@ export class SharedValue extends WritableManagedValue {
                 else {
                     this.value[k] = v;
                     if (this.ns) {
-                        writes.push([this.ns, k, v]);
+                        writes.push({ namespace: this.ns, id: k, value: v });
                     }
                 }
             }
@@ -92,13 +91,16 @@ export class SharedValue extends WritableManagedValue {
             this.processUpdate(values);
         }
         else {
-            await this.store.put(this.processUpdate(values));
+            await this.store.batch(this.processUpdate(values));
         }
     }
     async loadStore() {
         if (this.store && this.ns) {
-            const saved = await this.store.list([this.ns]);
-            this.value = saved[this.ns] || {};
+            const saved = await this.store.search(this.ns);
+            this.value = saved.reduce((acc, item) => {
+                acc[item.id] = item.value;
+                return acc;
+            }, {});
         }
         return false;
     }