@mastra/schema-compat 0.10.7 → 0.11.0

package/src/schema-compatibility.test.ts

@@ -1,8 +1,8 @@
 import { MockLanguageModelV1 } from 'ai/test';
 import { describe, it, expect, beforeEach } from 'vitest';
-import { z } from 'zod';
-import type { ModelInformation } from './schema-compatibility';
-import { isArr, isObj, isOptional, isString, isUnion, SchemaCompatLayer } from './schema-compatibility';
+import { z } from 'zod/v3';
+import { SchemaCompatLayer } from './schema-compatibility';
+import type { ModelInformation } from './types';
 
 class MockSchemaCompatibility extends SchemaCompatLayer {
   constructor(model: ModelInformation) {
@@ -18,16 +18,16 @@ class MockSchemaCompatibility extends SchemaCompatLayer {
   }
 
   processZodType(value: z.ZodTypeAny): z.ZodTypeAny {
-    if (isObj(value)) {
+    if (this.isObj(value)) {
       return this.defaultZodObjectHandler(value);
-    } else if (isArr(value)) {
+    } else if (this.isArr(value)) {
       // For these tests, we will handle all checks by converting them to descriptions.
       return this.defaultZodArrayHandler(value, ['min', 'max', 'length']);
-    } else if (isOptional(value)) {
+    } else if (this.isOptional(value)) {
       return this.defaultZodOptionalHandler(value);
-    } else if (isUnion(value)) {
+    } else if (this.isUnion(value)) {
       return this.defaultZodUnionHandler(value);
-    } else if (isString(value)) {
+    } else if (this.isString(value)) {
       // Add a marker to confirm it was processed
       return z.string().describe(`${value.description || 'string'}:processed`);
     } else {
@@ -45,20 +45,12 @@ describe('SchemaCompatLayer', () => {
   let compatibility: MockSchemaCompatibility;
 
   beforeEach(() => {
-    compatibility = new MockSchemaCompatibility({
-      modelId: mockModel.modelId,
-      supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
-      provider: mockModel.provider,
-    });
+    compatibility = new MockSchemaCompatibility(mockModel);
   });
 
   describe('constructor and getModel', () => {
     it('should store and return the model', () => {
-      expect(compatibility.getModel()).toEqual({
-        modelId: mockModel.modelId,
-        supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
-        provider: mockModel.provider,
-      });
+      expect(compatibility.getModel()).toBe(mockModel);
     });
   });
 
@@ -359,11 +351,7 @@ describe('SchemaCompatLayer', () => {
         }
       }
 
-      const testCompat = new TestCompatibility({
-        modelId: mockModel.modelId,
-        supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
-        provider: mockModel.provider,
-      });
+      const testCompat = new TestCompatibility(mockModel);
       const result = testCompat.defaultZodOptionalHandler(optionalSchema);
 
       expect(result._def.typeName).toBe('ZodOptional');
@@ -412,11 +400,7 @@ describe('SchemaCompatLayer', () => {
           return super.processZodType(value);
         }
       }
-      const preservingCompat = new PreservingMock({
-        modelId: mockModel.modelId,
-        supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
-        provider: mockModel.provider,
-      });
+      const preservingCompat = new PreservingMock(mockModel);
       const preservingResult = preservingCompat.processToAISDKSchema(arraySchema);
       expect(preservingResult.jsonSchema.description).toBeUndefined();
       expect(