UNPKG

@langchain/core

Version:
88 lines (87 loc) 4.18 kB
import { __exportAll } from "../_virtual/_rolldown/runtime.js"; import { interopZodObjectStrict, interopZodTransformInputSchema, isZodObjectV4, isZodSchemaV3, isZodSchemaV4 } from "./types/zod.js"; import { zodToJsonSchema } from "./zod-to-json-schema/zodToJsonSchema.js"; import "./zod-to-json-schema/index.js"; import { isStandardJsonSchema } from "./standard_schema.js"; import { toJSONSchema } from "zod/v4/core"; import { Validator, deepCompareStrict, dereference } from "@cfworker/json-schema"; //#region src/utils/json_schema.ts var json_schema_exports = /* @__PURE__ */ __exportAll({ Validator: () => Validator, deepCompareStrict: () => deepCompareStrict, toJsonSchema: () => toJsonSchema, validatesOnlyStrings: () => validatesOnlyStrings }); /** * WeakMap cache for Zod/Standard-Schema → JSON Schema conversions. * * Keyed on the schema object reference. Since Zod schemas are immutable and * the same `tool.schema` reference is passed on every LLM call, this * eliminates redundant serializations. For example, an agent with 6 tools * doing 15 steps across 3 parallel subagents would otherwise run 270 * identical conversions per invocation. * * Only used when no custom `params` are passed (the common case for tool * binding). WeakMap ensures cached entries are GC'd when the schema goes * out of scope. * * @internal */ const _jsonSchemaCache = /* @__PURE__ */ new WeakMap(); /** * Converts a Standard JSON schema, Zod schema or JSON schema to a JSON schema. * Results are cached by schema reference when no custom params are passed. * @param schema - The schema to convert. * @param params - The parameters to pass to the toJSONSchema function. * @returns The converted schema. */ function toJsonSchema(schema, params) { const canCache = !params && schema != null && typeof schema === "object"; if (canCache) { const cached = _jsonSchemaCache.get(schema); if (cached) return cached; } let result; if (isStandardJsonSchema(schema) && !isZodSchemaV4(schema)) result = schema["~standard"].jsonSchema.input({ target: "draft-07" }); else if (isZodSchemaV4(schema)) { const inputSchema = interopZodTransformInputSchema(schema, true); if (isZodObjectV4(inputSchema)) result = toJSONSchema(interopZodObjectStrict(inputSchema, true), params); else result = toJSONSchema(schema, params); } else if (isZodSchemaV3(schema)) result = zodToJsonSchema(schema); else result = schema; if (canCache && result != null && typeof result === "object") _jsonSchemaCache.set(schema, result); return result; } /** * Validates if a JSON schema validates only strings. May return false negatives in some edge cases * (like recursive or unresolvable refs). * * @param schema - The schema to validate. * @returns `true` if the schema validates only strings, `false` otherwise. */ function validatesOnlyStrings(schema) { if (!schema || typeof schema !== "object" || Object.keys(schema).length === 0 || Array.isArray(schema)) return false; if ("type" in schema) { if (typeof schema.type === "string") return schema.type === "string"; if (Array.isArray(schema.type)) return schema.type.every((t) => t === "string"); return false; } if ("enum" in schema) return Array.isArray(schema.enum) && schema.enum.length > 0 && schema.enum.every((val) => typeof val === "string"); if ("const" in schema) return typeof schema.const === "string"; if ("allOf" in schema && Array.isArray(schema.allOf)) return schema.allOf.some((subschema) => validatesOnlyStrings(subschema)); if ("anyOf" in schema && Array.isArray(schema.anyOf) || "oneOf" in schema && Array.isArray(schema.oneOf)) { const subschemas = "anyOf" in schema ? schema.anyOf : schema.oneOf; return subschemas.length > 0 && subschemas.every((subschema) => validatesOnlyStrings(subschema)); } if ("not" in schema) return false; if ("$ref" in schema && typeof schema.$ref === "string") { const ref = schema.$ref; const resolved = dereference(schema); if (resolved[ref]) return validatesOnlyStrings(resolved[ref]); return false; } return false; } //#endregion export { Validator, deepCompareStrict, json_schema_exports, toJsonSchema, validatesOnlyStrings }; //# sourceMappingURL=json_schema.js.map