@langchain/langgraph 0.3.1 → 0.3.3

package/dist/graph/zod/meta.cjs

@@ -0,0 +1,180 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.schemaMetaRegistry = exports.SchemaMetaRegistry = exports.META_EXTRAS_DESCRIPTION_PREFIX = void 0;
+exports.withLangGraph = withLangGraph;
+const types_1 = require("@langchain/core/utils/types");
+const binop_js_1 = require("../../channels/binop.cjs");
+const last_value_js_1 = require("../../channels/last_value.cjs");
+exports.META_EXTRAS_DESCRIPTION_PREFIX = "lg:";
+/**
+ * A registry for storing and managing metadata associated with schemas.
+ * This class provides methods to get, extend, remove, and check metadata for a given schema.
+ */
+class SchemaMetaRegistry {
+    constructor() {
+        /**
+         * Internal map storing schema metadata.
+         * @internal
+         */
+        Object.defineProperty(this, "_map", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new WeakMap()
+        });
+        /**
+         * Cache for extended schfemas.
+         * @internal
+         */
+        Object.defineProperty(this, "_extensionCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
+    }
+    /**
+     * Retrieves the metadata associated with a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to retrieve metadata for.
+     * @returns The associated SchemaMeta, or undefined if not present.
+     */
+    get(schema) {
+        return this._map.get(schema);
+    }
+    /**
+     * Extends or sets the metadata for a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to extend metadata for.
+     * @param predicate A function that receives the existing metadata (or undefined) and returns the new metadata.
+     */
+    extend(schema, predicate) {
+        const existingMeta = this.get(schema);
+        this._map.set(schema, predicate(existingMeta));
+    }
+    /**
+     * Removes the metadata associated with a given schema.
+     * @param schema The schema to remove metadata for.
+     * @returns The SchemaMetaRegistry instance (for chaining).
+     */
+    remove(schema) {
+        this._map.delete(schema);
+        return this;
+    }
+    /**
+     * Checks if metadata exists for a given schema.
+     * @param schema The schema to check.
+     * @returns True if metadata exists, false otherwise.
+     */
+    has(schema) {
+        return this._map.has(schema);
+    }
+    /**
+     * Returns a mapping of channel instances for each property in the schema
+     * using the associated metadata in the registry.
+     *
+     * This is used to create the `channels` object that's passed to the `Graph` constructor.
+     *
+     * @template T The shape of the schema.
+     * @param schema The schema to extract channels from.
+     * @returns A mapping from property names to channel instances.
+     */
+    getChannelsForSchema(schema) {
+        const channels = {};
+        const shape = (0, types_1.getInteropZodObjectShape)(schema);
+        for (const [key, channelSchema] of Object.entries(shape)) {
+            const meta = this.get(channelSchema);
+            if (meta?.reducer) {
+                channels[key] = new binop_js_1.BinaryOperatorAggregate(meta.reducer.fn, meta.default);
+            }
+            else {
+                channels[key] = new last_value_js_1.LastValue();
+            }
+        }
+        return channels;
+    }
+    /**
+     * Returns a modified schema that introspectively looks at all keys of the provided
+     * object schema, and applies the augmentations based on meta provided with those keys
+     * in the registry and the selectors provided in the `effects` parameter.
+     *
+     * This assumes that the passed in schema is the "root" schema object for a graph where
+     * the keys of the schema are the channels of the graph. Because we need to represent
+     * the input of a graph in a couple of different ways, the `effects` parameter allows
+     * us to apply those augmentations based on pre determined conditions.
+     *
+     * @param schema The root schema object to extend.
+     * @param effects The effects that are being applied.
+     * @returns The extended schema.
+     */
+    getExtendedChannelSchemas(schema, effects) {
+        // If no effects are being applied, return the schema unchanged
+        if (Object.keys(effects).length === 0) {
+            return schema;
+        }
+        // Cache key is determined by looking at the effects that are being applied
+        const cacheKey = Object.entries(effects)
+            .filter(([, v]) => v === true)
+            .sort(([a], [b]) => a.localeCompare(b))
+            .map(([k, v]) => `${k}:${v}`)
+            .join("|");
+        const cache = this._extensionCache.get(cacheKey) ?? new WeakMap();
+        if (cache.has(schema))
+            return cache.get(schema);
+        let modifiedSchema = schema;
+        if (effects.withReducerSchema ||
+            effects.withJsonSchemaExtrasAsDescription) {
+            const newShapeEntries = Object.entries((0, types_1.getInteropZodObjectShape)(schema)).map(([key, schema]) => {
+                const meta = this.get(schema);
+                let outputSchema = effects.withReducerSchema
+                    ? meta?.reducer?.schema ?? schema
+                    : schema;
+                if (effects.withJsonSchemaExtrasAsDescription &&
+                    meta?.jsonSchemaExtra) {
+                    const description = (0, types_1.getSchemaDescription)(outputSchema) ?? (0, types_1.getSchemaDescription)(schema);
+                    const strExtras = JSON.stringify({
+                        ...meta.jsonSchemaExtra,
+                        description,
+                    });
+                    outputSchema = outputSchema.describe(`${exports.META_EXTRAS_DESCRIPTION_PREFIX}${strExtras}`);
+                }
+                return [key, outputSchema];
+            });
+            modifiedSchema = (0, types_1.extendInteropZodObject)(schema, Object.fromEntries(newShapeEntries));
+            if ((0, types_1.isZodSchemaV3)(modifiedSchema)) {
+                modifiedSchema._def.unknownKeys = "strip";
+            }
+        }
+        if (effects.asPartial) {
+            modifiedSchema = (0, types_1.interopZodObjectPartial)(modifiedSchema);
+        }
+        cache.set(schema, modifiedSchema);
+        this._extensionCache.set(cacheKey, cache);
+        return modifiedSchema;
+    }
+}
+exports.SchemaMetaRegistry = SchemaMetaRegistry;
+exports.schemaMetaRegistry = new SchemaMetaRegistry();
+function withLangGraph(schema, meta) {
+    if (meta.reducer && !meta.default) {
+        const defaultValueGetter = (0, types_1.getInteropZodDefaultGetter)(schema);
+        if (defaultValueGetter != null) {
+            // eslint-disable-next-line no-param-reassign
+            meta.default = defaultValueGetter;
+        }
+    }
+    if (meta.reducer) {
+        const schemaWithReducer = Object.assign(schema, {
+            lg_reducer_schema: meta.reducer?.schema ?? schema,
+        });
+        exports.schemaMetaRegistry.extend(schemaWithReducer, () => meta);
+        return schemaWithReducer;
+    }
+    else {
+        exports.schemaMetaRegistry.extend(schema, () => meta);
+        return schema;
+    }
+}
+//# sourceMappingURL=meta.js.map

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

@@ -0,0 +1,110 @@
+import { InteropZodObject, InteropZodType, InteropZodObjectShape } from "@langchain/core/utils/types";
+import { BaseChannel } from "../../channels/base.js";
+export declare const META_EXTRAS_DESCRIPTION_PREFIX = "lg:";
+export type ReducedZodChannel<T extends InteropZodType, TReducerSchema extends InteropZodType> = T & {
+    lg_reducer_schema: TReducerSchema;
+};
+export type InteropZodToStateDefinition<T extends InteropZodObject, TShape = InteropZodObjectShape<T>> = {
+    [key in keyof TShape]: TShape[key] extends ReducedZodChannel<infer Schema, infer ReducerSchema> ? Schema extends InteropZodType<infer V> ? ReducerSchema extends InteropZodType<infer U> ? BaseChannel<V, U> : never : never : TShape[key] extends InteropZodType<infer V, infer U> ? BaseChannel<V, U> : never;
+};
+export interface SchemaMeta<TValue = any, TUpdate = TValue> {
+    jsonSchemaExtra?: {
+        langgraph_nodes?: string[];
+        langgraph_type?: "prompt" | "messages";
+        [key: string]: unknown;
+    };
+    reducer?: {
+        schema?: InteropZodType<TUpdate>;
+        fn: (a: TValue, b: TUpdate) => TValue;
+    };
+    default?: () => TValue;
+    [key: string]: unknown;
+}
+/**
+ * A registry for storing and managing metadata associated with schemas.
+ * This class provides methods to get, extend, remove, and check metadata for a given schema.
+ */
+export declare class SchemaMetaRegistry {
+    /**
+     * Internal map storing schema metadata.
+     * @internal
+     */
+    _map: WeakMap<InteropZodType, SchemaMeta<any, any>>;
+    /**
+     * Cache for extended schfemas.
+     * @internal
+     */
+    _extensionCache: Map<string, WeakMap<InteropZodType, InteropZodType>>;
+    /**
+     * Retrieves the metadata associated with a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to retrieve metadata for.
+     * @returns The associated SchemaMeta, or undefined if not present.
+     */
+    get<TValue, TUpdate = TValue>(schema: InteropZodType<TValue>): SchemaMeta<TValue, TUpdate> | undefined;
+    /**
+     * Extends or sets the metadata for a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to extend metadata for.
+     * @param predicate A function that receives the existing metadata (or undefined) and returns the new metadata.
+     */
+    extend<TValue, TUpdate>(schema: InteropZodType<TValue>, predicate: (meta: SchemaMeta<TValue, TUpdate> | undefined) => SchemaMeta<TValue, TUpdate>): void;
+    /**
+     * Removes the metadata associated with a given schema.
+     * @param schema The schema to remove metadata for.
+     * @returns The SchemaMetaRegistry instance (for chaining).
+     */
+    remove(schema: InteropZodType): this;
+    /**
+     * Checks if metadata exists for a given schema.
+     * @param schema The schema to check.
+     * @returns True if metadata exists, false otherwise.
+     */
+    has(schema: InteropZodType): boolean;
+    /**
+     * Returns a mapping of channel instances for each property in the schema
+     * using the associated metadata in the registry.
+     *
+     * This is used to create the `channels` object that's passed to the `Graph` constructor.
+     *
+     * @template T The shape of the schema.
+     * @param schema The schema to extract channels from.
+     * @returns A mapping from property names to channel instances.
+     */
+    getChannelsForSchema<T extends InteropZodObject>(schema: T): InteropZodToStateDefinition<T>;
+    /**
+     * Returns a modified schema that introspectively looks at all keys of the provided
+     * object schema, and applies the augmentations based on meta provided with those keys
+     * in the registry and the selectors provided in the `effects` parameter.
+     *
+     * This assumes that the passed in schema is the "root" schema object for a graph where
+     * the keys of the schema are the channels of the graph. Because we need to represent
+     * the input of a graph in a couple of different ways, the `effects` parameter allows
+     * us to apply those augmentations based on pre determined conditions.
+     *
+     * @param schema The root schema object to extend.
+     * @param effects The effects that are being applied.
+     * @returns The extended schema.
+     */
+    getExtendedChannelSchemas<T extends InteropZodObject>(schema: T, effects: {
+        /**
+         * Augments the shape by using the reducer's schema if it exists
+         */
+        withReducerSchema?: boolean;
+        /**
+         * Applies the stringified jsonSchemaExtra as a description to the schema.
+         */
+        withJsonSchemaExtrasAsDescription?: boolean;
+        /**
+         * Applies the `.partial()` modifier to the schema.
+         */
+        asPartial?: boolean;
+    }): InteropZodObject;
+}
+export declare const schemaMetaRegistry: SchemaMetaRegistry;
+export declare function withLangGraph<TValue, TUpdate, TSchema extends InteropZodType<TValue>>(schema: TSchema, meta: SchemaMeta<TValue, TUpdate> & {
+    reducer?: undefined;
+}): TSchema;
+export declare function withLangGraph<TValue, TUpdate, TSchema extends InteropZodType<TValue>>(schema: TSchema, meta: SchemaMeta<TValue, TUpdate>): ReducedZodChannel<TSchema, InteropZodType<TUpdate>>;

package/dist/graph/zod/meta.js

@@ -0,0 +1,175 @@
+import { getInteropZodObjectShape, extendInteropZodObject, getInteropZodDefaultGetter, interopZodObjectPartial, isZodSchemaV3, getSchemaDescription, } from "@langchain/core/utils/types";
+import { BinaryOperatorAggregate } from "../../channels/binop.js";
+import { LastValue } from "../../channels/last_value.js";
+export const META_EXTRAS_DESCRIPTION_PREFIX = "lg:";
+/**
+ * A registry for storing and managing metadata associated with schemas.
+ * This class provides methods to get, extend, remove, and check metadata for a given schema.
+ */
+export class SchemaMetaRegistry {
+    constructor() {
+        /**
+         * Internal map storing schema metadata.
+         * @internal
+         */
+        Object.defineProperty(this, "_map", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new WeakMap()
+        });
+        /**
+         * Cache for extended schfemas.
+         * @internal
+         */
+        Object.defineProperty(this, "_extensionCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
+    }
+    /**
+     * Retrieves the metadata associated with a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to retrieve metadata for.
+     * @returns The associated SchemaMeta, or undefined if not present.
+     */
+    get(schema) {
+        return this._map.get(schema);
+    }
+    /**
+     * Extends or sets the metadata for a given schema.
+     * @template TValue The value type of the schema.
+     * @template TUpdate The update type of the schema (defaults to TValue).
+     * @param schema The schema to extend metadata for.
+     * @param predicate A function that receives the existing metadata (or undefined) and returns the new metadata.
+     */
+    extend(schema, predicate) {
+        const existingMeta = this.get(schema);
+        this._map.set(schema, predicate(existingMeta));
+    }
+    /**
+     * Removes the metadata associated with a given schema.
+     * @param schema The schema to remove metadata for.
+     * @returns The SchemaMetaRegistry instance (for chaining).
+     */
+    remove(schema) {
+        this._map.delete(schema);
+        return this;
+    }
+    /**
+     * Checks if metadata exists for a given schema.
+     * @param schema The schema to check.
+     * @returns True if metadata exists, false otherwise.
+     */
+    has(schema) {
+        return this._map.has(schema);
+    }
+    /**
+     * Returns a mapping of channel instances for each property in the schema
+     * using the associated metadata in the registry.
+     *
+     * This is used to create the `channels` object that's passed to the `Graph` constructor.
+     *
+     * @template T The shape of the schema.
+     * @param schema The schema to extract channels from.
+     * @returns A mapping from property names to channel instances.
+     */
+    getChannelsForSchema(schema) {
+        const channels = {};
+        const shape = getInteropZodObjectShape(schema);
+        for (const [key, channelSchema] of Object.entries(shape)) {
+            const meta = this.get(channelSchema);
+            if (meta?.reducer) {
+                channels[key] = new BinaryOperatorAggregate(meta.reducer.fn, meta.default);
+            }
+            else {
+                channels[key] = new LastValue();
+            }
+        }
+        return channels;
+    }
+    /**
+     * Returns a modified schema that introspectively looks at all keys of the provided
+     * object schema, and applies the augmentations based on meta provided with those keys
+     * in the registry and the selectors provided in the `effects` parameter.
+     *
+     * This assumes that the passed in schema is the "root" schema object for a graph where
+     * the keys of the schema are the channels of the graph. Because we need to represent
+     * the input of a graph in a couple of different ways, the `effects` parameter allows
+     * us to apply those augmentations based on pre determined conditions.
+     *
+     * @param schema The root schema object to extend.
+     * @param effects The effects that are being applied.
+     * @returns The extended schema.
+     */
+    getExtendedChannelSchemas(schema, effects) {
+        // If no effects are being applied, return the schema unchanged
+        if (Object.keys(effects).length === 0) {
+            return schema;
+        }
+        // Cache key is determined by looking at the effects that are being applied
+        const cacheKey = Object.entries(effects)
+            .filter(([, v]) => v === true)
+            .sort(([a], [b]) => a.localeCompare(b))
+            .map(([k, v]) => `${k}:${v}`)
+            .join("|");
+        const cache = this._extensionCache.get(cacheKey) ?? new WeakMap();
+        if (cache.has(schema))
+            return cache.get(schema);
+        let modifiedSchema = schema;
+        if (effects.withReducerSchema ||
+            effects.withJsonSchemaExtrasAsDescription) {
+            const newShapeEntries = Object.entries(getInteropZodObjectShape(schema)).map(([key, schema]) => {
+                const meta = this.get(schema);
+                let outputSchema = effects.withReducerSchema
+                    ? meta?.reducer?.schema ?? schema
+                    : schema;
+                if (effects.withJsonSchemaExtrasAsDescription &&
+                    meta?.jsonSchemaExtra) {
+                    const description = getSchemaDescription(outputSchema) ?? getSchemaDescription(schema);
+                    const strExtras = JSON.stringify({
+                        ...meta.jsonSchemaExtra,
+                        description,
+                    });
+                    outputSchema = outputSchema.describe(`${META_EXTRAS_DESCRIPTION_PREFIX}${strExtras}`);
+                }
+                return [key, outputSchema];
+            });
+            modifiedSchema = extendInteropZodObject(schema, Object.fromEntries(newShapeEntries));
+            if (isZodSchemaV3(modifiedSchema)) {
+                modifiedSchema._def.unknownKeys = "strip";
+            }
+        }
+        if (effects.asPartial) {
+            modifiedSchema = interopZodObjectPartial(modifiedSchema);
+        }
+        cache.set(schema, modifiedSchema);
+        this._extensionCache.set(cacheKey, cache);
+        return modifiedSchema;
+    }
+}
+export const schemaMetaRegistry = new SchemaMetaRegistry();
+export function withLangGraph(schema, meta) {
+    if (meta.reducer && !meta.default) {
+        const defaultValueGetter = getInteropZodDefaultGetter(schema);
+        if (defaultValueGetter != null) {
+            // eslint-disable-next-line no-param-reassign
+            meta.default = defaultValueGetter;
+        }
+    }
+    if (meta.reducer) {
+        const schemaWithReducer = Object.assign(schema, {
+            lg_reducer_schema: meta.reducer?.schema ?? schema,
+        });
+        schemaMetaRegistry.extend(schemaWithReducer, () => meta);
+        return schemaWithReducer;
+    }
+    else {
+        schemaMetaRegistry.extend(schema, () => meta);
+        return schema;
+    }
+}
+//# sourceMappingURL=meta.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"meta.js","sourceRoot":"","sources":["../../../src/graph/zod/meta.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,wBAAwB,EACxB,sBAAsB,EACtB,0BAA0B,EAC1B,uBAAuB,EAEvB,aAAa,EACb,oBAAoB,GACrB,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,MAAM,CAAC,MAAM,8BAA8B,GAAG,KAAK,CAAC;AAyCpD;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IAA/B;QACE;;;WAGG;QACH;;;;mBAAO,IAAI,OAAO,EAA8B;WAAC;QAEjD;;;WAGG;QACH;;;;mBAAkB,IAAI,GAAG,EAAmD;WAAC;IA0K/E,CAAC;IAxKC;;;;;;OAMG;IACH,GAAG,CACD,MAA8B;QAE9B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CACJ,MAA8B,EAC9B,SAEgC;QAEhC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAkB,MAAM,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IACjD,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,MAAsB;QAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,GAAG,CAAC,MAAsB;QACxB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED;;;;;;;;;OASG;IACH,oBAAoB,CAClB,MAAS;QAET,MAAM,QAAQ,GAAG,EAAiC,CAAC;QACnD,MAAM,KAAK,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YACrC,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;gBAClB,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,uBAAuB,CAEzC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,SAAS,EAAE,CAAC;YAClC,CAAC;QACH,CAAC;QACD,OAAO,QAA0C,CAAC;IACpD,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,yBAAyB,CACvB,MAAS,EACT,OAaC;QAED,+DAA+D;QAC/D,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtC,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,2EAA2E;QAC3E,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;aACrC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;aACtC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;aAC5B,IAAI,CAAC,GAAG,CAAC,CAAC;QAEb,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,OAAO,EAAE,CAAC;QAClE,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO,KAAK,CAAC,GAAG,CAAC,MAAM,CAAO,CAAC;QAEtD,IAAI,cAAc,GAAqB,MAAM,CAAC;QAE9C,IACE,OAAO,CAAC,iBAAiB;YACzB,OAAO,CAAC,iCAAiC,EACzC,CAAC;YACD,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CACpC,wBAAwB,CAAC,MAAM,CAAC,CACjC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,EAAE;gBACtB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC9B,IAAI,YAAY,GAAG,OAAO,CAAC,iBAAiB;oBAC1C,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,IAAI,MAAM;oBACjC,CAAC,CAAC,MAAM,CAAC;gBACX,IACE,OAAO,CAAC,iCAAiC;oBACzC,IAAI,EAAE,eAAe,EACrB,CAAC;oBACD,MAAM,WAAW,GACf,oBAAoB,CAAC,YAAY,CAAC,IAAI,oBAAoB,CAAC,MAAM,CAAC,CAAC;oBACrE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;wBAC/B,GAAG,IAAI,CAAC,eAAe;wBACvB,WAAW;qBACZ,CAAC,CAAC;oBACH,YAAY,GAAG,YAAY,CAAC,QAAQ,CAClC,GAAG,8BAA8B,GAAG,SAAS,EAAE,CAChD,CAAC;gBACJ,CAAC;gBACD,OAAO,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAC7B,CAAC,CAAC,CAAC;YACH,cAAc,GAAG,sBAAsB,CACrC,MAAM,EACN,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CACpC,CAAC;YACF,IAAI,aAAa,CAAC,cAAc,CAAC,EAAE,CAAC;gBAClC,cAAc,CAAC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACtB,cAAc,GAAG,uBAAuB,CAAC,cAAc,CAAC,CAAC;QAC3D,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,OAAO,cAAc,CAAC;IACxB,CAAC;CACF;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,EAAE,CAAC;AAkB3D,MAAM,UAAU,aAAa,CAK3B,MAAe,EACf,IAAiC;IAEjC,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;QAC9D,IAAI,kBAAkB,IAAI,IAAI,EAAE,CAAC;YAC/B,6CAA6C;YAC7C,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC;QACpC,CAAC;IACH,CAAC;IACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;YAC9C,iBAAiB,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,MAAM;SAClD,CAAC,CAAC;QACH,kBAAkB,CAAC,MAAM,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QACzD,OAAO,iBAAiB,CAAC;IAC3B,CAAC;SAAM,CAAC;QACN,kBAAkB,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO,MAAM,CAAC;IAChB,CAAC;AACH,CAAC"}

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"}