@langchain/langgraph 0.1.10-rc.3 → 0.2.0-rc.0

package/dist/graph/messages_annotation.d.ts

@@ -1,9 +1,10 @@
 import { BaseMessage } from "@langchain/core/messages";
+import { Messages } from "./message.js";
 /**
  * Prebuilt state annotation that combines returned messages.
  * Can handle standard messages and special modifiers like {@link RemoveMessage}
  * instances.
  */
 export declare const MessagesAnnotation: import("./annotation.js").AnnotationRoot<{
-    messages: import("../web.js").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
+    messages: import("../web.js").BinaryOperatorAggregate<BaseMessage[], Messages>;
 }>;

package/dist/graph/state.cjs

@@ -112,6 +112,16 @@ class StateGraph extends graph_js_1.Graph {
             writable: true,
             value: void 0
         });
+        /**
+         * Map schemas to managed values
+         * @internal
+         */
+        Object.defineProperty(this, "_schemaDefinitions", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
         if (isStateGraphArgsWithInputOutputSchemas(fields)) {
             this._schemaDefinition = fields.input.spec;
             this._inputDefinition = fields.input.spec;
@@ -154,6 +164,11 @@ class StateGraph extends graph_js_1.Graph {
         ]);
     }
     _addSchema(stateDefinition) {
+        if (this._schemaDefinitions.has(stateDefinition)) {
+            return;
+        }
+        // TODO: Support managed values
+        this._schemaDefinitions.set(stateDefinition, stateDefinition);
         for (const [key, val] of Object.entries(stateDefinition)) {
             let channel;
             if (typeof val === "function") {
@@ -188,10 +203,14 @@ class StateGraph extends graph_js_1.Graph {
         if (key === graph_js_1.END || key === graph_js_1.START) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
+        if (options?.input !== undefined) {
+            this._addSchema(options.input.spec);
+        }
         const nodeSpec = {
             runnable: (0, runnables_1._coerceToRunnable)(action),
             retryPolicy: options?.retryPolicy,
             metadata: options?.metadata,
+            input: options?.input?.spec ?? this._schemaDefinition,
         };
         this.nodes[key] = nodeSpec;
         return this;
@@ -228,7 +247,7 @@ class StateGraph extends graph_js_1.Graph {
             ...(Array.isArray(interruptAfter) ? interruptAfter : []),
         ]);
         // prepare output channels
-        const outputKeys = Object.keys(this._outputDefinition);
+        const outputKeys = Object.keys(this._schemaDefinitions.get(this._outputDefinition));
         const outputChannels = outputKeys.length === 1 && outputKeys[0] === ROOT ? ROOT : outputKeys;
         const streamKeys = Object.keys(this.channels);
         const streamChannels = streamKeys.length === 1 && streamKeys[0] === ROOT ? ROOT : streamKeys;
@@ -284,7 +303,7 @@ function _getChannels(schema) {
 }
 class CompiledStateGraph extends graph_js_1.CompiledGraph {
     attachNode(key, node) {
-        const stateKeys = Object.keys(this.builder._inputDefinition);
+        const stateKeys = Object.keys(this.builder.channels);
         function getStateKey(key, input) {
             if (!input) {
                 return write_js_1.SKIP_WRITE;
@@ -319,21 +338,26 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             });
         }
         else {
+            const inputDefinition = node?.input ?? this.builder._schemaDefinition;
+            const inputValues = Object.fromEntries(Object.keys(this.builder._schemaDefinitions.get(inputDefinition)).map((k) => [k, k]));
+            const isSingleInput = Object.keys(inputValues).length === 1 && ROOT in inputValues;
             this.channels[key] = new ephemeral_value_js_1.EphemeralValue(false);
             this.nodes[key] = new read_js_1.PregelNode({
                 triggers: [],
                 // read state keys
-                channels: stateKeys.length === 1 && stateKeys[0] === ROOT
-                    ? stateKeys
-                    : stateKeys.reduce((acc, k) => {
-                        acc[k] = k;
-                        return acc;
-                    }, {}),
+                channels: isSingleInput ? Object.keys(inputValues) : inputValues,
                 // publish to this channel and state keys
                 writers: [
                     new write_js_1.ChannelWrite(stateWriteEntries.concat({ channel: key, value: key }), [constants_js_1.TAG_HIDDEN]),
                 ],
+                mapper: isSingleInput
+                    ? undefined
+                    : // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                        (input) => {
+                            return Object.fromEntries(Object.entries(input).filter(([k]) => k in inputValues));
+                        },
                 bound: node?.runnable,
+                metadata: node?.metadata,
                 retryPolicy: node?.retryPolicy,
             });
         }

package/dist/graph/state.d.ts

@@ -15,11 +15,12 @@ export interface StateGraphArgs<Channels extends object | unknown> {
     }>;
 }
 export type StateGraphNodeSpec<RunInput, RunOutput> = NodeSpec<RunInput, RunOutput> & {
-    input?: any;
+    input?: StateDefinition;
     retryPolicy?: RetryPolicy;
 };
 export type StateGraphAddNodeOptions = {
     retryPolicy?: RetryPolicy;
+    input?: AnnotationRoot<any>;
 } & AddNodeOptions;
 export type StateGraphArgsWithStateSchema<SD extends StateDefinition, I extends StateDefinition, O extends StateDefinition> = {
     stateSchema: AnnotationRoot<SD>;
@@ -101,10 +102,15 @@ export declare class StateGraph<SD extends StateDefinition | unknown, S = SD ext
     _inputDefinition: I;
     /** @internal */
     _outputDefinition: O;
+    /**
+     * Map schemas to managed values
+     * @internal
+     */
+    _schemaDefinitions: Map<any, any>;
     constructor(fields: SD extends StateDefinition ? SD | AnnotationRoot<SD> | StateGraphArgs<S> | StateGraphArgsWithStateSchema<SD, I, O> | StateGraphArgsWithInputOutputSchemas<SD, O> : StateGraphArgs<S>);
     get allEdges(): Set<[string, string]>;
     _addSchema(stateDefinition: StateDefinition): void;
-    addNode<K extends string, NodeInput = S>(key: K, action: RunnableLike<NodeInput, 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>;
     addEdge(startKey: typeof START | N | N[], endKey: N | typeof END): this;
     compile({ checkpointer, interruptBefore, interruptAfter, }?: {
         checkpointer?: BaseCheckpointSaver;

package/dist/graph/state.js

@@ -109,6 +109,16 @@ export class StateGraph extends Graph {
             writable: true,
             value: void 0
         });
+        /**
+         * Map schemas to managed values
+         * @internal
+         */
+        Object.defineProperty(this, "_schemaDefinitions", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
         if (isStateGraphArgsWithInputOutputSchemas(fields)) {
             this._schemaDefinition = fields.input.spec;
             this._inputDefinition = fields.input.spec;
@@ -151,6 +161,11 @@ export class StateGraph extends Graph {
         ]);
     }
     _addSchema(stateDefinition) {
+        if (this._schemaDefinitions.has(stateDefinition)) {
+            return;
+        }
+        // TODO: Support managed values
+        this._schemaDefinitions.set(stateDefinition, stateDefinition);
         for (const [key, val] of Object.entries(stateDefinition)) {
             let channel;
             if (typeof val === "function") {
@@ -185,10 +200,14 @@ export class StateGraph extends Graph {
         if (key === END || key === START) {
             throw new Error(`Node \`${key}\` is reserved.`);
         }
+        if (options?.input !== undefined) {
+            this._addSchema(options.input.spec);
+        }
         const nodeSpec = {
             runnable: _coerceToRunnable(action),
             retryPolicy: options?.retryPolicy,
             metadata: options?.metadata,
+            input: options?.input?.spec ?? this._schemaDefinition,
         };
         this.nodes[key] = nodeSpec;
         return this;
@@ -225,7 +244,7 @@ export class StateGraph extends Graph {
             ...(Array.isArray(interruptAfter) ? interruptAfter : []),
         ]);
         // prepare output channels
-        const outputKeys = Object.keys(this._outputDefinition);
+        const outputKeys = Object.keys(this._schemaDefinitions.get(this._outputDefinition));
         const outputChannels = outputKeys.length === 1 && outputKeys[0] === ROOT ? ROOT : outputKeys;
         const streamKeys = Object.keys(this.channels);
         const streamChannels = streamKeys.length === 1 && streamKeys[0] === ROOT ? ROOT : streamKeys;
@@ -280,7 +299,7 @@ function _getChannels(schema) {
 }
 export class CompiledStateGraph extends CompiledGraph {
     attachNode(key, node) {
-        const stateKeys = Object.keys(this.builder._inputDefinition);
+        const stateKeys = Object.keys(this.builder.channels);
         function getStateKey(key, input) {
             if (!input) {
                 return SKIP_WRITE;
@@ -315,21 +334,26 @@ export class CompiledStateGraph extends CompiledGraph {
             });
         }
         else {
+            const inputDefinition = node?.input ?? this.builder._schemaDefinition;
+            const inputValues = Object.fromEntries(Object.keys(this.builder._schemaDefinitions.get(inputDefinition)).map((k) => [k, k]));
+            const isSingleInput = Object.keys(inputValues).length === 1 && ROOT in inputValues;
             this.channels[key] = new EphemeralValue(false);
             this.nodes[key] = new PregelNode({
                 triggers: [],
                 // read state keys
-                channels: stateKeys.length === 1 && stateKeys[0] === ROOT
-                    ? stateKeys
-                    : stateKeys.reduce((acc, k) => {
-                        acc[k] = k;
-                        return acc;
-                    }, {}),
+                channels: isSingleInput ? Object.keys(inputValues) : inputValues,
                 // publish to this channel and state keys
                 writers: [
                     new ChannelWrite(stateWriteEntries.concat({ channel: key, value: key }), [TAG_HIDDEN]),
                 ],
+                mapper: isSingleInput
+                    ? undefined
+                    : // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                        (input) => {
+                            return Object.fromEntries(Object.entries(input).filter(([k]) => k in inputValues));
+                        },
                 bound: node?.runnable,
+                metadata: node?.metadata,
                 retryPolicy: node?.retryPolicy,
             });
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.1.10-rc.3",
+  "version": "0.2.0-rc.0",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -31,26 +31,19 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/core": ">=0.2.31 <0.3.0",
     "@langchain/langgraph-checkpoint": "~0.0.4",
-    "@langchain/langgraph-checkpoint-sqlite": "~0.0.1",
     "double-ended-queue": "^2.1.0-0",
     "uuid": "^10.0.0",
     "zod": "^3.23.8"
   },
   "peerDependencies": {
-    "better-sqlite3": "^9.5.0"
-  },
-  "peerDependenciesMeta": {
-    "better-sqlite3": {
-      "optional": true
-    }
+    "@langchain/core": ">=0.2.31 <0.3.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
     "@langchain/anthropic": "^0.2.12",
-    "@langchain/community": "^0.2.25",
-    "@langchain/core": "^0.2.29",
+    "@langchain/community": "^0.2.31",
+    "@langchain/core": "^0.2.31",
     "@langchain/openai": "^0.2.4",
     "@langchain/scripts": ">=0.1.2 <0.2.0",
     "@swc/core": "^1.3.90",