ai 6.0.151 → 6.0.153

package/dist/internal/index.js

@@ -153,7 +153,7 @@ var import_provider_utils2 = require("@ai-sdk/provider-utils");
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.151" : "0.0.0-test";
+var VERSION = true ? "6.0.153" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({

package/dist/internal/index.mjs

@@ -133,7 +133,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.151" : "0.0.0-test";
+var VERSION = true ? "6.0.153" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "6.0.151",
+  "version": "6.0.153",
   "description": "AI SDK by Vercel - build apps like ChatGPT, Claude, Gemini, and more with a single interface for any model using the Vercel AI Gateway or go direct to OpenAI, Anthropic, Google, or any other model provider.",
   "license": "Apache-2.0",
   "sideEffects": false,

package/src/model/resolve-model.ts

@@ -5,6 +5,7 @@ import {
   ImageModelV3,
   LanguageModelV3,
   ProviderV3,
+  RerankingModelV3,
   SpeechModelV3,
   TranscriptionModelV3,
 } from '@ai-sdk/provider';
@@ -19,6 +20,7 @@ import { asLanguageModelV3 } from './as-language-model-v3';
 import { asSpeechModelV3 } from './as-speech-model-v3';
 import { asTranscriptionModelV3 } from './as-transcription-model-v3';
 import { ImageModel } from '../types/image-model';
+import { RerankingModel } from '../types/reranking-model';
 import { VideoModel } from '../types/video-model';
 
 export function resolveLanguageModel(model: LanguageModel): LanguageModelV3 {
@@ -154,6 +156,33 @@ export function resolveVideoModel(
   return model;
 }
 
+export function resolveRerankingModel(model: RerankingModel): RerankingModelV3 {
+  if (typeof model === 'string') {
+    const provider = getGlobalProvider();
+    const rerankingModel = provider.rerankingModel;
+
+    if (!rerankingModel) {
+      throw new Error(
+        'The default provider does not support reranking models. ' +
+          'Please use a RerankingModel object from a provider (e.g., gateway.rerankingModel("model-id")).',
+      );
+    }
+
+    return rerankingModel(model);
+  }
+
+  if (model.specificationVersion !== 'v3') {
+    const unsupportedModel: any = model;
+    throw new UnsupportedModelVersionError({
+      version: unsupportedModel.specificationVersion,
+      provider: unsupportedModel.provider,
+      modelId: unsupportedModel.modelId,
+    });
+  }
+
+  return model;
+}
+
 function getGlobalProvider(): ProviderV3 {
   return globalThis.AI_SDK_DEFAULT_PROVIDER ?? gateway;
 }

package/src/rerank/rerank.ts

@@ -7,6 +7,7 @@ import { getTracer } from '../telemetry/get-tracer';
 import { recordSpan } from '../telemetry/record-span';
 import { selectTelemetryAttributes } from '../telemetry/select-telemetry-attributes';
 import { TelemetrySettings } from '../telemetry/telemetry-settings';
+import { resolveRerankingModel } from '../model/resolve-model';
 import { RerankingModel } from '../types';
 import { RerankResult } from './rerank-result';
 import { logWarnings } from '../logger/log-warnings';
@@ -28,7 +29,7 @@ import { logWarnings } from '../logger/log-warnings';
  * @returns A result object that contains the reranked documents, the reranked indices, and additional information.
  */
 export async function rerank<VALUE extends JSONObject | string>({
-  model,
+  model: modelArg,
   documents,
   query,
   topN,
@@ -88,6 +89,8 @@ export async function rerank<VALUE extends JSONObject | string>({
    */
   providerOptions?: ProviderOptions;
 }): Promise<RerankResult<VALUE>> {
+  const model = resolveRerankingModel(modelArg);
+
   if (documents.length === 0) {
     return new DefaultRerankResult({
       originalDocuments: [],

package/src/types/reranking-model.ts

@@ -3,4 +3,4 @@ import { RerankingModelV3 } from '@ai-sdk/provider';
 /**
  * Reranking model that is used by the AI SDK.
  */
-export type RerankingModel = RerankingModelV3;
+export type RerankingModel = string | RerankingModelV3;