@langchain/langgraph 0.2.4 → 0.2.5-rc.0

package/dist/graph/state.cjs

@@ -78,7 +78,7 @@ const ROOT = "__root__";
  * ```
  */
 class StateGraph extends graph_js_1.Graph {
-    constructor(fields) {
+    constructor(fields, configSchema) {
         super();
         Object.defineProperty(this, "channels", {
             enumerable: true,
@@ -124,6 +124,13 @@ class StateGraph extends graph_js_1.Graph {
             writable: true,
             value: new Map()
         });
+        /** @internal */
+        Object.defineProperty(this, "_configSchema", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         if (isStateGraphArgsWithInputOutputSchemas(fields)) {
             this._schemaDefinition = fields.input.spec;
             this._inputDefinition = fields.input.spec;
@@ -152,6 +159,7 @@ class StateGraph extends graph_js_1.Graph {
         this._addSchema(this._schemaDefinition);
         this._addSchema(this._inputDefinition);
         this._addSchema(this._outputDefinition);
+        this._configSchema = configSchema?.spec;
     }
     get allEdges() {
         return new Set([
@@ -251,6 +259,9 @@ class StateGraph extends graph_js_1.Graph {
         // create empty compiled graph
         const compiled = new CompiledStateGraph({
             builder: this,
+            configKeys: this._configSchema
+                ? Object.keys(this._configSchema)
+                : undefined,
             checkpointer,
             interruptAfter,
             interruptBefore,

package/dist/graph/state.d.ts

@@ -95,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> 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>> {
     channels: Record<string, BaseChannel | ManagedValueSpec>;
     waitingEdges: Set<[N[], N]>;
     /** @internal */
@@ -109,20 +109,22 @@ export declare class StateGraph<SD extends StateDefinition | unknown, S = SD ext
      * @internal
      */
     _schemaDefinitions: Map<any, any>;
-    constructor(fields: SD extends StateDefinition ? SD | AnnotationRoot<SD> | StateGraphArgs<S> | StateGraphArgsWithStateSchema<SD, I, O> | StateGraphArgsWithInputOutputSchemas<SD, O> : StateGraphArgs<S>);
+    /** @internal */
+    _configSchema: C | undefined;
+    constructor(fields: SD extends StateDefinition ? SD | AnnotationRoot<SD> | StateGraphArgs<S> | StateGraphArgsWithStateSchema<SD, I, O> | StateGraphArgsWithInputOutputSchemas<SD, O> : StateGraphArgs<S>, configSchema?: AnnotationRoot<C>);
     get allEdges(): Set<[string, string]>;
     _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>;
+    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>;
     addEdge(startKey: typeof START | N | N[], endKey: N | typeof END): this;
     compile({ checkpointer, store, interruptBefore, interruptAfter, }?: {
         checkpointer?: BaseCheckpointSaver;
         store?: BaseStore;
         interruptBefore?: N[] | All;
         interruptAfter?: N[] | All;
-    }): CompiledStateGraph<S, U, N, I, O>;
+    }): CompiledStateGraph<S, U, N, I, O, C>;
 }
-export declare class CompiledStateGraph<S, U, N extends string = typeof START, I extends StateDefinition = StateDefinition, O extends StateDefinition = StateDefinition> extends CompiledGraph<N, S, U> {
-    builder: StateGraph<unknown, S, U, N, I, O>;
+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> {
+    builder: StateGraph<unknown, S, U, N, I, O, C>;
     attachNode(key: typeof START, node?: never): void;
     attachNode(key: N, node: StateGraphNodeSpec<S, U>): void;
     attachEdge(start: N | N[] | "__start__", end: N | "__end__"): void;

package/dist/graph/state.js

@@ -75,7 +75,7 @@ const ROOT = "__root__";
  * ```
  */
 export class StateGraph extends Graph {
-    constructor(fields) {
+    constructor(fields, configSchema) {
         super();
         Object.defineProperty(this, "channels", {
             enumerable: true,
@@ -121,6 +121,13 @@ export class StateGraph extends Graph {
             writable: true,
             value: new Map()
         });
+        /** @internal */
+        Object.defineProperty(this, "_configSchema", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         if (isStateGraphArgsWithInputOutputSchemas(fields)) {
             this._schemaDefinition = fields.input.spec;
             this._inputDefinition = fields.input.spec;
@@ -149,6 +156,7 @@ export class StateGraph extends Graph {
         this._addSchema(this._schemaDefinition);
         this._addSchema(this._inputDefinition);
         this._addSchema(this._outputDefinition);
+        this._configSchema = configSchema?.spec;
     }
     get allEdges() {
         return new Set([
@@ -248,6 +256,9 @@ export class StateGraph extends Graph {
         // create empty compiled graph
         const compiled = new CompiledStateGraph({
             builder: this,
+            configKeys: this._configSchema
+                ? Object.keys(this._configSchema)
+                : undefined,
             checkpointer,
             interruptAfter,
             interruptBefore,

package/dist/prebuilt/agent_executor.d.ts

@@ -18,5 +18,5 @@ export interface AgentExecutorState {
 export declare function createAgentExecutor({ agentRunnable, tools, }: {
     agentRunnable: Runnable;
     tools: Array<Tool> | ToolExecutor;
-}): import("../graph/state.js").CompiledStateGraph<AgentExecutorState, Partial<AgentExecutorState>, "__start__" | "agent" | "action", import("../graph/annotation.js").StateDefinition, import("../graph/annotation.js").StateDefinition>;
+}): import("../graph/state.js").CompiledStateGraph<AgentExecutorState, Partial<AgentExecutorState>, "__start__" | "agent" | "action", import("../graph/annotation.js").StateDefinition, import("../graph/annotation.js").StateDefinition, import("../graph/annotation.js").StateDefinition>;
 export {};

package/dist/pregel/index.cjs

@@ -113,6 +113,12 @@ class Pregel extends runnables_1.Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "configKeys", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "autoValidate", {
             enumerable: true,
             configurable: true,
@@ -183,6 +189,7 @@ class Pregel extends runnables_1.Runnable {
         this.streamMode = streamMode ?? this.streamMode;
         this.inputChannels = fields.inputChannels;
         this.outputChannels = fields.outputChannels;
+        this.configKeys = fields.configKeys;
         this.streamChannels = fields.streamChannels ?? this.streamChannels;
         this.interruptAfter = fields.interruptAfter;
         this.interruptBefore = fields.interruptBefore;
@@ -425,6 +432,12 @@ class Pregel extends runnables_1.Runnable {
         else {
             defaultCheckpointer = this.checkpointer;
         }
+        if (this.configKeys !== undefined) {
+            const newConfigurable = Object.fromEntries(Object.entries(rest.configurable ?? {}).filter(([key]) => {
+                return this.configKeys?.includes(key);
+            }));
+            rest.configurable = newConfigurable;
+        }
         return [
             defaultDebug,
             defaultStreamMode,

package/dist/pregel/index.d.ts

@@ -45,6 +45,7 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     channels: Cc;
     inputChannels: keyof Cc | Array<keyof Cc>;
     outputChannels: keyof Cc | Array<keyof Cc>;
+    configKeys?: string[];
     autoValidate: boolean;
     streamMode: StreamMode[];
     streamChannels?: keyof Cc | Array<keyof Cc>;

package/dist/pregel/index.js

@@ -109,6 +109,12 @@ export class Pregel extends Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "configKeys", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "autoValidate", {
             enumerable: true,
             configurable: true,
@@ -179,6 +185,7 @@ export class Pregel extends Runnable {
         this.streamMode = streamMode ?? this.streamMode;
         this.inputChannels = fields.inputChannels;
         this.outputChannels = fields.outputChannels;
+        this.configKeys = fields.configKeys;
         this.streamChannels = fields.streamChannels ?? this.streamChannels;
         this.interruptAfter = fields.interruptAfter;
         this.interruptBefore = fields.interruptBefore;
@@ -421,6 +428,12 @@ export class Pregel extends Runnable {
         else {
             defaultCheckpointer = this.checkpointer;
         }
+        if (this.configKeys !== undefined) {
+            const newConfigurable = Object.fromEntries(Object.entries(rest.configurable ?? {}).filter(([key]) => {
+                return this.configKeys?.includes(key);
+            }));
+            rest.configurable = newConfigurable;
+        }
         return [
             defaultDebug,
             defaultStreamMode,

package/dist/pregel/types.d.ts

@@ -26,6 +26,7 @@ export interface PregelInterface<Nn extends StrRecord<string, PregelNode>, Cc ex
     streamMode?: StreamMode | StreamMode[];
     inputChannels: keyof Cc | Array<keyof Cc>;
     outputChannels: keyof Cc | Array<keyof Cc>;
+    configKeys?: string[];
     /**
      * @default []
      */

package/package.json

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