fix(provider): add specificationVersion to ProviderV3 (#9820)

Author: lgrammel

.changeset/angry-cobras-care.md

@@ -0,0 +1,35 @@
+---
+'@ai-sdk/openai-compatible': patch
+'@ai-sdk/amazon-bedrock': patch
+'@ai-sdk/google-vertex': patch
+'@ai-sdk/huggingface': patch
+'@ai-sdk/assemblyai': patch
+'@ai-sdk/elevenlabs': patch
+'@ai-sdk/perplexity': patch
+'@ai-sdk/togetherai': patch
+'@ai-sdk/anthropic': patch
+'@ai-sdk/deepinfra': patch
+'@ai-sdk/fireworks': patch
+'@ai-sdk/replicate': patch
+'@ai-sdk/cerebras': patch
+'@ai-sdk/deepgram': patch
+'@ai-sdk/deepseek': patch
+'@ai-sdk/provider': patch
+'@ai-sdk/baseten': patch
+'@ai-sdk/gateway': patch
+'@ai-sdk/mistral': patch
+'@ai-sdk/cohere': patch
+'@ai-sdk/gladia': patch
+'@ai-sdk/google': patch
+'@ai-sdk/openai': patch
+'@ai-sdk/vercel': patch
+'@ai-sdk/azure': patch
+'@ai-sdk/revai': patch
+'@ai-sdk/groq': patch
+'@ai-sdk/luma': patch
+'@ai-sdk/fal': patch
+'@ai-sdk/xai': patch
+'ai': patch
+---
+
+fix(provider): add specificationVersion to ProviderV3

packages/ai/src/middleware/wrap-provider.ts

@@ -1,6 +1,7 @@
-import type { LanguageModelV3, ProviderV2, ProviderV3 } from '@ai-sdk/provider';
+import type { ProviderV2, ProviderV3 } from '@ai-sdk/provider';
 import { LanguageModelMiddleware } from '../types/language-model-middleware';
 import { wrapLanguageModel } from './wrap-language-model';
+import { asProviderV3 } from '../model/as-provider-v3';
 
 /**
  * Wraps a ProviderV3 instance with middleware functionality.
@@ -20,20 +21,17 @@ export function wrapProvider({
   provider: ProviderV3 | ProviderV2;
   languageModelMiddleware: LanguageModelMiddleware | LanguageModelMiddleware[];
 }): ProviderV3 {
-  const wrappedProvider = {
-    languageModel(modelId: string) {
-      let model = provider.languageModel(modelId);
-      model = wrapLanguageModel({
-        model: model as LanguageModelV3,
+  const providerV3 = asProviderV3(provider);
+  return {
+    specificationVersion: 'v3',
+    languageModel: (modelId: string) =>
+      wrapLanguageModel({
+        model: providerV3.languageModel(modelId),
         middleware: languageModelMiddleware,
-      });
-      return model;
-    },
-    textEmbeddingModel: provider.textEmbeddingModel,
-    imageModel: provider.imageModel,
-    transcriptionModel: provider.transcriptionModel,
-    speechModel: provider.speechModel,
+      }),
+    textEmbeddingModel: providerV3.textEmbeddingModel,
+    imageModel: providerV3.imageModel,
+    transcriptionModel: providerV3.transcriptionModel,
+    speechModel: providerV3.speechModel,
   };
-
-  return wrappedProvider as ProviderV3;
 }

packages/ai/src/model/as-embedding-model-v3.ts

@@ -0,0 +1,18 @@
+import { EmbeddingModelV2, EmbeddingModelV3 } from '@ai-sdk/provider';
+
+export function asEmbeddingModelV3<VALUE>(
+  model: EmbeddingModelV2<VALUE> | EmbeddingModelV3<VALUE>,
+): EmbeddingModelV3<VALUE> {
+  if (model.specificationVersion === 'v3') {
+    return model;
+  }
+
+  // TODO this could break, we need to properly map v2 to v3
+  // and support all relevant v3 properties:
+  return new Proxy(model, {
+    get(target, prop: keyof EmbeddingModelV2<VALUE>) {
+      if (prop === 'specificationVersion') return 'v3';
+      return target[prop];
+    },
+  }) as unknown as EmbeddingModelV3<VALUE>;
+}

packages/ai/src/model/as-image-model-v3.ts

@@ -0,0 +1,18 @@
+import { ImageModelV2, ImageModelV3 } from '@ai-sdk/provider';
+
+export function asImageModelV3(
+  model: ImageModelV2 | ImageModelV3,
+): ImageModelV3 {
+  if (model.specificationVersion === 'v3') {
+    return model;
+  }
+
+  // TODO this could break, we need to properly map v2 to v3
+  // and support all relevant v3 properties:
+  return new Proxy(model, {
+    get(target, prop: keyof ImageModelV2) {
+      if (prop === 'specificationVersion') return 'v3';
+      return target[prop];
+    },
+  }) as unknown as ImageModelV3;
+}

packages/ai/src/model/as-language-model-v3.ts

@@ -0,0 +1,18 @@
+import { LanguageModelV2, LanguageModelV3 } from '@ai-sdk/provider';
+
+export function asLanguageModelV3(
+  model: LanguageModelV2 | LanguageModelV3,
+): LanguageModelV3 {
+  if (model.specificationVersion === 'v3') {
+    return model;
+  }
+
+  // TODO this could break, we need to properly map v2 to v3
+  // and support all relevant v3 properties:
+  return new Proxy(model, {
+    get(target, prop: keyof LanguageModelV2) {
+      if (prop === 'specificationVersion') return 'v3';
+      return target[prop];
+    },
+  }) as unknown as LanguageModelV3;
+}

packages/ai/src/model/as-provider-v3.ts

@@ -0,0 +1,32 @@
+import { ProviderV2, ProviderV3 } from '@ai-sdk/provider';
+import { asEmbeddingModelV3 } from './as-embedding-model-v3';
+import { asImageModelV3 } from './as-image-model-v3';
+import { asLanguageModelV3 } from './as-language-model-v3';
+import { asTranscriptionModelV3 } from './as-transcription-model-v3';
+import { asSpeechModelV3 } from './as-speech-model-v3';
+
+export function asProviderV3(provider: ProviderV2 | ProviderV3): ProviderV3 {
+  if (
+    'specificationVersion' in provider &&
+    provider.specificationVersion === 'v3'
+  ) {
+    return provider;
+  }
+
+  return {
+    specificationVersion: 'v3',
+    languageModel: (modelId: string) =>
+      asLanguageModelV3(provider.languageModel(modelId)),
+    textEmbeddingModel: (modelId: string) =>
+      asEmbeddingModelV3(provider.textEmbeddingModel(modelId)),
+    imageModel: (modelId: string) =>
+      asImageModelV3(provider.imageModel(modelId)),
+    transcriptionModel: provider.transcriptionModel
+      ? (modelId: string) =>
+          asTranscriptionModelV3(provider.transcriptionModel!(modelId))
+      : undefined,
+    speechModel: provider.speechModel
+      ? (modelId: string) => asSpeechModelV3(provider.speechModel!(modelId))
+      : undefined,
+  };
+}

packages/ai/src/model/as-speech-model-v3.ts

@@ -0,0 +1,18 @@
+import { SpeechModelV2, SpeechModelV3 } from '@ai-sdk/provider';
+
+export function asSpeechModelV3(
+  model: SpeechModelV3 | SpeechModelV2,
+): SpeechModelV3 {
+  if (model.specificationVersion === 'v3') {
+    return model;
+  }
+
+  // TODO this could break, we need to properly map v2 to v3
+  // and support all relevant v3 properties:
+  return new Proxy(model, {
+    get(target, prop: keyof SpeechModelV2) {
+      if (prop === 'specificationVersion') return 'v3';
+      return target[prop];
+    },
+  }) as unknown as SpeechModelV3;
+}

packages/ai/src/model/as-transcription-model-v3.ts

@@ -0,0 +1,18 @@
+import { TranscriptionModelV2, TranscriptionModelV3 } from '@ai-sdk/provider';
+
+export function asTranscriptionModelV3(
+  model: TranscriptionModelV3 | TranscriptionModelV2,
+): TranscriptionModelV3 {
+  if (model.specificationVersion === 'v3') {
+    return model;
+  }
+
+  // TODO this could break, we need to properly map v2 to v3
+  // and support all relevant v3 properties:
+  return new Proxy(model, {
+    get(target, prop: keyof TranscriptionModelV2) {
+      if (prop === 'specificationVersion') return 'v3';
+      return target[prop];
+    },
+  }) as unknown as TranscriptionModelV3;
+}

packages/ai/src/model/resolve-model.ts

@@ -1,60 +1,20 @@
 import { gateway } from '@ai-sdk/gateway';
 import {
-  EmbeddingModelV2,
   EmbeddingModelV3,
-  LanguageModelV2,
   LanguageModelV3,
   ProviderV3,
-  SpeechModelV2,
   SpeechModelV3,
-  TranscriptionModelV2,
   TranscriptionModelV3,
 } from '@ai-sdk/provider';
 import { UnsupportedModelVersionError } from '../error';
 import { EmbeddingModel } from '../types/embedding-model';
 import { LanguageModel } from '../types/language-model';
 import { SpeechModel } from '../types/speech-model';
 import { TranscriptionModel } from '../types/transcription-model';
-
-function transformToV3LanguageModel(model: LanguageModelV2): LanguageModelV3 {
-  return new Proxy(model, {
-    get(target, prop: keyof LanguageModelV2) {
-      if (prop === 'specificationVersion') return 'v3';
-      return target[prop];
-    },
-  }) as unknown as LanguageModelV3;
-}
-
-function transformToV3EmbeddingModel<VALUE>(
-  model: EmbeddingModelV2<VALUE>,
-): EmbeddingModelV3<VALUE> {
-  return new Proxy(model, {
-    get(target, prop: keyof EmbeddingModelV2<VALUE>) {
-      if (prop === 'specificationVersion') return 'v3';
-      return target[prop];
-    },
-  }) as unknown as EmbeddingModelV3<VALUE>;
-}
-
-function transformToV3TranscriptionModel(
-  model: TranscriptionModelV2,
-): TranscriptionModelV3 {
-  return new Proxy(model, {
-    get(target, prop: keyof TranscriptionModelV2) {
-      if (prop === 'specificationVersion') return 'v3';
-      return target[prop];
-    },
-  }) as unknown as TranscriptionModelV3;
-}
-
-function transformToV3SpeechModel(model: SpeechModelV2): SpeechModelV3 {
-  return new Proxy(model, {
-    get(target, prop: keyof SpeechModelV2) {
-      if (prop === 'specificationVersion') return 'v3';
-      return target[prop];
-    },
-  }) as unknown as SpeechModelV3;
-}
+import { asEmbeddingModelV3 } from './as-embedding-model-v3';
+import { asLanguageModelV3 } from './as-language-model-v3';
+import { asSpeechModelV3 } from './as-speech-model-v3';
+import { asTranscriptionModelV3 } from './as-transcription-model-v3';
 
 export function resolveLanguageModel(model: LanguageModel): LanguageModelV3 {
   if (typeof model !== 'string') {
@@ -69,10 +29,8 @@ export function resolveLanguageModel(model: LanguageModel): LanguageModelV3 {
         modelId: unsupportedModel.modelId,
       });
     }
-    if (model.specificationVersion === 'v2') {
-      return transformToV3LanguageModel(model);
-    }
-    return model;
+
+    return asLanguageModelV3(model);
   }
 
   return getGlobalProvider().languageModel(model);
@@ -93,11 +51,8 @@ export function resolveEmbeddingModel<VALUE = string>(
         modelId: unsupportedModel.modelId,
       });
     }
-    if (model.specificationVersion === 'v2') {
-      return transformToV3EmbeddingModel(model);
-    }
 
-    return model;
+    return asEmbeddingModelV3(model);
   }
 
   // TODO AI SDK 6: figure out how to cleanly support different generic types
@@ -121,10 +76,7 @@ export function resolveTranscriptionModel(
         modelId: unsupportedModel.modelId,
       });
     }
-    if (model.specificationVersion === 'v2') {
-      return transformToV3TranscriptionModel(model);
-    }
-    return model;
+    return asTranscriptionModelV3(model);
   }
 
   return getGlobalProvider().transcriptionModel?.(model);
@@ -145,10 +97,7 @@ export function resolveSpeechModel(
         modelId: unsupportedModel.modelId,
       });
     }
-    if (model.specificationVersion === 'v2') {
-      return transformToV3SpeechModel(model);
-    }
-    return model;
+    return asSpeechModelV3(model);
   }
 
   return getGlobalProvider().speechModel?.(model);

packages/ai/src/registry/custom-provider.ts

@@ -55,6 +55,7 @@ export function customProvider<
   speechModel(modelId: ExtractModelId<SPEECH_MODELS>): SpeechModelV3;
 } {
   return {
+    specificationVersion: 'v3',
     languageModel(modelId: ExtractModelId<LANGUAGE_MODELS>): LanguageModelV3 {
       if (languageModels != null && modelId in languageModels) {
         return languageModels[modelId];