ai 6.0.152 → 6.0.154

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # ai
 
+## 6.0.154
+
+### Patch Changes
+
+- Updated dependencies [37a378e]
+  - @ai-sdk/gateway@3.0.94
+
+## 6.0.153
+
+### Patch Changes
+
+- f152133: 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.
+
 ## 6.0.152
 
 ### Patch Changes

package/dist/index.d.mts

@@ -159,7 +159,7 @@ type LanguageModelResponseMetadata = {
 /**
  * Reranking model that is used by the AI SDK.
  */
-type RerankingModel = RerankingModelV3;
+type RerankingModel = string | RerankingModelV3;
 
 /**
  * Provider for language, text embedding, and image models.
@@ -6176,7 +6176,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, }: {
+declare function rerank<VALUE extends JSONObject | string>({ model: modelArg, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, }: {
     /**
      * The reranking model to use.
      */

package/dist/index.d.ts

@@ -159,7 +159,7 @@ type LanguageModelResponseMetadata = {
 /**
  * Reranking model that is used by the AI SDK.
  */
-type RerankingModel = RerankingModelV3;
+type RerankingModel = string | RerankingModelV3;
 
 /**
  * Provider for language, text embedding, and image models.
@@ -6176,7 +6176,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, }: {
+declare function rerank<VALUE extends JSONObject | string>({ model: modelArg, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, }: {
     /**
      * The reranking model to use.
      */

package/dist/index.js

@@ -968,6 +968,27 @@ function resolveVideoModel(model) {
   }
   return 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 !== "v3") {
+    const unsupportedModel = model;
+    throw new UnsupportedModelVersionError({
+      version: unsupportedModel.specificationVersion,
+      provider: unsupportedModel.provider,
+      modelId: unsupportedModel.modelId
+    });
+  }
+  return model;
+}
 function getGlobalProvider() {
   var _a21;
   return (_a21 = globalThis.AI_SDK_DEFAULT_PROVIDER) != null ? _a21 : import_gateway.gateway;
@@ -1231,7 +1252,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 ? "6.0.152" : "0.0.0-test";
+var VERSION = true ? "6.0.154" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -12418,7 +12439,7 @@ var DefaultProviderRegistry = class {
 
 // src/rerank/rerank.ts
 async function rerank({
-  model,
+  model: modelArg,
   documents,
   query,
   topN,
@@ -12428,6 +12449,7 @@ async function rerank({
   providerOptions,
   experimental_telemetry: telemetry
 }) {
+  const model = resolveRerankingModel(modelArg);
   if (documents.length === 0) {
     return new DefaultRerankResult({
       originalDocuments: [],