@mastra/schema-compat 0.0.0-fix-message-list-args-missing-20250807205055 → 0.0.0-fix-zod-to-json-schema-ref-strategy-20250910193441

package/src/provider-compats/anthropic.ts

@@ -1,43 +0,0 @@
-import type { ZodTypeAny } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import { SchemaCompatLayer, isArr, isObj, isOptional, isString, isUnion } from '../schema-compatibility';
-import type { AllZodType, ModelInformation } from '../schema-compatibility';
-
-export class AnthropicSchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return 'jsonSchema7';
-  }
-
-  shouldApply(): boolean {
-    return this.getModel().modelId.includes('claude');
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      const handleTypes: AllZodType[] = ['ZodObject', 'ZodArray', 'ZodUnion', 'ZodNever', 'ZodUndefined', 'ZodTuple'];
-      if (this.getModel().modelId.includes('claude-3.5-haiku')) handleTypes.push('ZodString');
-      return this.defaultZodOptionalHandler(value, handleTypes);
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value);
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value, []);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isString(value)) {
-      // the claude-3.5-haiku model support these properties but the model doesn't respect them, but it respects them when they're
-      // added to the tool description
-
-      if (this.getModel().modelId.includes('claude-3.5-haiku')) {
-        return this.defaultZodStringHandler(value, ['max', 'min']);
-      } else {
-        return value;
-      }
-    }
-
-    return this.defaultUnsupportedZodTypeHandler(value, ['ZodNever', 'ZodTuple', 'ZodUndefined']);
-  }
-}

package/src/provider-compats/deepseek.ts

@@ -1,35 +0,0 @@
-import type { ZodTypeAny } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import { SchemaCompatLayer, isArr, isObj, isOptional, isString, isUnion } from '../schema-compatibility';
-import type { ModelInformation } from '../schema-compatibility';
-
-export class DeepSeekSchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return 'jsonSchema7';
-  }
-
-  shouldApply(): boolean {
-    // Deepseek R1 performs perfectly without this compat layer
-    return this.getModel().modelId.includes('deepseek') && !this.getModel().modelId.includes('r1');
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      return this.defaultZodOptionalHandler(value, ['ZodObject', 'ZodArray', 'ZodUnion', 'ZodString', 'ZodNumber']);
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value);
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value, ['min', 'max']);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isString(value)) {
-      return this.defaultZodStringHandler(value);
-    }
-
-    return value;
-  }
-}

package/src/provider-compats/google.ts

@@ -1,63 +0,0 @@
-import type { ZodTypeAny } from 'zod';
-import { z } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import type { ModelInformation } from '../schema-compatibility';
-import {
-  SchemaCompatLayer,
-  UNSUPPORTED_ZOD_TYPES,
-  isArr,
-  isNull,
-  isNumber,
-  isObj,
-  isOptional,
-  isString,
-  isUnion,
-} from '../schema-compatibility';
-
-export class GoogleSchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return 'jsonSchema7';
-  }
-
-  shouldApply(): boolean {
-    return this.getModel().provider.includes('google') || this.getModel().modelId.includes('google');
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      return this.defaultZodOptionalHandler(value, [
-        'ZodObject',
-        'ZodArray',
-        'ZodUnion',
-        'ZodString',
-        'ZodNumber',
-        ...UNSUPPORTED_ZOD_TYPES,
-      ]);
-    } else if (isNull(value)) {
-      // Google models don't support null, so we need to convert it to any and then refine it to null
-      return z
-        .any()
-        .refine(v => v === null, { message: 'must be null' })
-        .describe(value._def.description || 'must be null');
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value);
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value, []);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isString(value)) {
-      // Google models support these properties but the model doesn't respect them, but it respects them when they're
-      // added to the tool description
-      return this.defaultZodStringHandler(value);
-    } else if (isNumber(value)) {
-      // Google models support these properties but the model doesn't respect them, but it respects them when they're
-      // added to the tool description
-      return this.defaultZodNumberHandler(value);
-    }
-    return this.defaultUnsupportedZodTypeHandler(value);
-  }
-}

package/src/provider-compats/meta.ts

@@ -1,36 +0,0 @@
-import type { ZodTypeAny } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import { SchemaCompatLayer, isArr, isNumber, isObj, isOptional, isString, isUnion } from '../schema-compatibility';
-import type { ModelInformation } from '../schema-compatibility';
-
-export class MetaSchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return 'jsonSchema7';
-  }
-
-  shouldApply(): boolean {
-    return this.getModel().modelId.includes('meta');
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      return this.defaultZodOptionalHandler(value, ['ZodObject', 'ZodArray', 'ZodUnion', 'ZodString', 'ZodNumber']);
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value);
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value, ['min', 'max']);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isNumber(value)) {
-      return this.defaultZodNumberHandler(value);
-    } else if (isString(value)) {
-      return this.defaultZodStringHandler(value);
-    }
-
-    return value;
-  }
-}

package/src/provider-compats/openai-reasoning.ts

@@ -1,91 +0,0 @@
-import { z } from 'zod';
-import type { ZodTypeAny } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import type { ModelInformation } from '../schema-compatibility';
-import {
-  SchemaCompatLayer,
-  isArr,
-  isDate,
-  isDefault,
-  isNumber,
-  isObj,
-  isOptional,
-  isString,
-  isUnion,
-} from '../schema-compatibility';
-
-export class OpenAIReasoningSchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return `openApi3`;
-  }
-
-  isReasoningModel(): boolean {
-    // there isn't a good way to automatically detect reasoning models besides doing this.
-    // in the future when o5 is released this compat wont apply and we'll want to come back and update this class + our tests
-    return (
-      this.getModel().modelId.includes(`o3`) ||
-      this.getModel().modelId.includes(`o4`) ||
-      this.getModel().modelId.includes(`o1`)
-    );
-  }
-
-  shouldApply(): boolean {
-    if (
-      (this.getModel().supportsStructuredOutputs || this.isReasoningModel()) &&
-      (this.getModel().provider.includes(`openai`) || this.getModel().modelId.includes(`openai`))
-    ) {
-      return true;
-    }
-
-    return false;
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      const innerZodType = this.processZodType(value._def.innerType);
-      return innerZodType.nullable();
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value, { passthrough: false });
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isDefault(value)) {
-      const defaultDef = value._def;
-      const innerType = defaultDef.innerType;
-      const defaultValue = defaultDef.defaultValue();
-      const constraints: { defaultValue?: unknown } = {};
-      if (defaultValue !== undefined) {
-        constraints.defaultValue = defaultValue;
-      }
-
-      const description = this.mergeParameterDescription(value.description, constraints);
-      let result = this.processZodType(innerType);
-      if (description) {
-        result = result.describe(description);
-      }
-      return result;
-    } else if (isNumber(value)) {
-      return this.defaultZodNumberHandler(value);
-    } else if (isString(value)) {
-      return this.defaultZodStringHandler(value);
-    } else if (isDate(value)) {
-      return this.defaultZodDateHandler(value);
-    } else if (value._def.typeName === 'ZodAny') {
-      // It's bad practice in the tool to use any, it's not reasonable for models that don't support that OOTB, to cast every single possible type
-      // in the schema. Usually when it's "any" it could be a json object or a union of specific types.
-      return z
-        .string()
-        .describe(
-          (value.description ?? '') +
-            `\nArgument was an "any" type, but you (the LLM) do not support "any", so it was cast to a "string" type`,
-        );
-    }
-
-    return this.defaultUnsupportedZodTypeHandler(value);
-  }
-}

package/src/provider-compats/openai.ts

@@ -1,55 +0,0 @@
-import type { ZodTypeAny } from 'zod';
-import type { Targets } from 'zod-to-json-schema';
-import { SchemaCompatLayer, isArr, isObj, isOptional, isString, isUnion } from '../schema-compatibility';
-import type { ModelInformation, StringCheckType } from '../schema-compatibility';
-
-export class OpenAISchemaCompatLayer extends SchemaCompatLayer {
-  constructor(model: ModelInformation) {
-    super(model);
-  }
-
-  getSchemaTarget(): Targets | undefined {
-    return `jsonSchema7`;
-  }
-
-  shouldApply(): boolean {
-    if (
-      !this.getModel().supportsStructuredOutputs &&
-      (this.getModel().provider.includes(`openai`) || this.getModel().modelId.includes(`openai`))
-    ) {
-      return true;
-    }
-
-    return false;
-  }
-
-  processZodType(value: ZodTypeAny): ZodTypeAny {
-    if (isOptional(value)) {
-      return this.defaultZodOptionalHandler(value, [
-        'ZodObject',
-        'ZodArray',
-        'ZodUnion',
-        'ZodString',
-        'ZodNever',
-        'ZodUndefined',
-        'ZodTuple',
-      ]);
-    } else if (isObj(value)) {
-      return this.defaultZodObjectHandler(value);
-    } else if (isUnion(value)) {
-      return this.defaultZodUnionHandler(value);
-    } else if (isArr(value)) {
-      return this.defaultZodArrayHandler(value);
-    } else if (isString(value)) {
-      const model = this.getModel();
-      const checks: StringCheckType[] = ['emoji'];
-
-      if (model.modelId.includes('gpt-4o-mini')) {
-        checks.push('regex');
-      }
-      return this.defaultZodStringHandler(value, checks);
-    }
-
-    return this.defaultUnsupportedZodTypeHandler(value, ['ZodNever', 'ZodUndefined', 'ZodTuple']);
-  }
-}