ai 7.0.0-beta.71 → 7.0.0-beta.72

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # ai
 
+## 7.0.0-beta.72
+
+### Patch Changes
+
+- 664a0eb: feat (ai/core): support plain string model IDs in `rerank()` function
+
+  The `rerank()` function now accepts plain model strings (e.g., `'cohere/rerank-v3.5'`) in addition to `RerankingModel` objects, matching the behavior of `generateText`, `embed`, and other core functions.
+
 ## 7.0.0-beta.71
 
 ### Patch Changes

package/dist/index.d.mts

@@ -185,7 +185,7 @@ type LanguageModelResponseMetadata = {
 /**
  * Reranking model that is used by the AI SDK.
  */
-type RerankingModel = RerankingModelV4 | RerankingModelV3;
+type RerankingModel = string | RerankingModelV4 | RerankingModelV3;
 
 /**
  * Provider for language, text embedding, and image models.
@@ -7138,7 +7138,7 @@ interface RerankResult<VALUE> {
  *
  * @returns A result object that contains the reranked documents, the reranked indices, and additional information.
  */
-declare function rerank<VALUE extends JSONObject | string>({ model, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, experimental_onStart: onStart, experimental_onFinish: onFinish, _internal: { generateCallId }, }: {
+declare function rerank<VALUE extends JSONObject | string>({ model: modelArg, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, experimental_onStart: onStart, experimental_onFinish: onFinish, _internal: { generateCallId }, }: {
     /**
      * The reranking model to use.
      */

package/dist/index.d.ts

@@ -185,7 +185,7 @@ type LanguageModelResponseMetadata = {
 /**
  * Reranking model that is used by the AI SDK.
  */
-type RerankingModel = RerankingModelV4 | RerankingModelV3;
+type RerankingModel = string | RerankingModelV4 | RerankingModelV3;
 
 /**
  * Provider for language, text embedding, and image models.
@@ -7138,7 +7138,7 @@ interface RerankResult<VALUE> {
  *
  * @returns A result object that contains the reranked documents, the reranked indices, and additional information.
  */
-declare function rerank<VALUE extends JSONObject | string>({ model, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, experimental_onStart: onStart, experimental_onFinish: onFinish, _internal: { generateCallId }, }: {
+declare function rerank<VALUE extends JSONObject | string>({ model: modelArg, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, experimental_onStart: onStart, experimental_onFinish: onFinish, _internal: { generateCallId }, }: {
     /**
      * The reranking model to use.
      */

package/dist/index.js

@@ -878,6 +878,20 @@ function asLanguageModelV4(model) {
   });
 }
 
+// src/model/as-reranking-model-v4.ts
+function asRerankingModelV4(model) {
+  if (model.specificationVersion === "v4") {
+    return model;
+  }
+  return new Proxy(model, {
+    get(target, prop) {
+      if (prop === "specificationVersion")
+        return "v4";
+      return target[prop];
+    }
+  });
+}
+
 // src/model/as-speech-model-v3.ts
 function asSpeechModelV3(model) {
   if (model.specificationVersion === "v3") {
@@ -976,20 +990,6 @@ function asProviderV3(provider) {
   };
 }
 
-// src/model/as-reranking-model-v4.ts
-function asRerankingModelV4(model) {
-  if (model.specificationVersion === "v4") {
-    return model;
-  }
-  return new Proxy(model, {
-    get(target, prop) {
-      if (prop === "specificationVersion")
-        return "v4";
-      return target[prop];
-    }
-  });
-}
-
 // src/model/as-provider-v4.ts
 function asProviderV4(provider) {
   if ("specificationVersion" in provider && provider.specificationVersion === "v4") {
@@ -1102,6 +1102,27 @@ function resolveVideoModel(model) {
   }
   return asVideoModelV4(model);
 }
+function resolveRerankingModel(model) {
+  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 !== "v4" && model.specificationVersion !== "v3") {
+    const unsupportedModel = model;
+    throw new UnsupportedModelVersionError({
+      version: unsupportedModel.specificationVersion,
+      provider: unsupportedModel.provider,
+      modelId: unsupportedModel.modelId
+    });
+  }
+  return asRerankingModelV4(model);
+}
 function getGlobalProvider() {
   var _a21;
   const provider = (_a21 = globalThis.AI_SDK_DEFAULT_PROVIDER) != null ? _a21 : import_gateway.gateway;
@@ -1380,7 +1401,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "7.0.0-beta.71" : "0.0.0-test";
+var VERSION = true ? "7.0.0-beta.72" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -12545,7 +12566,7 @@ var originalGenerateCallId5 = (0, import_provider_utils33.createIdGenerator)({
   size: 24
 });
 async function rerank({
-  model,
+  model: modelArg,
   documents,
   query,
   topN,
@@ -12559,6 +12580,7 @@ async function rerank({
   _internal: { generateCallId = originalGenerateCallId5 } = {}
 }) {
   var _a21, _b, _c, _d, _e;
+  const model = resolveRerankingModel(modelArg);
   const callId = generateCallId();
   const createGlobalTelemetry = getGlobalTelemetryIntegration();
   const globalTelemetry = createGlobalTelemetry({