@langchain/langgraph 0.2.74 → 0.3.1

package/dist/graph/zod/schema.cjs

@@ -5,42 +5,9 @@ exports.getUpdateTypeSchema = getUpdateTypeSchema;
 exports.getInputTypeSchema = getInputTypeSchema;
 exports.getOutputTypeSchema = getOutputTypeSchema;
 exports.getConfigTypeSchema = getConfigTypeSchema;
-const zod_1 = require("zod");
 const zod_to_json_schema_1 = require("zod-to-json-schema");
 const state_js_1 = require("./state.cjs");
-const TYPE_CACHE = {};
-const DESCRIPTION_PREFIX = "lg:";
-function applyPlugin(schema, actions) {
-    const cacheKey = [
-        `reducer:${actions.reducer ?? false}`,
-        `jsonSchemaExtra:${actions.jsonSchemaExtra ?? false}`,
-        `partial:${actions.partial ?? false}`,
-    ].join("|");
-    TYPE_CACHE[cacheKey] ??= new WeakMap();
-    const cache = TYPE_CACHE[cacheKey];
-    if (cache.has(schema))
-        return cache.get(schema);
-    let shape = zod_1.z.object({
-        ...Object.fromEntries(Object.entries(schema.shape).map(([key, input]) => {
-            const meta = (0, state_js_1.getMeta)(input);
-            let output = actions.reducer ? meta?.reducer?.schema ?? input : input;
-            if (actions.jsonSchemaExtra) {
-                const strMeta = JSON.stringify({
-                    ...meta?.jsonSchemaExtra,
-                    description: output.description ?? input.description,
-                });
-                if (strMeta !== "{}") {
-                    output = output.describe(`${DESCRIPTION_PREFIX}${strMeta}`);
-                }
-            }
-            return [key, output];
-        })),
-    });
-    if (actions.partial)
-        shape = shape.partial();
-    cache.set(schema, shape);
-    return shape;
-}
+const PartialStateSchema = Symbol.for("langgraph.state.partial");
 function isGraphWithZodLike(graph) {
     if (!graph || typeof graph !== "object")
         return false;
@@ -51,28 +18,8 @@ function isGraphWithZodLike(graph) {
     }
     return true;
 }
-function applyExtraFromDescription(schema) {
-    if (Array.isArray(schema)) {
-        return schema.map(applyExtraFromDescription);
-    }
-    if (typeof schema === "object" && schema != null) {
-        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
-            key,
-            applyExtraFromDescription(value),
-        ]));
-        if ("description" in output &&
-            typeof output.description === "string" &&
-            output.description.startsWith(DESCRIPTION_PREFIX)) {
-            const strMeta = output.description.slice(DESCRIPTION_PREFIX.length);
-            delete output.description;
-            Object.assign(output, JSON.parse(strMeta));
-        }
-        return output;
-    }
-    return schema;
-}
 function toJsonSchema(schema) {
-    return applyExtraFromDescription((0, zod_to_json_schema_1.zodToJsonSchema)(schema));
+    return (0, state_js_1.applyExtraFromDescription)((0, zod_to_json_schema_1.zodToJsonSchema)(schema));
 }
 /**
  * Get the state schema for a graph.
@@ -85,7 +32,7 @@ function getStateTypeSchema(graph) {
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(schemaDef);
+    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, { jsonSchemaExtra: true }));
 }
 /**
  * Get the update schema for a graph.
@@ -98,7 +45,7 @@ function getUpdateTypeSchema(graph) {
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, {
+    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, {
         reducer: true,
         jsonSchemaExtra: true,
         partial: true,
@@ -112,10 +59,14 @@ function getUpdateTypeSchema(graph) {
 function getInputTypeSchema(graph) {
     if (!isGraphWithZodLike(graph))
         return undefined;
-    const schemaDef = graph.builder._inputRuntimeDefinition;
+    let schemaDef = graph.builder._inputRuntimeDefinition;
+    if (schemaDef === PartialStateSchema) {
+        // No need to pass `.partial()` here, that's being done by `applyPlugin`
+        schemaDef = graph.builder._schemaRuntimeDefinition;
+    }
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, {
+    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, {
         reducer: true,
         jsonSchemaExtra: true,
         partial: true,
@@ -132,7 +83,7 @@ function getOutputTypeSchema(graph) {
     const schemaDef = graph.builder._outputRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema((0, state_js_1.applyZodPlugin)(schemaDef, { jsonSchemaExtra: true }));
 }
 /**
  * Get the config schema for a graph.
@@ -145,6 +96,6 @@ function getConfigTypeSchema(graph) {
     const configDef = graph.builder._configRuntimeSchema;
     if (!configDef)
         return undefined;
-    return toJsonSchema(applyPlugin(configDef, { jsonSchemaExtra: true }));
+    return toJsonSchema((0, state_js_1.applyZodPlugin)(configDef, { jsonSchemaExtra: true }));
 }
 //# sourceMappingURL=schema.js.map

package/dist/graph/zod/schema.js

@@ -1,39 +1,6 @@
-import { z } from "zod";
 import { zodToJsonSchema as _zodToJsonSchema } from "zod-to-json-schema";
-import { getMeta } from "./state.js";
-const TYPE_CACHE = {};
-const DESCRIPTION_PREFIX = "lg:";
-function applyPlugin(schema, actions) {
-    const cacheKey = [
-        `reducer:${actions.reducer ?? false}`,
-        `jsonSchemaExtra:${actions.jsonSchemaExtra ?? false}`,
-        `partial:${actions.partial ?? false}`,
-    ].join("|");
-    TYPE_CACHE[cacheKey] ??= new WeakMap();
-    const cache = TYPE_CACHE[cacheKey];
-    if (cache.has(schema))
-        return cache.get(schema);
-    let shape = z.object({
-        ...Object.fromEntries(Object.entries(schema.shape).map(([key, input]) => {
-            const meta = getMeta(input);
-            let output = actions.reducer ? meta?.reducer?.schema ?? input : input;
-            if (actions.jsonSchemaExtra) {
-                const strMeta = JSON.stringify({
-                    ...meta?.jsonSchemaExtra,
-                    description: output.description ?? input.description,
-                });
-                if (strMeta !== "{}") {
-                    output = output.describe(`${DESCRIPTION_PREFIX}${strMeta}`);
-                }
-            }
-            return [key, output];
-        })),
-    });
-    if (actions.partial)
-        shape = shape.partial();
-    cache.set(schema, shape);
-    return shape;
-}
+import { applyZodPlugin, applyExtraFromDescription } from "./state.js";
+const PartialStateSchema = Symbol.for("langgraph.state.partial");
 function isGraphWithZodLike(graph) {
     if (!graph || typeof graph !== "object")
         return false;
@@ -44,26 +11,6 @@ function isGraphWithZodLike(graph) {
     }
     return true;
 }
-function applyExtraFromDescription(schema) {
-    if (Array.isArray(schema)) {
-        return schema.map(applyExtraFromDescription);
-    }
-    if (typeof schema === "object" && schema != null) {
-        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
-            key,
-            applyExtraFromDescription(value),
-        ]));
-        if ("description" in output &&
-            typeof output.description === "string" &&
-            output.description.startsWith(DESCRIPTION_PREFIX)) {
-            const strMeta = output.description.slice(DESCRIPTION_PREFIX.length);
-            delete output.description;
-            Object.assign(output, JSON.parse(strMeta));
-        }
-        return output;
-    }
-    return schema;
-}
 function toJsonSchema(schema) {
     return applyExtraFromDescription(_zodToJsonSchema(schema));
 }
@@ -78,7 +25,7 @@ export function getStateTypeSchema(graph) {
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(schemaDef);
+    return toJsonSchema(applyZodPlugin(schemaDef, { jsonSchemaExtra: true }));
 }
 /**
  * Get the update schema for a graph.
@@ -91,7 +38,7 @@ export function getUpdateTypeSchema(graph) {
     const schemaDef = graph.builder._schemaRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, {
+    return toJsonSchema(applyZodPlugin(schemaDef, {
         reducer: true,
         jsonSchemaExtra: true,
         partial: true,
@@ -105,10 +52,14 @@ export function getUpdateTypeSchema(graph) {
 export function getInputTypeSchema(graph) {
     if (!isGraphWithZodLike(graph))
         return undefined;
-    const schemaDef = graph.builder._inputRuntimeDefinition;
+    let schemaDef = graph.builder._inputRuntimeDefinition;
+    if (schemaDef === PartialStateSchema) {
+        // No need to pass `.partial()` here, that's being done by `applyPlugin`
+        schemaDef = graph.builder._schemaRuntimeDefinition;
+    }
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, {
+    return toJsonSchema(applyZodPlugin(schemaDef, {
         reducer: true,
         jsonSchemaExtra: true,
         partial: true,
@@ -125,7 +76,7 @@ export function getOutputTypeSchema(graph) {
     const schemaDef = graph.builder._outputRuntimeDefinition;
     if (!schemaDef)
         return undefined;
-    return toJsonSchema(applyPlugin(schemaDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(applyZodPlugin(schemaDef, { jsonSchemaExtra: true }));
 }
 /**
  * Get the config schema for a graph.
@@ -138,6 +89,6 @@ export function getConfigTypeSchema(graph) {
     const configDef = graph.builder._configRuntimeSchema;
     if (!configDef)
         return undefined;
-    return toJsonSchema(applyPlugin(configDef, { jsonSchemaExtra: true }));
+    return toJsonSchema(applyZodPlugin(configDef, { jsonSchemaExtra: 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":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,eAAe,IAAI,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,MAAM,UAAU,GAA4D,EAAE,CAAC;AAC/E,MAAM,kBAAkB,GAAG,KAAK,CAAC;AAEjC,SAAS,WAAW,CAClB,MAAsB,EACtB,OASC;IAED,MAAM,QAAQ,GAAG;QACf,WAAW,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE;QACrC,mBAAmB,OAAO,CAAC,eAAe,IAAI,KAAK,EAAE;QACrD,WAAW,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE;KACtC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEZ,UAAU,CAAC,QAAQ,CAAC,KAAK,IAAI,OAAO,EAAE,CAAC;IACvC,MAAM,KAAK,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAEnC,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;IACjD,IAAI,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC;QACnB,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAqC,CAAC,CAAC,GAAG,CAC9D,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAA0B,EAAE;YACvC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;YAEtE,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC7B,GAAG,IAAI,EAAE,eAAe;oBACxB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW;iBACrD,CAAC,CAAC;gBAEH,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;oBACrB,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,kBAAkB,GAAG,OAAO,EAAE,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;YAED,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACvB,CAAC,CACF,CACF;KACF,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,OAAO;QAAE,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC;IAC7C,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACzB,OAAO,KAAK,CAAC;AACf,CAAC;AAYD,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;AAID,SAAS,yBAAyB,CAAC,MAAe;IAChD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,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,yBAAyB,CAAC,KAAK,CAAC;SACjC,CAAC,CACH,CAAC;QAEF,IACE,aAAa,IAAI,MAAM;YACvB,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;YACtC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,kBAAkB,CAAC,EACjD,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;YACpE,OAAO,MAAM,CAAC,WAAW,CAAC;YAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,MAAoB,CAAC;AAC9B,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,SAAS,CAAC,CAAC;AACjC,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,WAAW,CAAC,SAAS,EAAE;QACrB,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,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,uBAAuB,CAAC;IACxD,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,YAAY,CACjB,WAAW,CAAC,SAAS,EAAE;QACrB,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,WAAW,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AACzE,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,WAAW,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AACzE,CAAC"}
+{"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"}

package/dist/graph/zod/state.cjs

@@ -6,6 +6,8 @@ exports.withLangGraph = withLangGraph;
 exports.getMeta = getMeta;
 exports.extendMeta = extendMeta;
 exports.getChannelsFromZod = getChannelsFromZod;
+exports.applyZodPlugin = applyZodPlugin;
+exports.applyExtraFromDescription = applyExtraFromDescription;
 const binop_js_1 = require("../../channels/binop.cjs");
 const last_value_js_1 = require("../../channels/last_value.cjs");
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -69,4 +71,65 @@ function getChannelsFromZod(schema) {
     }
     return channels;
 }
+const ZOD_TYPE_CACHE = {};
+const ZOD_DESCRIPTION_PREFIX = "lg:";
+function applyZodPlugin(schema, actions) {
+    const cacheKey = [
+        `reducer:${actions.reducer ?? false}`,
+        `jsonSchemaExtra:${actions.jsonSchemaExtra ?? false}`,
+        `partial:${actions.partial ?? false}`,
+    ].join("|");
+    ZOD_TYPE_CACHE[cacheKey] ??= new WeakMap();
+    const cache = ZOD_TYPE_CACHE[cacheKey];
+    if (cache.has(schema))
+        return cache.get(schema);
+    let shape = schema.extend({
+        ...Object.fromEntries(Object.entries(schema.shape).map(([key, input]) => {
+            const meta = getMeta(input);
+            let output = actions.reducer ? meta?.reducer?.schema ?? input : input;
+            if (actions.jsonSchemaExtra) {
+                const strMeta = JSON.stringify({
+                    ...meta?.jsonSchemaExtra,
+                    description: output.description ?? input.description,
+                });
+                if (strMeta !== "{}") {
+                    output = output.describe(`${ZOD_DESCRIPTION_PREFIX}${strMeta}`);
+                }
+            }
+            return [key, output];
+        })),
+    });
+    // using zObject.extend() will set `unknownKeys` to `passthrough`
+    // which trips up `zod-to-json-schema`
+    if ("_def" in shape &&
+        shape._def != null &&
+        typeof shape._def === "object" &&
+        "unknownKeys" in shape._def) {
+        shape._def.unknownKeys = "strip";
+    }
+    if (actions.partial)
+        shape = shape.partial();
+    cache.set(schema, shape);
+    return shape;
+}
+function applyExtraFromDescription(schema) {
+    if (Array.isArray(schema)) {
+        return schema.map(applyExtraFromDescription);
+    }
+    if (typeof schema === "object" && schema != null) {
+        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
+            key,
+            applyExtraFromDescription(value),
+        ]));
+        if ("description" in output &&
+            typeof output.description === "string" &&
+            output.description.startsWith(ZOD_DESCRIPTION_PREFIX)) {
+            const strMeta = output.description.slice(ZOD_DESCRIPTION_PREFIX.length);
+            delete output.description;
+            Object.assign(output, JSON.parse(strMeta));
+        }
+        return output;
+    }
+    return schema;
+}
 //# sourceMappingURL=state.js.map

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

@@ -3,7 +3,7 @@ import { BaseChannel } from "../../channels/base.js";
 export interface Meta<ValueType, UpdateType = ValueType> {
     jsonSchemaExtra?: {
         langgraph_nodes?: string[];
-        langgraph_type?: "prompt";
+        langgraph_type?: "prompt" | "messages";
         [key: string]: unknown;
     };
     reducer?: {
@@ -28,3 +28,12 @@ export type ZodToStateDefinition<T extends AnyZodObject> = {
     [key in keyof T["shape"]]: T["shape"][key] extends z.ZodType<infer V, z.ZodTypeDef, infer U> ? BaseChannel<V, U> : never;
 };
 export declare function getChannelsFromZod<T extends z.ZodRawShape>(schema: z.ZodObject<T>): ZodToStateDefinition<z.ZodObject<T>>;
+export declare function applyZodPlugin(schema: z.AnyZodObject, actions: {
+    /** Apply .langgraph.reducer calls */
+    reducer?: boolean;
+    /** Apply .langgraph.metadata() calls  */
+    jsonSchemaExtra?: boolean;
+    /** Apply .partial() */
+    partial?: boolean;
+}): z.AnyZodObject;
+export declare function applyExtraFromDescription(schema: unknown): unknown;

package/dist/graph/zod/state.js

@@ -61,4 +61,65 @@ export function getChannelsFromZod(schema) {
     }
     return channels;
 }
+const ZOD_TYPE_CACHE = {};
+const ZOD_DESCRIPTION_PREFIX = "lg:";
+export function applyZodPlugin(schema, actions) {
+    const cacheKey = [
+        `reducer:${actions.reducer ?? false}`,
+        `jsonSchemaExtra:${actions.jsonSchemaExtra ?? false}`,
+        `partial:${actions.partial ?? false}`,
+    ].join("|");
+    ZOD_TYPE_CACHE[cacheKey] ??= new WeakMap();
+    const cache = ZOD_TYPE_CACHE[cacheKey];
+    if (cache.has(schema))
+        return cache.get(schema);
+    let shape = schema.extend({
+        ...Object.fromEntries(Object.entries(schema.shape).map(([key, input]) => {
+            const meta = getMeta(input);
+            let output = actions.reducer ? meta?.reducer?.schema ?? input : input;
+            if (actions.jsonSchemaExtra) {
+                const strMeta = JSON.stringify({
+                    ...meta?.jsonSchemaExtra,
+                    description: output.description ?? input.description,
+                });
+                if (strMeta !== "{}") {
+                    output = output.describe(`${ZOD_DESCRIPTION_PREFIX}${strMeta}`);
+                }
+            }
+            return [key, output];
+        })),
+    });
+    // using zObject.extend() will set `unknownKeys` to `passthrough`
+    // which trips up `zod-to-json-schema`
+    if ("_def" in shape &&
+        shape._def != null &&
+        typeof shape._def === "object" &&
+        "unknownKeys" in shape._def) {
+        shape._def.unknownKeys = "strip";
+    }
+    if (actions.partial)
+        shape = shape.partial();
+    cache.set(schema, shape);
+    return shape;
+}
+export function applyExtraFromDescription(schema) {
+    if (Array.isArray(schema)) {
+        return schema.map(applyExtraFromDescription);
+    }
+    if (typeof schema === "object" && schema != null) {
+        const output = Object.fromEntries(Object.entries(schema).map(([key, value]) => [
+            key,
+            applyExtraFromDescription(value),
+        ]));
+        if ("description" in output &&
+            typeof output.description === "string" &&
+            output.description.startsWith(ZOD_DESCRIPTION_PREFIX)) {
+            const strMeta = output.description.slice(ZOD_DESCRIPTION_PREFIX.length);
+            delete output.description;
+            Object.assign(output, JSON.parse(strMeta));
+        }
+        return output;
+    }
+    return schema;
+}
 //# sourceMappingURL=state.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"state.js","sourceRoot":"","sources":["../../../src/graph/zod/state.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,8DAA8D;AAC9D,MAAM,QAAQ,GAAG,IAAI,OAAO,EAA6B,CAAC;AAkB1D,SAAS,SAAS,CAAC,KAAc;IAC/B,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,IAAI,IAAI;QACb,QAAQ,IAAI,KAAK;QACjB,OAAO,KAAK,CAAC,MAAM,KAAK,UAAU,CACnC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAc;IAEd,OAAO,CACL,SAAS,CAAC,KAAK,CAAC;QAChB,eAAe,IAAI,KAAK;QACxB,OAAO,KAAK,CAAC,aAAa,KAAK,UAAU,CAC1C,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,OAAO,CACL,SAAS,CAAC,KAAK,CAAC;QAChB,SAAS,IAAI,KAAK;QAClB,OAAO,KAAK,CAAC,OAAO,KAAK,UAAU,CACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,MAAwC,EACxC,IAAiC;IAEjC,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC;YACvC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY;YAC1B,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YACzB,6CAA6C;YAC7C,IAAI,CAAC,OAAO,GAAG,YAAY,CAAC;QAC9B,CAAC;IACH,CAAC;IACD,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC3B,OAAO,MAAwD,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,OAAO,CACrB,MAA4B;IAE5B,OAAO,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,MAA4B,EAC5B,MAEgC;IAEhC,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAEtB,CAAC;IACd,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IACrC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAChC,CAAC;AAYD,MAAM,UAAU,kBAAkB,CAChC,MAAsB;IAEtB,MAAM,QAAQ,GAAG,EAAiC,CAAC;IACnD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;YAC5D,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACpC,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;gBAClB,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,uBAAuB,CACzC,IAAI,CAAC,OAAO,CAAC,EAAE,EACf,IAAI,CAAC,OAAO,CACb,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,SAAS,EAAE,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,QAAgD,CAAC;AAC1D,CAAC"}
+{"version":3,"file":"state.js","sourceRoot":"","sources":["../../../src/graph/zod/state.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,8DAA8D;AAC9D,MAAM,QAAQ,GAAG,IAAI,OAAO,EAA6B,CAAC;AAkB1D,SAAS,SAAS,CAAC,KAAc;IAC/B,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,IAAI,IAAI;QACb,QAAQ,IAAI,KAAK;QACjB,OAAO,KAAK,CAAC,MAAM,KAAK,UAAU,CACnC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAc;IAEd,OAAO,CACL,SAAS,CAAC,KAAK,CAAC;QAChB,eAAe,IAAI,KAAK;QACxB,OAAO,KAAK,CAAC,aAAa,KAAK,UAAU,CAC1C,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,OAAO,CACL,SAAS,CAAC,KAAK,CAAC;QAChB,SAAS,IAAI,KAAK;QAClB,OAAO,KAAK,CAAC,OAAO,KAAK,UAAU,CACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,MAAwC,EACxC,IAAiC;IAEjC,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC;YACvC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY;YAC1B,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;YACzB,6CAA6C;YAC7C,IAAI,CAAC,OAAO,GAAG,YAAY,CAAC;QAC9B,CAAC;IACH,CAAC;IACD,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC3B,OAAO,MAAwD,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,OAAO,CACrB,MAA4B;IAE5B,OAAO,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,MAA4B,EAC5B,MAEgC;IAEhC,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAEtB,CAAC;IACd,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IACrC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAChC,CAAC;AAYD,MAAM,UAAU,kBAAkB,CAChC,MAAsB;IAEtB,MAAM,QAAQ,GAAG,EAAiC,CAAC;IACnD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;YAC5D,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACpC,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;gBAClB,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,uBAAuB,CACzC,IAAI,CAAC,OAAO,CAAC,EAAE,EACf,IAAI,CAAC,OAAO,CACb,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,GAAG,CAAC,GAAG,IAAI,SAAS,EAAE,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,QAAgD,CAAC;AAC1D,CAAC;AAED,MAAM,cAAc,GAGhB,EAAE,CAAC;AAEP,MAAM,sBAAsB,GAAG,KAAK,CAAC;AAErC,MAAM,UAAU,cAAc,CAC5B,MAAsB,EACtB,OASC;IAED,MAAM,QAAQ,GAAG;QACf,WAAW,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE;QACrC,mBAAmB,OAAO,CAAC,eAAe,IAAI,KAAK,EAAE;QACrD,WAAW,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE;KACtC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEZ,cAAc,CAAC,QAAQ,CAAC,KAAK,IAAI,OAAO,EAAE,CAAC;IAC3C,MAAM,KAAK,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAEvC,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;IAEjD,IAAI,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC;QACxB,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAqC,CAAC,CAAC,GAAG,CAC9D,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAA0B,EAAE;YACvC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;YAEtE,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC7B,GAAG,IAAI,EAAE,eAAe;oBACxB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW;iBACrD,CAAC,CAAC;gBAEH,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;oBACrB,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,sBAAsB,GAAG,OAAO,EAAE,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC;YAED,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACvB,CAAC,CACF,CACF;KACF,CAAC,CAAC;IAEH,iEAAiE;IACjE,sCAAsC;IACtC,IACE,MAAM,IAAI,KAAK;QACf,KAAK,CAAC,IAAI,IAAI,IAAI;QAClB,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAC9B,aAAa,IAAI,KAAK,CAAC,IAAI,EAC3B,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;IACnC,CAAC;IAED,IAAI,OAAO,CAAC,OAAO;QAAE,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC;IAC7C,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACzB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,MAAe;IACvD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,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,yBAAyB,CAAC,KAAK,CAAC;SACjC,CAAC,CACH,CAAC;QAEF,IACE,aAAa,IAAI,MAAM;YACvB,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;YACtC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,sBAAsB,CAAC,EACrD,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;YACxE,OAAO,MAAM,CAAC,WAAW,CAAC;YAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/prebuilt/react_agent_executor.cjs

@@ -56,6 +56,9 @@ function _getPromptRunnable(prompt) {
     }
     return promptRunnable;
 }
+function isClientTool(tool) {
+    return runnables_1.Runnable.isRunnable(tool);
+}
 function _getPrompt(prompt, stateModifier, messageModifier) {
     // Check if multiple modifiers exist
     const definedCount = [prompt, stateModifier, messageModifier].filter((x) => x != null).length;
@@ -113,7 +116,7 @@ async function _shouldBindTools(llm, tools) {
     if (tools.length !== boundTools.length) {
         throw new Error("Number of tools in the model.bindTools() and tools passed to createReactAgent must match");
     }
-    const toolNames = new Set(tools.map((tool) => tool.name));
+    const toolNames = new Set(tools.flatMap((tool) => (isClientTool(tool) ? tool.name : [])));
     const boundToolNames = new Set();
     for (const boundTool of boundTools) {
         let boundToolName;
@@ -121,7 +124,7 @@ async function _shouldBindTools(llm, tools) {
         if ("type" in boundTool && boundTool.type === "function") {
             boundToolName = boundTool.function.name;
         }
-        // Anthropic- or Google-style tool
+        // Anthropic or Google-style tool
         else if ("name" in boundTool) {
             boundToolName = boundTool.name;
         }
@@ -173,6 +176,12 @@ const createReactAgentAnnotation = () => annotation_js_1.Annotation.Root({
     structuredResponse: (annotation_js_1.Annotation),
 });
 exports.createReactAgentAnnotation = createReactAgentAnnotation;
+const PreHookAnnotation = annotation_js_1.Annotation.Root({
+    llmInputMessages: (0, annotation_js_1.Annotation)({
+        reducer: message_js_1.messagesStateReducer,
+        default: () => [],
+    }),
+});
 /**
  * Creates a StateGraph agent that relies on a chat model utilizing tool calling.
  *
@@ -216,7 +225,7 @@ exports.createReactAgentAnnotation = createReactAgentAnnotation;
  * ```
  */
 function createReactAgent(params) {
-    const { llm, tools, messageModifier, stateModifier, prompt, stateSchema, checkpointSaver, checkpointer, interruptBefore, interruptAfter, store, responseFormat, name, includeAgentName, } = params;
+    const { llm, tools, messageModifier, stateModifier, prompt, stateSchema, checkpointSaver, checkpointer, interruptBefore, interruptAfter, store, responseFormat, preModelHook, postModelHook, name, includeAgentName, } = params;
     let toolClasses;
     let toolNode;
     if (!Array.isArray(tools)) {
@@ -225,7 +234,7 @@ function createReactAgent(params) {
     }
     else {
         toolClasses = tools;
-        toolNode = new tool_node_js_1.ToolNode(tools);
+        toolNode = new tool_node_js_1.ToolNode(toolClasses.filter(isClientTool));
     }
     let cachedModelRunnable = null;
     const getModelRunnable = async (llm) => {
@@ -252,19 +261,16 @@ function createReactAgent(params) {
     // If any of the tools are configured to return_directly after running,
     // our graph needs to check if these were called
     const shouldReturnDirect = new Set(toolClasses
+        .filter(isClientTool)
         .filter((tool) => "returnDirect" in tool && tool.returnDirect)
         .map((tool) => tool.name));
-    const shouldContinue = (state) => {
-        const { messages } = state;
-        const lastMessage = messages[messages.length - 1];
-        if ((0, messages_1.isAIMessage)(lastMessage) &&
-            (!lastMessage.tool_calls || lastMessage.tool_calls.length === 0)) {
-            return responseFormat != null ? "generate_structured_response" : constants_js_1.END;
-        }
-        else {
-            return "continue";
+    function getModelInputState(state) {
+        const { messages, llmInputMessages, ...rest } = state;
+        if (llmInputMessages != null && llmInputMessages.length > 0) {
+            return { messages: llmInputMessages, ...rest };
         }
-    };
+        return { messages, ...rest };
+    }
     const generateStructuredResponse = async (state, config) => {
         if (responseFormat == null) {
             throw new Error("Attempted to generate structured output with no passed response schema. Please contact us for help.");
@@ -289,54 +295,102 @@ function createReactAgent(params) {
         // to ensure that we can validate ConfigurableModel properly
         const modelRunnable = await getModelRunnable(llm);
         // TODO: Auto-promote streaming.
-        const response = (await modelRunnable.invoke(state, config));
+        const response = (await modelRunnable.invoke(getModelInputState(state), config));
         // add agent name to the AIMessage
         // TODO: figure out if we can avoid mutating the message directly
         response.name = name;
         response.lc_kwargs.name = name;
         return { messages: [response] };
     };
-    const workflow = new index_js_1.StateGraph(stateSchema ?? (0, exports.createReactAgentAnnotation)())
-        .addNode("agent", callModel)
-        .addNode("tools", toolNode)
-        .addEdge(constants_js_1.START, "agent");
-    if (responseFormat !== undefined) {
-        workflow
-            .addNode("generate_structured_response", generateStructuredResponse)
-            .addEdge("generate_structured_response", constants_js_1.END)
-            .addConditionalEdges("agent", shouldContinue, {
-            continue: "tools",
-            [constants_js_1.END]: constants_js_1.END,
-            generate_structured_response: "generate_structured_response",
+    const schema = stateSchema ?? (0, exports.createReactAgentAnnotation)();
+    const workflow = new index_js_1.StateGraph(schema).addNode("tools", toolNode);
+    const allNodeWorkflows = workflow;
+    const conditionalMap = (map) => {
+        return Object.fromEntries(Object.entries(map).filter(([_, v]) => v != null));
+    };
+    let entrypoint = "agent";
+    let inputSchema;
+    if (preModelHook != null) {
+        allNodeWorkflows
+            .addNode("pre_model_hook", preModelHook)
+            .addEdge("pre_model_hook", "agent");
+        entrypoint = "pre_model_hook";
+        inputSchema = annotation_js_1.Annotation.Root({
+            ...schema.spec,
+            ...PreHookAnnotation.spec,
         });
     }
     else {
-        workflow.addConditionalEdges("agent", shouldContinue, {
-            continue: "tools",
-            [constants_js_1.END]: constants_js_1.END,
-        });
+        entrypoint = "agent";
     }
-    const routeToolResponses = (state) => {
-        // Check the last consecutive tool calls
-        for (let i = state.messages.length - 1; i >= 0; i -= 1) {
-            const message = state.messages[i];
-            if (!(0, messages_1.isToolMessage)(message)) {
-                break;
+    allNodeWorkflows
+        .addNode("agent", callModel, { input: inputSchema })
+        .addEdge(constants_js_1.START, entrypoint);
+    if (postModelHook != null) {
+        allNodeWorkflows
+            .addNode("post_model_hook", postModelHook)
+            .addEdge("agent", "post_model_hook")
+            .addConditionalEdges("post_model_hook", (state) => {
+            const { messages } = state;
+            const lastMessage = messages[messages.length - 1];
+            if ((0, messages_1.isAIMessage)(lastMessage) && lastMessage.tool_calls?.length) {
+                return "tools";
             }
-            // Check if this tool is configured to return directly
-            if (message.name !== undefined && shouldReturnDirect.has(message.name)) {
+            if ((0, messages_1.isToolMessage)(lastMessage))
+                return entrypoint;
+            if (responseFormat != null)
+                return "generate_structured_response";
+            return constants_js_1.END;
+        }, conditionalMap({
+            tools: "tools",
+            [entrypoint]: entrypoint,
+            generate_structured_response: responseFormat != null ? "generate_structured_response" : null,
+            [constants_js_1.END]: responseFormat != null ? null : constants_js_1.END,
+        }));
+    }
+    if (responseFormat !== undefined) {
+        workflow
+            .addNode("generate_structured_response", generateStructuredResponse)
+            .addEdge("generate_structured_response", constants_js_1.END);
+    }
+    if (postModelHook == null) {
+        allNodeWorkflows.addConditionalEdges("agent", (state) => {
+            const { messages } = state;
+            const lastMessage = messages[messages.length - 1];
+            // if there's no function call, we finish
+            if (!(0, messages_1.isAIMessage)(lastMessage) || !lastMessage.tool_calls?.length) {
+                if (responseFormat != null)
+                    return "generate_structured_response";
                 return constants_js_1.END;
             }
-        }
-        return "agent";
-    };
+            // there are function calls, we continue
+            return "tools";
+        }, conditionalMap({
+            tools: "tools",
+            generate_structured_response: responseFormat != null ? "generate_structured_response" : null,
+            [constants_js_1.END]: responseFormat != null ? null : constants_js_1.END,
+        }));
+    }
     if (shouldReturnDirect.size > 0) {
-        workflow.addConditionalEdges("tools", routeToolResponses, ["agent", constants_js_1.END]);
+        allNodeWorkflows.addConditionalEdges("tools", (state) => {
+            // Check the last consecutive tool calls
+            for (let i = state.messages.length - 1; i >= 0; i -= 1) {
+                const message = state.messages[i];
+                if (!(0, messages_1.isToolMessage)(message))
+                    break;
+                // Check if this tool is configured to return directly
+                if (message.name !== undefined &&
+                    shouldReturnDirect.has(message.name)) {
+                    return constants_js_1.END;
+                }
+            }
+            return entrypoint;
+        }, conditionalMap({ [entrypoint]: entrypoint, [constants_js_1.END]: constants_js_1.END }));
     }
     else {
-        workflow.addEdge("tools", "agent");
+        allNodeWorkflows.addEdge("tools", entrypoint);
     }
-    return workflow.compile({
+    return allNodeWorkflows.compile({
         checkpointer: checkpointer ?? checkpointSaver,
         interruptBefore,
         interruptAfter,