@mastra/schema-compat 0.10.6 → 0.10.7-alpha.0

package/src/schema-compatibility.ts

@@ -1,4 +1,4 @@
-import type { Schema, LanguageModelV1 } from 'ai';
+import type { Schema } from 'ai';
 import type { JSONSchema7 } from 'json-schema';
 import { z, ZodOptional, ZodObject, ZodArray, ZodUnion, ZodString, ZodNumber, ZodDate, ZodDefault, ZodNull } from 'zod';
 import type { ZodTypeAny } from 'zod';
@@ -142,6 +142,12 @@ type DateConstraints = {
   dateFormat?: string;
 };
 
+export type ModelInformation = {
+  modelId: string;
+  provider: string;
+  supportsStructuredOutputs: boolean;
+};
+
 /**
  * Abstract base class for creating schema compatibility layers for different AI model providers.
  *
@@ -182,14 +188,14 @@ type DateConstraints = {
  * ```
  */
 export abstract class SchemaCompatLayer {
-  private model: LanguageModelV1;
+  private model: ModelInformation;
 
   /**
    * Creates a new schema compatibility instance.
    *
    * @param model - The language model this compatibility layer applies to
    */
-  constructor(model: LanguageModelV1) {
+  constructor(model: ModelInformation) {
     this.model = model;
   }
 
@@ -198,7 +204,7 @@ export abstract class SchemaCompatLayer {
    *
    * @returns The language model instance
    */
-  getModel(): LanguageModelV1 {
+  getModel(): ModelInformation {
     return this.model;
   }
 

package/src/utils.test.ts

@@ -1,8 +1,9 @@
 import { jsonSchema } from 'ai';
-import type { LanguageModelV1, Schema } from 'ai';
+import type { Schema } from 'ai';
 import { MockLanguageModelV1 } from 'ai/test';
 import { describe, it, expect, beforeEach, vi } from 'vitest';
 import { z } from 'zod';
+import type { ModelInformation } from './schema-compatibility';
 import { SchemaCompatLayer } from './schema-compatibility';
 import { convertZodSchemaToAISDKSchema, convertSchemaToZod, applyCompatLayer } from './utils';
 
@@ -13,7 +14,7 @@ const mockModel = new MockLanguageModelV1({
 
 class MockSchemaCompatibility extends SchemaCompatLayer {
   constructor(
-    model: LanguageModelV1,
+    model: ModelInformation,
     private shouldApplyValue: boolean = true,
   ) {
     super(model);
@@ -191,7 +192,11 @@ describe('Builder Functions', () => {
     let mockCompatibility: MockSchemaCompatibility;
 
     beforeEach(() => {
-      mockCompatibility = new MockSchemaCompatibility(mockModel);
+      mockCompatibility = new MockSchemaCompatibility({
+        modelId: mockModel.modelId,
+        supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
+        provider: mockModel.provider,
+      });
     });
 
     it('should process Zod object schema with compatibility', () => {
@@ -257,7 +262,14 @@ describe('Builder Functions', () => {
     });
 
     it('should return fallback when no compatibility applies', () => {
-      const nonApplyingCompatibility = new MockSchemaCompatibility(mockModel, false);
+      const nonApplyingCompatibility = new MockSchemaCompatibility(
+        {
+          modelId: mockModel.modelId,
+          supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
+          provider: mockModel.provider,
+        },
+        false,
+      );
       const zodSchema = z.object({
         name: z.string(),
       });
@@ -303,8 +315,22 @@ describe('Builder Functions', () => {
     });
 
     it('should handle complex schema with multiple compatLayers', () => {
-      const compat1 = new MockSchemaCompatibility(mockModel, false);
-      const compat2 = new MockSchemaCompatibility(mockModel, true);
+      const compat1 = new MockSchemaCompatibility(
+        {
+          modelId: mockModel.modelId,
+          supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? true,
+          provider: mockModel.provider,
+        },
+        false,
+      );
+      const compat2 = new MockSchemaCompatibility(
+        {
+          modelId: mockModel.modelId,
+          supportsStructuredOutputs: mockModel.supportsStructuredOutputs ?? false,
+          provider: mockModel.provider,
+        },
+        true,
+      );
 
       vi.spyOn(compat1, 'processZodType');
       vi.spyOn(compat2, 'processZodType');