@langchain/langgraph 0.3.0 → 0.3.2

package/dist/graph/zod/plugin.cjs

@@ -1,39 +1,42 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const zod_1 = require("zod");
-const state_js_1 = require("./state.cjs");
+const v3_1 = require("zod/v3");
+const types_1 = require("@langchain/core/utils/types");
+const meta_js_1 = require("./meta.cjs");
 const metaSymbol = Symbol.for("langgraph-zod");
 if (!(metaSymbol in globalThis)) {
     globalThis[metaSymbol] = new WeakSet();
 }
-try {
+function applyPluginPrototype(prototype) {
     const cache = globalThis[metaSymbol];
-    if (!cache?.has(zod_1.z.ZodType.prototype)) {
-        Object.defineProperty(zod_1.z.ZodType.prototype, "langgraph", {
-            get() {
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                const zodThis = this;
-                return {
-                    metadata(jsonSchemaExtra) {
-                        (0, state_js_1.extendMeta)(zodThis, (meta) => ({ ...meta, jsonSchemaExtra }));
-                        return zodThis;
-                    },
-                    reducer(fn, schema) {
-                        const defaultFn = (0, state_js_1.isZodDefault)(zodThis)
-                            ? // @ts-expect-error Due to `_def` being `any`
-                                zodThis._def.defaultValue
-                            : undefined;
-                        (0, state_js_1.extendMeta)(zodThis, (meta) => ({
-                            ...meta,
-                            default: defaultFn ?? meta?.default,
-                            reducer: { schema, fn },
-                        }));
-                        return zodThis;
-                    },
-                };
-            },
-        });
+    if (cache.has(prototype)) {
+        return; // Already applied
     }
+    Object.defineProperty(prototype, "langgraph", {
+        get() {
+            // Return type is any, actual type provided by module augmentation
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const zodThis = this;
+            return {
+                metadata(jsonSchemaExtra) {
+                    return (0, meta_js_1.withLangGraph)(zodThis, { jsonSchemaExtra });
+                },
+                reducer(fn, schema) {
+                    const defaultFn = (0, types_1.getInteropZodDefaultGetter)(zodThis);
+                    return (0, meta_js_1.withLangGraph)(zodThis, {
+                        default: defaultFn,
+                        reducer: { schema, fn },
+                    });
+                },
+            };
+        },
+    });
+    cache.add(prototype);
+}
+try {
+    applyPluginPrototype(v3_1.z.ZodType.prototype);
+    applyPluginPrototype(zod_1.z.ZodType.prototype);
 }
 catch (error) {
     throw new Error("Failed to extend Zod with LangGraph-related methods. This is most likely a bug, consider opening an issue and/or using `withLangGraph` to augment your Zod schema.", { cause: error });

package/dist/graph/zod/plugin.d.ts

@@ -1,6 +1,6 @@
-import { z } from "zod";
-interface ZodLangGraphTypes<T extends z.ZodTypeAny, Output> {
-    reducer<Input = z.output<T>>(transform: (a: Output, arg: Input) => Output, options?: z.ZodType<Input>): z.ZodType<Output, z.ZodEffectsDef<T>, Input>;
+import { z as zd } from "zod";
+interface ZodLangGraphTypes<T extends zd.ZodTypeAny, Output> {
+    reducer<Input = zd.output<T>>(transform: (a: Output, arg: Input) => Output, options?: zd.ZodType<Input>): zd.ZodType<Output, zd.ZodEffectsDef<T>, Input>;
     metadata(payload: {
         langgraph_nodes?: string[];
         langgraph_type?: "prompt";
@@ -9,6 +9,19 @@ interface ZodLangGraphTypes<T extends z.ZodTypeAny, Output> {
 }
 declare module "zod" {
     interface ZodType<Output> {
+        /**
+         * @deprecated Using the langgraph zod plugin is deprecated and will be removed in future versions
+         * Consider upgrading to zod 4 and using the exported langgraph meta registry. {@link langgraphRegistry}
+         */
+        langgraph: ZodLangGraphTypes<this, Output>;
+    }
+}
+declare module "zod/v3" {
+    interface ZodType<Output> {
+        /**
+         * @deprecated Using the langgraph zod plugin is deprecated and will be removed in future versions
+         * Consider upgrading to zod 4 and using the exported langgraph meta registry. {@link langgraphRegistry}
+         */
         langgraph: ZodLangGraphTypes<this, Output>;
     }
 }

package/dist/graph/zod/plugin.js

@@ -1,37 +1,40 @@
-import { z } from "zod";
-import { extendMeta, isZodDefault } from "./state.js";
+import { z as zd } from "zod";
+import { z as z3 } from "zod/v3";
+import { getInteropZodDefaultGetter } from "@langchain/core/utils/types";
+import { withLangGraph } from "./meta.js";
 const metaSymbol = Symbol.for("langgraph-zod");
 if (!(metaSymbol in globalThis)) {
     globalThis[metaSymbol] = new WeakSet();
 }
-try {
+function applyPluginPrototype(prototype) {
     const cache = globalThis[metaSymbol];
-    if (!cache?.has(z.ZodType.prototype)) {
-        Object.defineProperty(z.ZodType.prototype, "langgraph", {
-            get() {
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                const zodThis = this;
-                return {
-                    metadata(jsonSchemaExtra) {
-                        extendMeta(zodThis, (meta) => ({ ...meta, jsonSchemaExtra }));
-                        return zodThis;
-                    },
-                    reducer(fn, schema) {
-                        const defaultFn = isZodDefault(zodThis)
-                            ? // @ts-expect-error Due to `_def` being `any`
-                                zodThis._def.defaultValue
-                            : undefined;
-                        extendMeta(zodThis, (meta) => ({
-                            ...meta,
-                            default: defaultFn ?? meta?.default,
-                            reducer: { schema, fn },
-                        }));
-                        return zodThis;
-                    },
-                };
-            },
-        });
+    if (cache.has(prototype)) {
+        return; // Already applied
     }
+    Object.defineProperty(prototype, "langgraph", {
+        get() {
+            // Return type is any, actual type provided by module augmentation
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const zodThis = this;
+            return {
+                metadata(jsonSchemaExtra) {
+                    return withLangGraph(zodThis, { jsonSchemaExtra });
+                },
+                reducer(fn, schema) {
+                    const defaultFn = getInteropZodDefaultGetter(zodThis);
+                    return withLangGraph(zodThis, {
+                        default: defaultFn,
+                        reducer: { schema, fn },
+                    });
+                },
+            };
+        },
+    });
+    cache.add(prototype);
+}
+try {
+    applyPluginPrototype(z3.ZodType.prototype);
+    applyPluginPrototype(zd.ZodType.prototype);
 }
 catch (error) {
     throw new Error("Failed to extend Zod with LangGraph-related methods. This is most likely a bug, consider opening an issue and/or using `withLangGraph` to augment your Zod schema.", { cause: error });

package/dist/graph/zod/plugin.js.map

@@ -1 +1 @@
-{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../../src/graph/zod/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,YAAY,EAAa,MAAM,YAAY,CAAC;AAEjE,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;AA0B/C,IAAI,CAAC,CAAC,UAAU,IAAI,UAAU,CAAC,EAAE,CAAC;IAC/B,UAA+B,CAAC,UAAU,CAAC,GAAG,IAAI,OAAO,EAAE,CAAC;AAC/D,CAAC;AAED,IAAI,CAAC;IACH,MAAM,KAAK,GAAI,UAA+B,CAAC,UAAU,CAAC,CAAC;IAE3D,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,MAAM,CAAC,cAAc,CACnB,CAAC,CAAC,OAAO,CAAC,SAAS,EACnB,WAAqC,EACrC;YACE,GAAG;gBACD,8DAA8D;gBAC9D,MAAM,OAAO,GAAG,IAAiD,CAAC;gBAGlE,OAAO;oBACL,QAAQ,CAAC,eAAwD;wBAC/D,UAAU,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;wBAC9D,OAAO,OAAO,CAAC;oBACjB,CAAC;oBACD,OAAO,CACL,EAAqC,EACrC,MAAyB;wBAEzB,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;4BACrC,CAAC,CAAC,6CAA6C;gCAC7C,OAAO,CAAC,IAAI,CAAC,YAAY;4BAC3B,CAAC,CAAC,SAAS,CAAC;wBAEd,UAAU,CAAgB,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;4BAC5C,GAAG,IAAI;4BACP,OAAO,EAAE,SAAS,IAAI,IAAI,EAAE,OAAO;4BACnC,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;yBACxB,CAAC,CAAC,CAAC;wBAEJ,OAAO,OAAO,CAAC;oBACjB,CAAC;iBACF,CAAC;YACJ,CAAC;SACF,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IACf,MAAM,IAAI,KAAK,CACb,oKAAoK,EACpK,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;AACJ,CAAC"}
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../../src/graph/zod/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,MAAM,KAAK,CAAC;AAC9B,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAc,aAAa,EAAE,MAAM,WAAW,CAAC;AAEtD,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;AAwC/C,IAAI,CAAC,CAAC,UAAU,IAAI,UAAU,CAAC,EAAE,CAAC;IAC/B,UAA+B,CAAC,UAAU,CAAC,GAAG,IAAI,OAAO,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,oBAAoB,CAI3B,SAA4B;IAC5B,MAAM,KAAK,GAAI,UAA+B,CAAC,UAAU,CAAE,CAAC;IAC5D,IAAI,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,kBAAkB;IAC5B,CAAC;IAED,MAAM,CAAC,cAAc,CACnB,SAAS,EACT,WAA6C,EAC7C;QACE,GAAG;YACD,kEAAkE;YAClE,8DAA8D;YAC9D,MAAM,OAAO,GAAG,IAAqB,CAAC;YAGtC,OAAO;gBACL,QAAQ,CACN,eAA8D;oBAE9D,OAAO,aAAa,CAAC,OAAO,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC;gBACrD,CAAC;gBACD,OAAO,CACL,EAAqC,EACrC,MAA0B;oBAE1B,MAAM,SAAS,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;oBACtD,OAAO,aAAa,CAAC,OAAO,EAAE;wBAC5B,OAAO,EAAE,SAAS;wBAClB,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;qBACxB,CAAC,CAAC;gBACL,CAAC;aACF,CAAC;QACJ,CAAC;KACF,CACF,CAAC;IACF,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AACvB,CAAC;AAED,IAAI,CAAC;IACH,oBAAoB,CAAC,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3C,oBAAoB,CAAC,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IACf,MAAM,IAAI,KAAK,CACb,oKAAoK,EACpK,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;AACJ,CAAC"}

package/dist/graph/zod/schema.cjs

@@ -5,8 +5,8 @@ exports.getUpdateTypeSchema = getUpdateTypeSchema;
 exports.getInputTypeSchema = getInputTypeSchema;
 exports.getOutputTypeSchema = getOutputTypeSchema;
 exports.getConfigTypeSchema = getConfigTypeSchema;
-const zod_to_json_schema_1 = require("zod-to-json-schema");
-const state_js_1 = require("./state.cjs");
+const json_schema_1 = require("@langchain/core/utils/json_schema");
+const meta_js_1 = require("./meta.cjs");
 const PartialStateSchema = Symbol.for("langgraph.state.partial");
 function isGraphWithZodLike(graph) {
     if (!graph || typeof graph !== "object")
@@ -18,37 +18,59 @@ function isGraphWithZodLike(graph) {
     }
     return true;
 }
+function applyJsonSchemaExtrasFromDescription(schema) {
+    if (Array.isArray(schema)) {
+        return schema.map(applyJsonSchemaExtrasFromDescription);
+    }
+    if (typeof schema === "object" && schema != null) {
+        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
+            key,
+            applyJsonSchemaExtrasFromDescription(value),
+        ]));
+        if ("description" in output &&
+            typeof output.description === "string" &&
+            output.description.startsWith(meta_js_1.META_EXTRAS_DESCRIPTION_PREFIX)) {
+            const strMeta = output.description.slice(meta_js_1.META_EXTRAS_DESCRIPTION_PREFIX.length);
+            delete output.description;
+            Object.assign(output, JSON.parse(strMeta));
+        }
+        return output;
+    }
+    return schema;
+}
 function toJsonSchema(schema) {
-    return (0, state_js_1.applyExtraFromDescription)((0, zod_to_json_schema_1.zodToJsonSchema)(schema));
+    return applyJsonSchemaExtrasFromDescription((0, json_schema_1.toJsonSchema)(schema));
 }
 /**
  * Get the state schema for a graph.
  * @param graph - The graph to get the state schema for.
  * @returns The state schema for the graph.
  */
-function getStateTypeSchema(graph) {
+function getStateTypeSchema(graph, registry = meta_js_1.schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 /**
  * Get the update schema for a graph.
  * @param graph - The graph to get the update schema for.
  * @returns The update schema for the graph.
  */
-function getUpdateTypeSchema(graph) {
+function getUpdateTypeSchema(graph, registry = meta_js_1.schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, {
-        reducer: true,
-        jsonSchemaExtra: true,
-        partial: true,
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withReducerSchema: true,
+        withJsonSchemaExtrasAsDescription: true,
+        asPartial: true,
     }));
 }
 /**
@@ -56,7 +78,7 @@ function getUpdateTypeSchema(graph) {
  * @param graph - The graph to get the input schema for.
  * @returns The input schema for the graph.
  */
-function getInputTypeSchema(graph) {
+function getInputTypeSchema(graph, registry = meta_js_1.schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     let schemaDef = graph.builder._inputRuntimeDefinition;
@@ -66,10 +88,10 @@ function getInputTypeSchema(graph) {
     }
     if (!schemaDef)
         return undefined;
-    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, {
-        reducer: true,
-        jsonSchemaExtra: true,
-        partial: true,
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withReducerSchema: true,
+        withJsonSchemaExtrasAsDescription: true,
+        asPartial: true,
     }));
 }
 /**
@@ -77,25 +99,29 @@ function getInputTypeSchema(graph) {
  * @param graph - The graph to get the output schema for.
  * @returns The output schema for the graph.
  */
-function getOutputTypeSchema(graph) {
+function getOutputTypeSchema(graph, registry = meta_js_1.schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._outputRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 /**
  * Get the config schema for a graph.
  * @param graph - The graph to get the config schema for.
  * @returns The config schema for the graph.
  */
-function getConfigTypeSchema(graph) {
+function getConfigTypeSchema(graph, registry = meta_js_1.schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const configDef = graph.builder._configRuntimeSchema;
     if (!configDef)
         return undefined;
-    return toJsonSchema((0, state_js_1.applyZodPlugin)(configDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(configDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 //# sourceMappingURL=schema.js.map

package/dist/graph/zod/schema.d.ts

@@ -1,33 +1,32 @@
-import { zodToJsonSchema as _zodToJsonSchema } from "zod-to-json-schema";
-type JsonSchema = ReturnType<typeof _zodToJsonSchema>;
+import { type JSONSchema } from "@langchain/core/utils/json_schema";
+import { SchemaMetaRegistry } from "./meta.js";
 /**
  * Get the state schema for a graph.
  * @param graph - The graph to get the state schema for.
  * @returns The state schema for the graph.
  */
-export declare function getStateTypeSchema(graph: unknown): JsonSchema | undefined;
+export declare function getStateTypeSchema(graph: unknown, registry?: SchemaMetaRegistry): JSONSchema | undefined;
 /**
  * Get the update schema for a graph.
  * @param graph - The graph to get the update schema for.
  * @returns The update schema for the graph.
  */
-export declare function getUpdateTypeSchema(graph: unknown): JsonSchema | undefined;
+export declare function getUpdateTypeSchema(graph: unknown, registry?: SchemaMetaRegistry): JSONSchema | undefined;
 /**
  * Get the input schema for a graph.
  * @param graph - The graph to get the input schema for.
  * @returns The input schema for the graph.
  */
-export declare function getInputTypeSchema(graph: unknown): JsonSchema | undefined;
+export declare function getInputTypeSchema(graph: unknown, registry?: SchemaMetaRegistry): JSONSchema | undefined;
 /**
  * Get the output schema for a graph.
  * @param graph - The graph to get the output schema for.
  * @returns The output schema for the graph.
  */
-export declare function getOutputTypeSchema(graph: unknown): JsonSchema | undefined;
+export declare function getOutputTypeSchema(graph: unknown, registry?: SchemaMetaRegistry): JSONSchema | undefined;
 /**
  * Get the config schema for a graph.
  * @param graph - The graph to get the config schema for.
  * @returns The config schema for the graph.
  */
-export declare function getConfigTypeSchema(graph: unknown): JsonSchema | undefined;
-export {};
+export declare function getConfigTypeSchema(graph: unknown, registry?: SchemaMetaRegistry): JSONSchema | undefined;

package/dist/graph/zod/schema.js

@@ -1,5 +1,5 @@
-import { zodToJsonSchema as _zodToJsonSchema } from "zod-to-json-schema";
-import { applyZodPlugin, applyExtraFromDescription } from "./state.js";
+import { toJsonSchema as interopToJsonSchema, } from "@langchain/core/utils/json_schema";
+import { META_EXTRAS_DESCRIPTION_PREFIX, schemaMetaRegistry, } from "./meta.js";
 const PartialStateSchema = Symbol.for("langgraph.state.partial");
 function isGraphWithZodLike(graph) {
     if (!graph || typeof graph !== "object")
@@ -11,37 +11,59 @@ function isGraphWithZodLike(graph) {
     }
     return true;
 }
+function applyJsonSchemaExtrasFromDescription(schema) {
+    if (Array.isArray(schema)) {
+        return schema.map(applyJsonSchemaExtrasFromDescription);
+    }
+    if (typeof schema === "object" && schema != null) {
+        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
+            key,
+            applyJsonSchemaExtrasFromDescription(value),
+        ]));
+        if ("description" in output &&
+            typeof output.description === "string" &&
+            output.description.startsWith(META_EXTRAS_DESCRIPTION_PREFIX)) {
+            const strMeta = output.description.slice(META_EXTRAS_DESCRIPTION_PREFIX.length);
+            delete output.description;
+            Object.assign(output, JSON.parse(strMeta));
+        }
+        return output;
+    }
+    return schema;
+}
 function toJsonSchema(schema) {
-    return applyExtraFromDescription(_zodToJsonSchema(schema));
+    return applyJsonSchemaExtrasFromDescription(interopToJsonSchema(schema));
 }
 /**
  * Get the state schema for a graph.
  * @param graph - The graph to get the state schema for.
  * @returns The state schema for the graph.
  */
-export function getStateTypeSchema(graph) {
+export function getStateTypeSchema(graph, registry = schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyZodPlugin(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 /**
  * Get the update schema for a graph.
  * @param graph - The graph to get the update schema for.
  * @returns The update schema for the graph.
  */
-export function getUpdateTypeSchema(graph) {
+export function getUpdateTypeSchema(graph, registry = schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyZodPlugin(schemaDef, {
-        reducer: true,
-        jsonSchemaExtra: true,
-        partial: true,
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withReducerSchema: true,
+        withJsonSchemaExtrasAsDescription: true,
+        asPartial: true,
     }));
 }
 /**
@@ -49,7 +71,7 @@ export function getUpdateTypeSchema(graph) {
  * @param graph - The graph to get the input schema for.
  * @returns The input schema for the graph.
  */
-export function getInputTypeSchema(graph) {
+export function getInputTypeSchema(graph, registry = schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     let schemaDef = graph.builder._inputRuntimeDefinition;
@@ -59,10 +81,10 @@ export function getInputTypeSchema(graph) {
     }
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyZodPlugin(schemaDef, {
-        reducer: true,
-        jsonSchemaExtra: true,
-        partial: true,
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withReducerSchema: true,
+        withJsonSchemaExtrasAsDescription: true,
+        asPartial: true,
     }));
 }
 /**
@@ -70,25 +92,29 @@ export function getInputTypeSchema(graph) {
  * @param graph - The graph to get the output schema for.
  * @returns The output schema for the graph.
  */
-export function getOutputTypeSchema(graph) {
+export function getOutputTypeSchema(graph, registry = schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const schemaDef = graph.builder._outputRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyZodPlugin(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(schemaDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 /**
  * Get the config schema for a graph.
  * @param graph - The graph to get the config schema for.
  * @returns The config schema for the graph.
  */
-export function getConfigTypeSchema(graph) {
+export function getConfigTypeSchema(graph, registry = schemaMetaRegistry) {
     if (!isGraphWithZodLike(graph))
         return undefined;
     const configDef = graph.builder._configRuntimeSchema;
     if (!configDef)
         return undefined;
-    return toJsonSchema(applyZodPlugin(configDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(registry.getExtendedChannelSchemas(configDef, {
+        withJsonSchemaExtrasAsDescription: true,
+    }));
 }
 //# sourceMappingURL=schema.js.map

package/dist/graph/zod/schema.js.map

@@ -1 +1 @@
-{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../../src/graph/zod/schema.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,IAAI,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AAEvE,MAAM,kBAAkB,GAAG,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;AAcjE,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,IACE,CAAC,CAAC,SAAS,IAAI,KAAK,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QACjC,KAAK,CAAC,OAAO,IAAI,IAAI,EACrB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,MAAiB;IACrC,OAAO,yBAAyB,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAe,CAAC;AAC3E,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,YAAY,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,cAAc,CAAC,SAAS,EAAE;QACxB,OAAO,EAAE,IAAI;QACb,eAAe,EAAE,IAAI;QACrB,OAAO,EAAE,IAAI;KACd,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,IAAI,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,uBAAuB,CAAC;IACtD,IAAI,SAAS,KAAK,kBAAkB,EAAE,CAAC;QACrC,wEAAwE;QACxE,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACrD,CAAC;IAED,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,YAAY,CACjB,cAAc,CAAC,SAAS,EAAE;QACxB,OAAO,EAAE,IAAI;QACb,eAAe,EAAE,IAAI;QACrB,OAAO,EAAE,IAAI;KACd,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,YAAY,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,oBAAoB,CAAC;IACrD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,YAAY,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC5E,CAAC"}
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../../src/graph/zod/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,YAAY,IAAI,mBAAmB,GACpC,MAAM,mCAAmC,CAAC;AAE3C,OAAO,EACL,8BAA8B,EAE9B,kBAAkB,GACnB,MAAM,WAAW,CAAC;AAEnB,MAAM,kBAAkB,GAAG,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;AAYjE,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,IACE,CAAC,CAAC,SAAS,IAAI,KAAK,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QACjC,KAAK,CAAC,OAAO,IAAI,IAAI,EACrB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,oCAAoC,CAAI,MAAS;IACxD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAC/B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;YAC3C,GAAG;YACH,oCAAoC,CAAC,KAAK,CAAC;SAC5C,CAAC,CACH,CAAC;QAEF,IACE,aAAa,IAAI,MAAM;YACvB,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;YACtC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,8BAA8B,CAAC,EAC7D,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CACtC,8BAA8B,CAAC,MAAM,CACtC,CAAC;YACF,OAAO,MAAM,CAAC,WAAW,CAAC;YAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,MAAW,CAAC;IACrB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,MAAwB;IAC5C,OAAO,oCAAoC,CACzC,mBAAmB,CAAC,MAAM,CAAC,CACd,CAAC;AAClB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,KAAc,EACd,WAA+B,kBAAkB;IAEjD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,QAAQ,CAAC,yBAAyB,CAAC,SAAS,EAAE;QAC5C,iCAAiC,EAAE,IAAI;KACxC,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAc,EACd,WAA+B,kBAAkB;IAEjD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,QAAQ,CAAC,yBAAyB,CAAC,SAAS,EAAE;QAC5C,iBAAiB,EAAE,IAAI;QACvB,iCAAiC,EAAE,IAAI;QACvC,SAAS,EAAE,IAAI;KAChB,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,KAAc,EACd,WAA+B,kBAAkB;IAEjD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,IAAI,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,uBAAuB,CAAC;IACtD,IAAI,SAAS,KAAK,kBAAkB,EAAE,CAAC;QACrC,wEAAwE;QACxE,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACrD,CAAC;IACD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,QAAQ,CAAC,yBAAyB,CAAC,SAAS,EAAE;QAC5C,iBAAiB,EAAE,IAAI;QACvB,iCAAiC,EAAE,IAAI;QACvC,SAAS,EAAE,IAAI;KAChB,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAc,EACd,WAA+B,kBAAkB;IAEjD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC;IACzD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,QAAQ,CAAC,yBAAyB,CAAC,SAAS,EAAE;QAC5C,iCAAiC,EAAE,IAAI;KACxC,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAc,EACd,WAA+B,kBAAkB;IAEjD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,oBAAoB,CAAC;IACrD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,OAAO,YAAY,CACjB,QAAQ,CAAC,yBAAyB,CAAC,SAAS,EAAE;QAC5C,iCAAiC,EAAE,IAAI;KACxC,CAAC,CACH,CAAC;AACJ,CAAC"}

package/dist/graph/zod/zod-registry.cjs

@@ -0,0 +1,49 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registry = exports.LanggraphZodMetaRegistry = void 0;
+const types_1 = require("@langchain/core/utils/types");
+const core_1 = require("zod/v4/core");
+const meta_js_1 = require("./meta.cjs");
+/**
+ * A Zod v4-compatible meta registry that extends the base registry.
+ *
+ * This registry allows you to associate and retrieve metadata for Zod schemas,
+ * leveraging the base registry for storage. It is compatible with Zod v4 and
+ * interoperates with the base registry to ensure consistent metadata management
+ * across different Zod versions.
+ *
+ * @template Meta - The type of metadata associated with each schema.
+ * @template Schema - The Zod schema type.
+ */
+class LanggraphZodMetaRegistry extends core_1.$ZodRegistry {
+    /**
+     * Creates a new LanggraphZodMetaRegistry instance.
+     *
+     * @param parent - The base SchemaMetaRegistry to use for metadata storage.
+     */
+    constructor(parent) {
+        super();
+        Object.defineProperty(this, "parent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: parent
+        });
+        // Use the parent's map for metadata storage
+        this._map = this.parent._map;
+    }
+    add(schema, ..._meta) {
+        const firstMeta = _meta[0];
+        if (firstMeta && !firstMeta?.default) {
+            const defaultValueGetter = (0, types_1.getInteropZodDefaultGetter)(schema);
+            if (defaultValueGetter != null) {
+                // eslint-disable-next-line no-param-reassign
+                firstMeta.default = defaultValueGetter;
+            }
+        }
+        return super.add(schema, ..._meta);
+    }
+}
+exports.LanggraphZodMetaRegistry = LanggraphZodMetaRegistry;
+exports.registry = new LanggraphZodMetaRegistry(meta_js_1.schemaMetaRegistry);
+//# sourceMappingURL=zod-registry.js.map

package/dist/graph/zod/zod-registry.d.ts

@@ -0,0 +1,24 @@
+import { $ZodType, $ZodRegistry, $replace } from "zod/v4/core";
+import { SchemaMeta, SchemaMetaRegistry } from "./meta.js";
+/**
+ * A Zod v4-compatible meta registry that extends the base registry.
+ *
+ * This registry allows you to associate and retrieve metadata for Zod schemas,
+ * leveraging the base registry for storage. It is compatible with Zod v4 and
+ * interoperates with the base registry to ensure consistent metadata management
+ * across different Zod versions.
+ *
+ * @template Meta - The type of metadata associated with each schema.
+ * @template Schema - The Zod schema type.
+ */
+export declare class LanggraphZodMetaRegistry<Meta extends SchemaMeta = SchemaMeta, Schema extends $ZodType = $ZodType> extends $ZodRegistry<Meta, Schema> {
+    protected parent: SchemaMetaRegistry;
+    /**
+     * Creates a new LanggraphZodMetaRegistry instance.
+     *
+     * @param parent - The base SchemaMetaRegistry to use for metadata storage.
+     */
+    constructor(parent: SchemaMetaRegistry);
+    add<S extends Schema>(schema: S, ..._meta: undefined extends Meta ? [$replace<Meta, S>?] : [$replace<Meta, S>]): this;
+}
+export declare const registry: LanggraphZodMetaRegistry<SchemaMeta<any, any>, $ZodType<unknown, unknown, import("zod/v4/core").$ZodTypeInternals<unknown, unknown>>>;

package/dist/graph/zod/zod-registry.js

@@ -0,0 +1,45 @@
+import { getInteropZodDefaultGetter } from "@langchain/core/utils/types";
+import { $ZodRegistry } from "zod/v4/core";
+import { schemaMetaRegistry } from "./meta.js";
+/**
+ * A Zod v4-compatible meta registry that extends the base registry.
+ *
+ * This registry allows you to associate and retrieve metadata for Zod schemas,
+ * leveraging the base registry for storage. It is compatible with Zod v4 and
+ * interoperates with the base registry to ensure consistent metadata management
+ * across different Zod versions.
+ *
+ * @template Meta - The type of metadata associated with each schema.
+ * @template Schema - The Zod schema type.
+ */
+export class LanggraphZodMetaRegistry extends $ZodRegistry {
+    /**
+     * Creates a new LanggraphZodMetaRegistry instance.
+     *
+     * @param parent - The base SchemaMetaRegistry to use for metadata storage.
+     */
+    constructor(parent) {
+        super();
+        Object.defineProperty(this, "parent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: parent
+        });
+        // Use the parent's map for metadata storage
+        this._map = this.parent._map;
+    }
+    add(schema, ..._meta) {
+        const firstMeta = _meta[0];
+        if (firstMeta && !firstMeta?.default) {
+            const defaultValueGetter = getInteropZodDefaultGetter(schema);
+            if (defaultValueGetter != null) {
+                // eslint-disable-next-line no-param-reassign
+                firstMeta.default = defaultValueGetter;
+            }
+        }
+        return super.add(schema, ..._meta);
+    }
+}
+export const registry = new LanggraphZodMetaRegistry(schemaMetaRegistry);
+//# sourceMappingURL=zod-registry.js.map

package/dist/graph/zod/zod-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zod-registry.js","sourceRoot":"","sources":["../../../src/graph/zod/zod-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAY,YAAY,EAAY,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAkC,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAE/E;;;;;;;;;;GAUG;AACH,MAAM,OAAO,wBAGX,SAAQ,YAA0B;IAClC;;;;OAIG;IACH,YAAsB,MAA0B;QAC9C,KAAK,EAAE,CAAC;QADE;;;;mBAAU,MAAM;WAAoB;QAE9C,4CAA4C;QAC5C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAA+C,CAAC;IAC1E,CAAC;IAED,GAAG,CACD,MAAS,EACT,GAAG,KAEoB;QAEvB,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,SAAS,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC;YACrC,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;YAC9D,IAAI,kBAAkB,IAAI,IAAI,EAAE,CAAC;gBAC/B,6CAA6C;gBAC7C,SAAS,CAAC,OAAO,GAAG,kBAAkB,CAAC;YACzC,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAAC,kBAAkB,CAAC,CAAC"}