@mastra/schema-compat 1.2.9-alpha.1 → 1.2.10-alpha.0

package/dist/index.cjs

@@ -10,7 +10,7 @@ var zodFromJsonSchema = require('zod-from-json-schema');
 var zodFromJsonSchemaV3 = require('zod-from-json-schema-v3');
 var v4 = require('zod/v4');
 
-// ../_vendored/ai_v4/dist/chunk-OPIPXJLE.js
+// ../_vendored/ai_v4/dist/chunk-HB6FLTHO.js
 var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -5905,56 +5905,62 @@ var DeepSeekSchemaCompatLayer = class extends SchemaCompatLayer3 {
     return schema;
   }
 };
-function fixNullableUnionTypes(schema) {
+function fixAISDKNullableUnionTypes(schema) {
   if (typeof schema !== "object" || schema === null) {
     return schema;
   }
   const result = { ...schema };
   if (Array.isArray(result.type)) {
     const nonNullTypes = result.type.filter((t) => t !== "null");
-    if (nonNullTypes.length < result.type.length) {
+    if (nonNullTypes.length === 1) {
+      result.type = nonNullTypes[0];
       result.nullable = true;
-      if (nonNullTypes.length === 1) {
-        result.type = nonNullTypes[0];
-      } else if (nonNullTypes.length > 1) {
-        result.type = nonNullTypes;
-      } else {
-        delete result.type;
-      }
+    } else {
+      delete result.type;
+      delete result.nullable;
     }
   }
-  if (result.anyOf && Array.isArray(result.anyOf) && result.anyOf.length === 2) {
+  if (Array.isArray(result.enum) && result.enum.some((value) => typeof value !== "string")) {
+    delete result.enum;
+  }
+  if ("const" in result && typeof result.const !== "string") {
+    delete result.const;
+  }
+  if (result.anyOf && Array.isArray(result.anyOf)) {
     const nullSchema = result.anyOf.find((s) => typeof s === "object" && s !== null && s.type === "null");
-    const otherSchema = result.anyOf.find((s) => typeof s === "object" && s !== null && s.type !== "null");
-    if (nullSchema && otherSchema && typeof otherSchema === "object") {
+    const nonNullSchemas = result.anyOf.filter((s) => !(typeof s === "object" && s !== null && s.type === "null"));
+    if (nullSchema) {
       const { anyOf: _, ...rest } = result;
-      const fixedOther = fixNullableUnionTypes(otherSchema);
-      return { ...rest, ...fixedOther, nullable: true };
+      if (nonNullSchemas.length === 1 && typeof nonNullSchemas[0] === "object" && nonNullSchemas[0] !== null) {
+        const fixedOther = fixAISDKNullableUnionTypes(nonNullSchemas[0]);
+        return fixedOther.type ? { ...rest, ...fixedOther, nullable: true } : { ...rest, ...fixedOther };
+      }
+      return rest;
     }
   }
   if (result.properties && typeof result.properties === "object") {
     result.properties = Object.fromEntries(
-      Object.entries(result.properties).map(([key, value]) => [key, fixNullableUnionTypes(value)])
+      Object.entries(result.properties).map(([key, value]) => [key, fixAISDKNullableUnionTypes(value)])
     );
   }
   if (result.items) {
     if (Array.isArray(result.items)) {
-      result.items = result.items.map((item) => fixNullableUnionTypes(item));
+      result.items = result.items.map((item) => fixAISDKNullableUnionTypes(item));
     } else {
-      result.items = fixNullableUnionTypes(result.items);
+      result.items = fixAISDKNullableUnionTypes(result.items);
     }
   }
   if (result.additionalProperties && typeof result.additionalProperties === "object") {
-    result.additionalProperties = fixNullableUnionTypes(result.additionalProperties);
+    result.additionalProperties = fixAISDKNullableUnionTypes(result.additionalProperties);
   }
   if (result.anyOf && Array.isArray(result.anyOf)) {
-    result.anyOf = result.anyOf.map((s) => fixNullableUnionTypes(s));
+    result.anyOf = result.anyOf.map((s) => fixAISDKNullableUnionTypes(s));
   }
   if (result.oneOf && Array.isArray(result.oneOf)) {
-    result.oneOf = result.oneOf.map((s) => fixNullableUnionTypes(s));
+    result.oneOf = result.oneOf.map((s) => fixAISDKNullableUnionTypes(s));
   }
   if (result.allOf && Array.isArray(result.allOf)) {
-    result.allOf = result.allOf.map((s) => fixNullableUnionTypes(s));
+    result.allOf = result.allOf.map((s) => fixAISDKNullableUnionTypes(s));
   }
   if (result.anyOf && Array.isArray(result.anyOf)) {
     if (result.description) {
@@ -6007,15 +6013,13 @@ var GoogleSchemaCompatLayer = class extends SchemaCompatLayer3 {
     }
     return this.defaultUnsupportedZodTypeHandler(value);
   }
-  // public processToJSONSchema(zodSchema: PublicSchema<any>, io?: 'input' | 'output'): JSONSchema7 {
-  //   const result = super.processToJSONSchema(zodSchema, io);
-  //   // Fix union type arrays that Gemini doesn't support
-  //   return fixNullableUnionTypes(result as Record<string, any>) as JSONSchema7;
-  // }
+  processToJSONSchema(schema, io) {
+    return super.processToJSONSchema(schema, io);
+  }
   processToAISDKSchema(zodSchema2) {
     const compat = this.processToCompatSchema(zodSchema2);
     const transformedJsonSchema = chunkQ6XZFGLX_cjs.standardSchemaToJSONSchema(compat);
-    const fixedJsonSchema = fixNullableUnionTypes(transformedJsonSchema);
+    const fixedJsonSchema = fixAISDKNullableUnionTypes(transformedJsonSchema);
     return jsonSchema(fixedJsonSchema, {
       validate: (value) => {
         const transformed = this.#traverse(value, fixedJsonSchema);