@langchain/langgraph 0.3.0 → 0.3.2

package/dist/func/index.cjs

@@ -59,18 +59,23 @@ const write_js_1 = require("../pregel/write.cjs");
  * ```
  */
 function task(optionsOrName, func) {
-    const { name, retry, cache: userCache, } = typeof optionsOrName === "string"
-        ? { name: optionsOrName, retry: undefined, cache: undefined }
+    const options = typeof optionsOrName === "string"
+        ? { name: optionsOrName, retry: undefined, cachePolicy: undefined }
         : optionsOrName;
+    const { name, retry } = options;
     if ((0, utils_js_1.isAsyncGeneratorFunction)(func) || (0, utils_js_1.isGeneratorFunction)(func)) {
         throw new Error("Generators are disallowed as tasks. For streaming responses, use config.write.");
     }
+    const cachePolicy = options.cachePolicy ??
+        // `cache` was mistakingly used as an alias for `cachePolicy` in v0.3.x,
+        // TODO: remove in 1.x
+        ("cache" in options ? options.cache : undefined);
     let cache;
-    if (typeof userCache === "boolean") {
-        cache = userCache ? {} : undefined;
+    if (typeof cachePolicy === "boolean") {
+        cache = cachePolicy ? {} : undefined;
     }
     else {
-        cache = userCache;
+        cache = cachePolicy;
     }
     return (...args) => {
         return (0, call_js_1.call)({ func, name, retry, cache }, ...args);

package/dist/func/index.d.ts

@@ -9,7 +9,7 @@ import { EntrypointFinal, EntrypointReturnT, EntrypointFinalSaveT, EntrypointFun
 /**
  * Options for the {@link task} function
  */
-export type TaskOptions = {
+export interface TaskOptions {
     /**
      * The name of the task, analogous to the node name in {@link StateGraph}.
      */
@@ -22,8 +22,8 @@ export type TaskOptions = {
     /**
      * The cache policy for the task. Configures how the task should be cached.
      */
-    cache?: CachePolicy | boolean;
-};
+    cachePolicy?: CachePolicy;
+}
 /**
  * Define a LangGraph task using the `task` function.
  *

package/dist/func/index.js

@@ -54,18 +54,23 @@ import { ChannelWrite, PASSTHROUGH } from "../pregel/write.js";
  * ```
  */
 export function task(optionsOrName, func) {
-    const { name, retry, cache: userCache, } = typeof optionsOrName === "string"
-        ? { name: optionsOrName, retry: undefined, cache: undefined }
+    const options = typeof optionsOrName === "string"
+        ? { name: optionsOrName, retry: undefined, cachePolicy: undefined }
         : optionsOrName;
+    const { name, retry } = options;
     if (isAsyncGeneratorFunction(func) || isGeneratorFunction(func)) {
         throw new Error("Generators are disallowed as tasks. For streaming responses, use config.write.");
     }
+    const cachePolicy = options.cachePolicy ??
+        // `cache` was mistakingly used as an alias for `cachePolicy` in v0.3.x,
+        // TODO: remove in 1.x
+        ("cache" in options ? options.cache : undefined);
     let cache;
-    if (typeof userCache === "boolean") {
-        cache = userCache ? {} : undefined;
+    if (typeof cachePolicy === "boolean") {
+        cache = cachePolicy ? {} : undefined;
     }
     else {
-        cache = userCache;
+        cache = cachePolicy;
     }
     return (...args) => {
         return call({ func, name, retry, cache }, ...args);

package/dist/func/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/func/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,kCAAkC,EAAE,MAAM,4BAA4B,CAAC;AAChF,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EACL,yBAAyB,EACzB,GAAG,EACH,QAAQ,EACR,KAAK,EACL,UAAU,GACX,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAEnE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAStD,OAAO,EACL,gBAAgB,EAChB,wBAAwB,EACxB,mBAAmB,GACpB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAsB/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6CG;AACH,MAAM,UAAU,IAAI,CAClB,aAAmC,EACnC,IAA8B;IAE9B,MAAM,EACJ,IAAI,EACJ,KAAK,EACL,KAAK,EAAE,SAAS,GACjB,GAAG,OAAO,aAAa,KAAK,QAAQ;QACnC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;QAC7D,CAAC,CAAC,aAAa,CAAC;IAClB,IAAI,wBAAwB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CACb,gFAAgF,CACjF,CAAC;IACJ,CAAC;IAED,IAAI,KAA8B,CAAC;IACnC,IAAI,OAAO,SAAS,KAAK,SAAS,EAAE,CAAC;QACnC,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACrC,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,SAAS,CAAC;IACpB,CAAC;IAED,OAAO,CAAC,GAAG,IAAW,EAAE,EAAE;QACxB,OAAO,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC;AACJ,CAAC;AA2ED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkIG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,SAAS,UAAU,CAC3C,aAAyC,EACzC,IAAqC;IAErC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,GACxC,OAAO,aAAa,KAAK,QAAQ;QAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;QACpE,CAAC,CAAC,aAAa,CAAC;IACpB,IAAI,wBAAwB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,SAAS,CAAC;IAC7B,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAEnD,mDAAmD;IACnD,SAAS,iBAAiB,CACxB,KAAc;QAEd,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,WAAW,IAAI,KAAK;YACpB,KAAK,CAAC,SAAS,KAAK,gBAAgB,CACrC,CAAC;IACJ,CAAC;IAED,gFAAgF;IAChF,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC;QAC5C,IAAI,EAAE,kBAAkB;QACxB,IAAI,EAAE,CAAC,KAAc,EAAE,EAAE;YACvB,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QACxD,CAAC;KACF,CAAC,CAAC;IAEH,8EAA8E;IAC9E,MAAM,cAAc,GAAG,IAAI,gBAAgB,CAAC;QAC1C,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,CAAC,KAAc,EAAE,EAAE;YACvB,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QACvD,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,IAAI,UAAU,CAAqC;QACxE,KAAK;QACL,QAAQ,EAAE,CAAC,KAAK,CAAC;QACjB,QAAQ,EAAE,CAAC,KAAK,CAAC;QACjB,OAAO,EAAE;YACP,IAAI,YAAY,CACd;gBACE,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,gBAAgB,EAAE;gBAC9D,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE;aAClE,EACD,CAAC,UAAU,CAAC,CACb;SACF;KACF,CAAC,CAAC;IAEH,OAAO,IAAI,MAAM,CAUf;QACA,IAAI;QACJ,YAAY;QACZ,KAAK,EAAE;YACL,CAAC,IAAI,CAAC,EAAE,cAAc;SACvB;QACD,QAAQ,EAAE;YACR,CAAC,KAAK,CAAC,EAAE,IAAI,cAAc,EAAU;YACrC,CAAC,GAAG,CAAC,EAAE,IAAI,SAAS,EAA8B;YAClD,CAAC,QAAQ,CAAC,EAAE,IAAI,SAAS,EAAiC;SAC3D;QACD,aAAa,EAAE,KAAK;QACpB,cAAc,EAAE,GAAG;QACnB,cAAc,EAAE,GAAG;QACnB,UAAU;QACV,KAAK;QACL,KAAK;KACN,CAAC,CAAC;AACL,CAAuB,CAAC;AAExB,iDAAiD;AACjD,UAAU,CAAC,KAAK,GAAG,SAAS,KAAK,CAAgB,EAC/C,KAAK,EACL,IAAI,GAIL;IACC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC;AACtD,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,MAAM,GACV,kCAAkC,CAAC,iBAAiB,EAAE,CAAC;IACzD,OAAO,MAAM,CAAC,YAAY,EAAE,CAAC,yBAAyB,CAAW,CAAC;AACpE,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/func/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,kCAAkC,EAAE,MAAM,4BAA4B,CAAC;AAChF,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EACL,yBAAyB,EACzB,GAAG,EACH,QAAQ,EACR,KAAK,EACL,UAAU,GACX,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAEnE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAStD,OAAO,EACL,gBAAgB,EAChB,wBAAwB,EACxB,mBAAmB,GACpB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAsB/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6CG;AACH,MAAM,UAAU,IAAI,CAClB,aAAmC,EACnC,IAA8B;IAE9B,MAAM,OAAO,GACX,OAAO,aAAa,KAAK,QAAQ;QAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE;QACnE,CAAC,CAAC,aAAa,CAAC;IAEpB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAChC,IAAI,wBAAwB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CACb,gFAAgF,CACjF,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GACf,OAAO,CAAC,WAAW;QACnB,wEAAwE;QACxE,sBAAsB;QACtB,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,CAAE,OAAO,CAAC,KAAqB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpE,IAAI,KAA8B,CAAC;IACnC,IAAI,OAAO,WAAW,KAAK,SAAS,EAAE,CAAC;QACrC,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,WAAW,CAAC;IACtB,CAAC;IAED,OAAO,CAAC,GAAG,IAAW,EAAE,EAAE;QACxB,OAAO,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC;AACJ,CAAC;AA2ED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkIG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,SAAS,UAAU,CAC3C,aAAyC,EACzC,IAAqC;IAErC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,GACxC,OAAO,aAAa,KAAK,QAAQ;QAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;QACpE,CAAC,CAAC,aAAa,CAAC;IACpB,IAAI,wBAAwB,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,SAAS,CAAC;IAC7B,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAEnD,mDAAmD;IACnD,SAAS,iBAAiB,CACxB,KAAc;QAEd,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,WAAW,IAAI,KAAK;YACpB,KAAK,CAAC,SAAS,KAAK,gBAAgB,CACrC,CAAC;IACJ,CAAC;IAED,gFAAgF;IAChF,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC;QAC5C,IAAI,EAAE,kBAAkB;QACxB,IAAI,EAAE,CAAC,KAAc,EAAE,EAAE;YACvB,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QACxD,CAAC;KACF,CAAC,CAAC;IAEH,8EAA8E;IAC9E,MAAM,cAAc,GAAG,IAAI,gBAAgB,CAAC;QAC1C,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,CAAC,KAAc,EAAE,EAAE;YACvB,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QACvD,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,IAAI,UAAU,CAAqC;QACxE,KAAK;QACL,QAAQ,EAAE,CAAC,KAAK,CAAC;QACjB,QAAQ,EAAE,CAAC,KAAK,CAAC;QACjB,OAAO,EAAE;YACP,IAAI,YAAY,CACd;gBACE,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,gBAAgB,EAAE;gBAC9D,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE;aAClE,EACD,CAAC,UAAU,CAAC,CACb;SACF;KACF,CAAC,CAAC;IAEH,OAAO,IAAI,MAAM,CAUf;QACA,IAAI;QACJ,YAAY;QACZ,KAAK,EAAE;YACL,CAAC,IAAI,CAAC,EAAE,cAAc;SACvB;QACD,QAAQ,EAAE;YACR,CAAC,KAAK,CAAC,EAAE,IAAI,cAAc,EAAU;YACrC,CAAC,GAAG,CAAC,EAAE,IAAI,SAAS,EAA8B;YAClD,CAAC,QAAQ,CAAC,EAAE,IAAI,SAAS,EAAiC;SAC3D;QACD,aAAa,EAAE,KAAK;QACpB,cAAc,EAAE,GAAG;QACnB,cAAc,EAAE,GAAG;QACnB,UAAU;QACV,KAAK;QACL,KAAK;KACN,CAAC,CAAC;AACL,CAAuB,CAAC;AAExB,iDAAiD;AACjD,UAAU,CAAC,KAAK,GAAG,SAAS,KAAK,CAAgB,EAC/C,KAAK,EACL,IAAI,GAIL;IACC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC;AACtD,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,MAAM,GACV,kCAAkC,CAAC,iBAAiB,EAAE,CAAC;IACzD,OAAO,MAAM,CAAC,YAAY,EAAE,CAAC,yBAAyB,CAAW,CAAC;AACpE,CAAC"}

package/dist/graph/messages_annotation.cjs

@@ -5,7 +5,7 @@ exports.MessagesZodState = exports.MessagesAnnotation = void 0;
 const zod_1 = require("zod");
 const annotation_js_1 = require("./annotation.cjs");
 const message_js_1 = require("./message.cjs");
-const state_js_1 = require("./zod/state.cjs");
+const meta_js_1 = require("./zod/meta.cjs");
 /**
  * Prebuilt state annotation that combines returned messages.
  * Can handle standard messages and special modifiers like {@link RemoveMessage}
@@ -87,7 +87,7 @@ exports.MessagesAnnotation = annotation_js_1.Annotation.Root({
  * ```
  */
 exports.MessagesZodState = zod_1.z.object({
-    messages: (0, state_js_1.withLangGraph)(zod_1.z.custom(), {
+    messages: (0, meta_js_1.withLangGraph)(zod_1.z.custom(), {
         reducer: {
             schema: zod_1.z.custom(),
             fn: message_js_1.messagesStateReducer,

package/dist/graph/messages_annotation.d.ts

@@ -79,9 +79,9 @@ export declare const MessagesAnnotation: import("./annotation.js").AnnotationRoo
  * ```
  */
 export declare const MessagesZodState: z.ZodObject<{
-    messages: z.ZodType<BaseMessage[], z.ZodTypeDef, Messages>;
+    messages: import("./zod/meta.js").ReducedZodChannel<z.ZodType<BaseMessage[], z.ZodTypeDef, BaseMessage[]>, import("@langchain/core/utils/types").InteropZodType<Messages>>;
 }, "strip", z.ZodTypeAny, {
     messages: BaseMessage[];
 }, {
-    messages: Messages;
+    messages: BaseMessage[];
 }>;

package/dist/graph/messages_annotation.js

@@ -2,7 +2,7 @@
 import { z } from "zod";
 import { Annotation } from "./annotation.js";
 import { messagesStateReducer } from "./message.js";
-import { withLangGraph } from "./zod/state.js";
+import { withLangGraph } from "./zod/meta.js";
 /**
  * Prebuilt state annotation that combines returned messages.
  * Can handle standard messages and special modifiers like {@link RemoveMessage}

package/dist/graph/messages_annotation.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages_annotation.js","sourceRoot":"","sources":["../../src/graph/messages_annotation.ts"],"names":[],"mappings":"AAAA,0CAA0C;AAG1C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAY,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC,IAAI,CAAC;IAChD,QAAQ,EAAE,UAAU,CAA0B;QAC5C,OAAO,EAAE,oBAAoB;QAC7B,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;KAClB,CAAC;CACH,CAAC,CAAC;AAEH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,MAAM,EAAiB,EAAE;QACjD,OAAO,EAAE;YACP,MAAM,EAAE,CAAC,CAAC,MAAM,EAAY;YAC5B,EAAE,EAAE,oBAAoB;SACzB;QACD,eAAe,EAAE;YACf,cAAc,EAAE,UAAU;SAC3B;QACD,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;KAClB,CAAC;CACH,CAAC,CAAC"}
+{"version":3,"file":"messages_annotation.js","sourceRoot":"","sources":["../../src/graph/messages_annotation.ts"],"names":[],"mappings":"AAAA,0CAA0C;AAG1C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAY,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC,IAAI,CAAC;IAChD,QAAQ,EAAE,UAAU,CAA0B;QAC5C,OAAO,EAAE,oBAAoB;QAC7B,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;KAClB,CAAC;CACH,CAAC,CAAC;AAEH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,MAAM,EAAiB,EAAE;QACjD,OAAO,EAAE;YACP,MAAM,EAAE,CAAC,CAAC,MAAM,EAAY;YAC5B,EAAE,EAAE,oBAAoB;SACzB;QACD,eAAe,EAAE;YACf,cAAc,EAAE,UAAU;SAC3B;QACD,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE;KAClB,CAAC;CACH,CAAC,CAAC"}

package/dist/graph/state.cjs

@@ -4,6 +4,7 @@ exports.CompiledStateGraph = exports.StateGraph = void 0;
 exports.typedNode = typedNode;
 /* eslint-disable @typescript-eslint/no-use-before-define */
 const runnables_1 = require("@langchain/core/runnables");
+const types_1 = require("@langchain/core/utils/types");
 const base_js_1 = require("../channels/base.cjs");
 const graph_js_1 = require("./graph.cjs");
 const write_js_1 = require("../pregel/write.cjs");
@@ -16,8 +17,8 @@ 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 state_js_1 = require("./zod/state.cjs");
 const last_value_js_1 = require("../channels/last_value.cjs");
+const meta_js_1 = require("./zod/meta.cjs");
 const ROOT = "__root__";
 const PartialStateSchema = Symbol.for("langgraph.state.partial");
 /**
@@ -150,6 +151,13 @@ class StateGraph extends graph_js_1.Graph {
             writable: true,
             value: new Map()
         });
+        /** @internal */
+        Object.defineProperty(this, "_metaRegistry", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: meta_js_1.schemaMetaRegistry
+        });
         /** @internal Used only for typing. */
         Object.defineProperty(this, "_configSchema", {
             enumerable: true,
@@ -165,9 +173,13 @@ class StateGraph extends graph_js_1.Graph {
             value: void 0
         });
         if (isZodStateGraphArgsWithStateSchema(fields)) {
-            const stateDef = (0, state_js_1.getChannelsFromZod)(fields.state);
-            const inputDef = fields.input != null ? (0, state_js_1.getChannelsFromZod)(fields.input) : stateDef;
-            const outputDef = fields.output != null ? (0, state_js_1.getChannelsFromZod)(fields.output) : stateDef;
+            const stateDef = this._metaRegistry.getChannelsForSchema(fields.state);
+            const inputDef = fields.input != null
+                ? this._metaRegistry.getChannelsForSchema(fields.input)
+                : stateDef;
+            const outputDef = fields.output != null
+                ? this._metaRegistry.getChannelsForSchema(fields.output)
+                : stateDef;
             this._schemaDefinition = stateDef;
             this._schemaRuntimeDefinition = fields.state;
             this._inputDefinition = inputDef;
@@ -175,8 +187,8 @@ class StateGraph extends graph_js_1.Graph {
             this._outputDefinition = outputDef;
             this._outputRuntimeDefinition = fields.output ?? fields.state;
         }
-        else if ((0, state_js_1.isAnyZodObject)(fields)) {
-            const stateDef = (0, state_js_1.getChannelsFromZod)(fields);
+        else if ((0, types_1.isInteropZodObject)(fields)) {
+            const stateDef = this._metaRegistry.getChannelsForSchema(fields);
             this._schemaDefinition = stateDef;
             this._schemaRuntimeDefinition = fields;
             this._inputDefinition = stateDef;
@@ -212,8 +224,8 @@ class StateGraph extends graph_js_1.Graph {
         this._addSchema(this._schemaDefinition);
         this._addSchema(this._inputDefinition);
         this._addSchema(this._outputDefinition);
-        if ((0, state_js_1.isAnyZodObject)(configSchema)) {
-            this._configRuntimeSchema = configSchema.passthrough();
+        if ((0, types_1.isInteropZodObject)(configSchema)) {
+            this._configRuntimeSchema = configSchema;
         }
     }
     get allEdges() {
@@ -287,8 +299,8 @@ class StateGraph extends graph_js_1.Graph {
             }
             let inputSpec = this._schemaDefinition;
             if (options?.input !== undefined) {
-                if ((0, state_js_1.isAnyZodObject)(options.input)) {
-                    inputSpec = (0, state_js_1.getChannelsFromZod)(options.input);
+                if ((0, types_1.isInteropZodObject)(options.input)) {
+                    inputSpec = this._metaRegistry.getChannelsForSchema(options.input);
                 }
                 else if (options.input.spec !== undefined) {
                     inputSpec = options.input.spec;
@@ -461,6 +473,16 @@ function _getChannels(schema) {
  * instance method.
  */
 class CompiledStateGraph extends graph_js_1.CompiledGraph {
+    constructor() {
+        super(...arguments);
+        /** @internal */
+        Object.defineProperty(this, "_metaRegistry", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: meta_js_1.schemaMetaRegistry
+        });
+    }
     attachNode(key, node) {
         let outputKeys;
         if (key === constants_js_1.START) {
@@ -643,28 +665,31 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             const apply = (schema) => {
                 if (schema == null)
                     return undefined;
-                return (0, state_js_1.applyZodPlugin)(schema, { reducer: true });
+                return this._metaRegistry.getExtendedChannelSchemas(schema, {
+                    withReducerSchema: true,
+                });
             };
-            if ((0, state_js_1.isAnyZodObject)(input))
+            if ((0, types_1.isInteropZodObject)(input))
                 return apply(input);
-            if (input === PartialStateSchema)
-                return apply(schema)?.partial();
+            if (input === PartialStateSchema) {
+                return (0, types_1.interopZodObjectPartial)(apply(schema));
+            }
             return undefined;
         })();
         if ((0, constants_js_1.isCommand)(input)) {
             const parsedInput = input;
             if (input.update && schema != null)
-                parsedInput.update = schema.parse(input.update);
+                parsedInput.update = (0, types_1.interopParse)(schema, input.update);
             return parsedInput;
         }
         if (schema != null)
-            return schema.parse(input);
+            return (0, types_1.interopParse)(schema, input);
         return input;
     }
     async _validateConfigurable(config) {
         const configSchema = this.builder._configRuntimeSchema;
-        if ((0, state_js_1.isAnyZodObject)(configSchema))
-            configSchema.parse(config);
+        if ((0, types_1.isInteropZodObject)(configSchema))
+            (0, types_1.interopParse)(configSchema, config);
         return config;
     }
 }
@@ -704,13 +729,13 @@ function isZodStateGraphArgsWithStateSchema(value) {
     if (typeof value !== "object" || value == null) {
         return false;
     }
-    if (!("state" in value) || !(0, state_js_1.isAnyZodObject)(value.state)) {
+    if (!("state" in value) || !(0, types_1.isInteropZodObject)(value.state)) {
         return false;
     }
-    if ("input" in value && !(0, state_js_1.isAnyZodObject)(value.input)) {
+    if ("input" in value && !(0, types_1.isInteropZodObject)(value.input)) {
         return false;
     }
-    if ("output" in value && !(0, state_js_1.isAnyZodObject)(value.output)) {
+    if ("output" in value && !(0, types_1.isInteropZodObject)(value.output)) {
         return false;
     }
     return true;

package/dist/graph/state.d.ts

@@ -1,5 +1,6 @@
 import { RunnableLike } from "@langchain/core/runnables";
 import { All, type BaseCache, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
+import { type InteropZodObject } from "@langchain/core/utils/types";
 import { BaseChannel } from "../channels/base.js";
 import { CompiledGraph, Graph, Branch, AddNodeOptions, NodeSpec } from "./graph.js";
 import { Command, END, START } from "../constants.js";
@@ -7,7 +8,7 @@ import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType }
 import type { CachePolicy, RetryPolicy } from "../pregel/utils/index.js";
 import { ManagedValueSpec } from "../managed/base.js";
 import type { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
-import { AnyZodObject, ZodToStateDefinition } from "./zod/state.js";
+import { type SchemaMetaRegistry, InteropZodToStateDefinition } from "./zod/meta.js";
 export type ChannelReducers<Channels extends object> = {
     [K in keyof Channels]: SingleReducer<Channels[K], any>;
 };
@@ -26,7 +27,7 @@ export type StateGraphNodeSpec<RunInput, RunOutput> = NodeSpec<RunInput, RunOutp
 export type StateGraphAddNodeOptions<Nodes extends string = string> = {
     retryPolicy?: RetryPolicy;
     cachePolicy?: CachePolicy | boolean;
-    input?: AnnotationRoot<any> | AnyZodObject;
+    input?: AnnotationRoot<any> | InteropZodObject;
 } & AddNodeOptions<Nodes>;
 export type StateGraphArgsWithStateSchema<SD extends StateDefinition, I extends StateDefinition, O extends StateDefinition> = {
     stateSchema: AnnotationRoot<SD>;
@@ -37,13 +38,13 @@ export type StateGraphArgsWithInputOutputSchemas<SD extends StateDefinition, O e
     input: AnnotationRoot<SD>;
     output: AnnotationRoot<O>;
 };
-type ZodStateGraphArgsWithStateSchema<SD extends AnyZodObject, I extends SDZod, O extends SDZod> = {
+type ZodStateGraphArgsWithStateSchema<SD extends InteropZodObject, I extends SDZod, O extends SDZod> = {
     state: SD;
     input?: I;
     output?: O;
 };
-type SDZod = StateDefinition | AnyZodObject;
-type ToStateDefinition<T> = T extends AnyZodObject ? ZodToStateDefinition<T> : T extends StateDefinition ? T : never;
+type SDZod = StateDefinition | InteropZodObject;
+type ToStateDefinition<T> = T extends InteropZodObject ? InteropZodToStateDefinition<T> : T extends StateDefinition ? T : never;
 type NodeAction<S, U, C extends SDZod> = RunnableLike<S, U extends object ? U & Record<string, any> : U, // eslint-disable-line @typescript-eslint/no-explicit-any
 LangGraphRunnableConfig<StateType<ToStateDefinition<C>>>>;
 declare const PartialStateSchema: unique symbol;
@@ -116,27 +117,29 @@ export declare class StateGraph<SD extends SDZod | unknown, S = SD extends SDZod
     /** @internal */
     _schemaDefinition: StateDefinition;
     /** @internal */
-    _schemaRuntimeDefinition: AnyZodObject | undefined;
+    _schemaRuntimeDefinition: InteropZodObject | undefined;
     /** @internal */
     _inputDefinition: I;
     /** @internal */
-    _inputRuntimeDefinition: AnyZodObject | PartialStateSchema | undefined;
+    _inputRuntimeDefinition: InteropZodObject | PartialStateSchema | undefined;
     /** @internal */
     _outputDefinition: O;
     /** @internal */
-    _outputRuntimeDefinition: AnyZodObject | undefined;
+    _outputRuntimeDefinition: InteropZodObject | undefined;
     /**
      * Map schemas to managed values
      * @internal
      */
     _schemaDefinitions: Map<any, any>;
+    /** @internal */
+    _metaRegistry: SchemaMetaRegistry;
     /** @internal Used only for typing. */
     _configSchema: ToStateDefinition<C> | undefined;
     /** @internal */
-    _configRuntimeSchema: AnyZodObject | undefined;
+    _configRuntimeSchema: InteropZodObject | undefined;
     constructor(fields: SD extends StateDefinition ? StateGraphArgsWithInputOutputSchemas<SD, ToStateDefinition<O>> : never, configSchema?: C | AnnotationRoot<ToStateDefinition<C>>);
     constructor(fields: SD extends StateDefinition ? SD | AnnotationRoot<SD> | StateGraphArgs<S> | StateGraphArgsWithStateSchema<SD, ToStateDefinition<I>, ToStateDefinition<O>> : StateGraphArgs<S>, configSchema?: C | AnnotationRoot<ToStateDefinition<C>>);
-    constructor(fields: SD extends AnyZodObject ? SD | ZodStateGraphArgsWithStateSchema<SD, I, O> : never, configSchema?: C | AnnotationRoot<ToStateDefinition<C>>);
+    constructor(fields: SD extends InteropZodObject ? SD | ZodStateGraphArgsWithStateSchema<SD, I, O> : never, configSchema?: C | AnnotationRoot<ToStateDefinition<C>>);
     get allEdges(): Set<[string, string]>;
     _addSchema(stateDefinition: SDZod): void;
     addNode<K extends string>(nodes: Record<K, NodeAction<S, U, C>> | [
@@ -168,6 +171,8 @@ export declare class StateGraph<SD extends SDZod | unknown, S = SD extends SDZod
  */
 export declare class CompiledStateGraph<S, U, N extends string = typeof START, I extends SDZod = StateDefinition, O extends SDZod = StateDefinition, C extends SDZod = StateDefinition> extends CompiledGraph<N, S, U, StateType<ToStateDefinition<C>>, UpdateType<ToStateDefinition<I>>, StateType<ToStateDefinition<O>>> {
     builder: StateGraph<unknown, S, U, N, I, O, C>;
+    /** @internal */
+    _metaRegistry: SchemaMetaRegistry;
     attachNode(key: typeof START, node?: never): void;
     attachNode(key: N, node: StateGraphNodeSpec<S, U>): void;
     attachEdge(starts: N | N[] | "__start__", end: N | "__end__"): void;
@@ -177,11 +182,13 @@ export declare class CompiledStateGraph<S, U, N extends string = typeof START, I
     protected _validateInput(input: UpdateType<ToStateDefinition<I>>): Promise<UpdateType<ToStateDefinition<I>>>;
     protected _validateConfigurable(config: Partial<LangGraphRunnableConfig["configurable"]>): Promise<LangGraphRunnableConfig["configurable"]>;
 }
-type TypedNodeAction<SD extends StateDefinition, Nodes extends string> = (state: StateType<SD>, config: LangGraphRunnableConfig) => UpdateType<SD> | Command<unknown, UpdateType<SD>, Nodes>;
-export declare function typedNode<SD extends SDZod, Nodes extends string>(_state: SD extends StateDefinition ? AnnotationRoot<SD> : never, _options?: {
+type TypedNodeAction<SD extends StateDefinition, Nodes extends string, C extends StateDefinition = StateDefinition> = RunnableLike<StateType<SD>, UpdateType<SD> | Command<unknown, UpdateType<SD>, Nodes>, LangGraphRunnableConfig<StateType<C>>>;
+export declare function typedNode<SD extends SDZod, Nodes extends string, C extends SDZod = StateDefinition>(_state: SD extends StateDefinition ? AnnotationRoot<SD> : never, _options?: {
     nodes?: Nodes[];
-}): (func: TypedNodeAction<ToStateDefinition<SD>, Nodes>, options?: StateGraphAddNodeOptions<Nodes>) => TypedNodeAction<ToStateDefinition<SD>, Nodes>;
-export declare function typedNode<SD extends SDZod, Nodes extends string>(_state: SD extends AnyZodObject ? SD : never, _options?: {
+    config?: C extends StateDefinition ? AnnotationRoot<C> : never;
+}): (func: TypedNodeAction<ToStateDefinition<SD>, Nodes, ToStateDefinition<C>>, options?: StateGraphAddNodeOptions<Nodes>) => TypedNodeAction<ToStateDefinition<SD>, Nodes, ToStateDefinition<C>>;
+export declare function typedNode<SD extends SDZod, Nodes extends string, C extends SDZod = StateDefinition>(_state: SD extends InteropZodObject ? SD : never, _options?: {
     nodes?: Nodes[];
-}): (func: TypedNodeAction<ToStateDefinition<SD>, Nodes>, options?: StateGraphAddNodeOptions<Nodes>) => TypedNodeAction<ToStateDefinition<SD>, Nodes>;
+    config?: C extends InteropZodObject ? C : never;
+}): (func: TypedNodeAction<ToStateDefinition<SD>, Nodes, ToStateDefinition<C>>, options?: StateGraphAddNodeOptions<Nodes>) => TypedNodeAction<ToStateDefinition<SD>, Nodes, ToStateDefinition<C>>;
 export {};

package/dist/graph/state.js

@@ -1,5 +1,6 @@
 /* eslint-disable @typescript-eslint/no-use-before-define */
 import { _coerceToRunnable, Runnable, } from "@langchain/core/runnables";
+import { interopParse, interopZodObjectPartial, isInteropZodObject, } from "@langchain/core/utils/types";
 import { isBaseChannel } from "../channels/base.js";
 import { CompiledGraph, Graph, Branch, } from "./graph.js";
 import { ChannelWrite, PASSTHROUGH, } from "../pregel/write.js";
@@ -12,8 +13,8 @@ import { InvalidUpdateError, ParentCommand } from "../errors.js";
 import { getChannel, } from "./annotation.js";
 import { isConfiguredManagedValue } from "../managed/base.js";
 import { isPregelLike } from "../pregel/utils/subgraph.js";
-import { getChannelsFromZod, applyZodPlugin, isAnyZodObject, } from "./zod/state.js";
 import { LastValueAfterFinish } from "../channels/last_value.js";
+import { schemaMetaRegistry, } from "./zod/meta.js";
 const ROOT = "__root__";
 const PartialStateSchema = Symbol.for("langgraph.state.partial");
 /**
@@ -146,6 +147,13 @@ export class StateGraph extends Graph {
             writable: true,
             value: new Map()
         });
+        /** @internal */
+        Object.defineProperty(this, "_metaRegistry", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: schemaMetaRegistry
+        });
         /** @internal Used only for typing. */
         Object.defineProperty(this, "_configSchema", {
             enumerable: true,
@@ -161,9 +169,13 @@ export class StateGraph extends Graph {
             value: void 0
         });
         if (isZodStateGraphArgsWithStateSchema(fields)) {
-            const stateDef = getChannelsFromZod(fields.state);
-            const inputDef = fields.input != null ? getChannelsFromZod(fields.input) : stateDef;
-            const outputDef = fields.output != null ? getChannelsFromZod(fields.output) : stateDef;
+            const stateDef = this._metaRegistry.getChannelsForSchema(fields.state);
+            const inputDef = fields.input != null
+                ? this._metaRegistry.getChannelsForSchema(fields.input)
+                : stateDef;
+            const outputDef = fields.output != null
+                ? this._metaRegistry.getChannelsForSchema(fields.output)
+                : stateDef;
             this._schemaDefinition = stateDef;
             this._schemaRuntimeDefinition = fields.state;
             this._inputDefinition = inputDef;
@@ -171,8 +183,8 @@ export class StateGraph extends Graph {
             this._outputDefinition = outputDef;
             this._outputRuntimeDefinition = fields.output ?? fields.state;
         }
-        else if (isAnyZodObject(fields)) {
-            const stateDef = getChannelsFromZod(fields);
+        else if (isInteropZodObject(fields)) {
+            const stateDef = this._metaRegistry.getChannelsForSchema(fields);
             this._schemaDefinition = stateDef;
             this._schemaRuntimeDefinition = fields;
             this._inputDefinition = stateDef;
@@ -208,8 +220,8 @@ export class StateGraph extends Graph {
         this._addSchema(this._schemaDefinition);
         this._addSchema(this._inputDefinition);
         this._addSchema(this._outputDefinition);
-        if (isAnyZodObject(configSchema)) {
-            this._configRuntimeSchema = configSchema.passthrough();
+        if (isInteropZodObject(configSchema)) {
+            this._configRuntimeSchema = configSchema;
         }
     }
     get allEdges() {
@@ -283,8 +295,8 @@ export class StateGraph extends Graph {
             }
             let inputSpec = this._schemaDefinition;
             if (options?.input !== undefined) {
-                if (isAnyZodObject(options.input)) {
-                    inputSpec = getChannelsFromZod(options.input);
+                if (isInteropZodObject(options.input)) {
+                    inputSpec = this._metaRegistry.getChannelsForSchema(options.input);
                 }
                 else if (options.input.spec !== undefined) {
                     inputSpec = options.input.spec;
@@ -456,6 +468,16 @@ function _getChannels(schema) {
  * instance method.
  */
 export class CompiledStateGraph extends CompiledGraph {
+    constructor() {
+        super(...arguments);
+        /** @internal */
+        Object.defineProperty(this, "_metaRegistry", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: schemaMetaRegistry
+        });
+    }
     attachNode(key, node) {
         let outputKeys;
         if (key === START) {
@@ -638,28 +660,31 @@ export class CompiledStateGraph extends CompiledGraph {
             const apply = (schema) => {
                 if (schema == null)
                     return undefined;
-                return applyZodPlugin(schema, { reducer: true });
+                return this._metaRegistry.getExtendedChannelSchemas(schema, {
+                    withReducerSchema: true,
+                });
             };
-            if (isAnyZodObject(input))
+            if (isInteropZodObject(input))
                 return apply(input);
-            if (input === PartialStateSchema)
-                return apply(schema)?.partial();
+            if (input === PartialStateSchema) {
+                return interopZodObjectPartial(apply(schema));
+            }
             return undefined;
         })();
         if (isCommand(input)) {
             const parsedInput = input;
             if (input.update && schema != null)
-                parsedInput.update = schema.parse(input.update);
+                parsedInput.update = interopParse(schema, input.update);
             return parsedInput;
         }
         if (schema != null)
-            return schema.parse(input);
+            return interopParse(schema, input);
         return input;
     }
     async _validateConfigurable(config) {
         const configSchema = this.builder._configRuntimeSchema;
-        if (isAnyZodObject(configSchema))
-            configSchema.parse(config);
+        if (isInteropZodObject(configSchema))
+            interopParse(configSchema, config);
         return config;
     }
 }
@@ -698,13 +723,13 @@ function isZodStateGraphArgsWithStateSchema(value) {
     if (typeof value !== "object" || value == null) {
         return false;
     }
-    if (!("state" in value) || !isAnyZodObject(value.state)) {
+    if (!("state" in value) || !isInteropZodObject(value.state)) {
         return false;
     }
-    if ("input" in value && !isAnyZodObject(value.input)) {
+    if ("input" in value && !isInteropZodObject(value.input)) {
         return false;
     }
-    if ("output" in value && !isAnyZodObject(value.output)) {
+    if ("output" in value && !isInteropZodObject(value.output)) {
         return false;
     }
     return true;