ai 6.0.0-beta.78 → 6.0.0-beta.79

package/dist/index.js

@@ -118,6 +118,7 @@ __export(src_exports, {
   pipeUIMessageStreamToResponse: () => pipeUIMessageStreamToResponse,
   pruneMessages: () => pruneMessages,
   readUIMessageStream: () => readUIMessageStream,
+  rerank: () => rerank,
   safeValidateUIMessages: () => safeValidateUIMessages,
   simulateReadableStream: () => simulateReadableStream,
   simulateStreamingMiddleware: () => simulateStreamingMiddleware,
@@ -874,7 +875,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.78" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.79" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -9915,7 +9916,9 @@ function asProviderV3(provider) {
     textEmbeddingModel: (modelId) => asEmbeddingModelV3(provider.textEmbeddingModel(modelId)),
     imageModel: (modelId) => asImageModelV3(provider.imageModel(modelId)),
     transcriptionModel: provider.transcriptionModel ? (modelId) => asTranscriptionModelV3(provider.transcriptionModel(modelId)) : void 0,
-    speechModel: provider.speechModel ? (modelId) => asSpeechModelV3(provider.speechModel(modelId)) : void 0
+    speechModel: provider.speechModel ? (modelId) => asSpeechModelV3(provider.speechModel(modelId)) : void 0,
+    rerankingModel: void 0
+    // v2 providers don't have reranking models
   };
 }
 
@@ -9934,7 +9937,8 @@ function wrapProvider({
     textEmbeddingModel: providerV3.textEmbeddingModel,
     imageModel: providerV3.imageModel,
     transcriptionModel: providerV3.transcriptionModel,
-    speechModel: providerV3.speechModel
+    speechModel: providerV3.speechModel,
+    rerankingModel: providerV3.rerankingModel
   };
 }
 
@@ -9946,8 +9950,10 @@ function customProvider({
   imageModels,
   transcriptionModels,
   speechModels,
-  fallbackProvider
+  rerankingModels,
+  fallbackProvider: fallbackProviderArg
 }) {
+  const fallbackProvider = fallbackProviderArg ? asProviderV3(fallbackProviderArg) : void 0;
   return {
     specificationVersion: "v3",
     languageModel(modelId) {
@@ -9994,6 +10000,15 @@ function customProvider({
         return fallbackProvider.speechModel(modelId);
       }
       throw new import_provider29.NoSuchModelError({ modelId, modelType: "speechModel" });
+    },
+    rerankingModel(modelId) {
+      if (rerankingModels != null && modelId in rerankingModels) {
+        return rerankingModels[modelId];
+      }
+      if (fallbackProvider == null ? void 0 : fallbackProvider.rerankingModel) {
+        return fallbackProvider.rerankingModel(modelId);
+      }
+      throw new import_provider29.NoSuchModelError({ modelId, modelType: "rerankingModel" });
     }
   };
 }
@@ -10142,6 +10157,150 @@ var DefaultProviderRegistry = class {
     }
     return model;
   }
+  rerankingModel(id) {
+    var _a16;
+    const [providerId, modelId] = this.splitId(id, "rerankingModel");
+    const provider = this.getProvider(providerId, "rerankingModel");
+    const model = (_a16 = provider.rerankingModel) == null ? void 0 : _a16.call(provider, modelId);
+    if (model == null) {
+      throw new import_provider31.NoSuchModelError({ modelId: id, modelType: "rerankingModel" });
+    }
+    return model;
+  }
+};
+
+// src/rerank/rerank.ts
+async function rerank({
+  model,
+  documents,
+  query,
+  topN,
+  maxRetries: maxRetriesArg,
+  abortSignal,
+  headers,
+  providerOptions,
+  experimental_telemetry: telemetry
+}) {
+  if (documents.length === 0) {
+    return new DefaultRerankResult({
+      originalDocuments: [],
+      ranking: [],
+      providerMetadata: void 0,
+      response: {
+        timestamp: /* @__PURE__ */ new Date(),
+        modelId: model.modelId
+      }
+    });
+  }
+  const { maxRetries, retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal
+  });
+  const documentsToSend = typeof documents[0] === "string" ? { type: "text", values: documents } : { type: "object", values: documents };
+  const baseTelemetryAttributes = getBaseTelemetryAttributes({
+    model,
+    telemetry,
+    headers,
+    settings: { maxRetries }
+  });
+  const tracer = getTracer(telemetry);
+  return recordSpan({
+    name: "ai.rerank",
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...assembleOperationName({ operationId: "ai.rerank", telemetry }),
+        ...baseTelemetryAttributes,
+        "ai.documents": {
+          input: () => documents.map((document) => JSON.stringify(document))
+        }
+      }
+    }),
+    tracer,
+    fn: async () => {
+      var _a16, _b;
+      const { ranking, response, providerMetadata, warnings } = await retry(
+        () => recordSpan({
+          name: "ai.rerank.doRerank",
+          attributes: selectTelemetryAttributes({
+            telemetry,
+            attributes: {
+              ...assembleOperationName({
+                operationId: "ai.rerank.doRerank",
+                telemetry
+              }),
+              ...baseTelemetryAttributes,
+              // specific settings that only make sense on the outer level:
+              "ai.documents": {
+                input: () => documents.map((document) => JSON.stringify(document))
+              }
+            }
+          }),
+          tracer,
+          fn: async (doRerankSpan) => {
+            const modelResponse = await model.doRerank({
+              documents: documentsToSend,
+              query,
+              topN,
+              providerOptions,
+              abortSignal,
+              headers
+            });
+            const ranking2 = modelResponse.ranking;
+            doRerankSpan.setAttributes(
+              await selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.ranking.type": documentsToSend.type,
+                  "ai.ranking": {
+                    output: () => ranking2.map((ranking3) => JSON.stringify(ranking3))
+                  }
+                }
+              })
+            );
+            return {
+              ranking: ranking2,
+              providerMetadata: modelResponse.providerMetadata,
+              response: modelResponse.response,
+              warnings: modelResponse.warnings
+            };
+          }
+        })
+      );
+      logWarnings({
+        warnings: warnings != null ? warnings : [],
+        provider: model.provider,
+        model: model.modelId
+      });
+      return new DefaultRerankResult({
+        originalDocuments: documents,
+        ranking: ranking.map((ranking2) => ({
+          originalIndex: ranking2.index,
+          relevanceScore: ranking2.relevanceScore,
+          document: documents[ranking2.index]
+        })),
+        providerMetadata,
+        response: {
+          id: response == null ? void 0 : response.id,
+          timestamp: (_a16 = response == null ? void 0 : response.timestamp) != null ? _a16 : /* @__PURE__ */ new Date(),
+          modelId: (_b = response == null ? void 0 : response.modelId) != null ? _b : model.modelId,
+          headers: response == null ? void 0 : response.headers,
+          body: response == null ? void 0 : response.body
+        }
+      });
+    }
+  });
+}
+var DefaultRerankResult = class {
+  constructor(options) {
+    this.originalDocuments = options.originalDocuments;
+    this.ranking = options.ranking;
+    this.response = options.response;
+    this.providerMetadata = options.providerMetadata;
+  }
+  get rerankedDocuments() {
+    return this.ranking.map((ranking) => ranking.document);
+  }
 };
 
 // src/transcribe/transcribe.ts
@@ -11072,6 +11231,7 @@ var TextStreamChatTransport = class extends HttpChatTransport {
   pipeUIMessageStreamToResponse,
   pruneMessages,
   readUIMessageStream,
+  rerank,
   safeValidateUIMessages,
   simulateReadableStream,
   simulateStreamingMiddleware,