@ai-sdk/amazon-bedrock 4.0.0-beta.46 → 4.0.0-beta.48

package/dist/index.mjs

@@ -1,4 +1,5 @@
 // src/bedrock-provider.ts
+import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 import {
   generateId,
   loadOptionalSetting,
@@ -7,12 +8,6 @@ import {
   withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
 
-// src/version.ts
-var VERSION = true ? "4.0.0-beta.46" : "0.0.0-test";
-
-// src/bedrock-provider.ts
-import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
-
 // src/bedrock-chat-language-model.ts
 import {
   combineHeaders,
@@ -1529,6 +1524,11 @@ import {
   getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 import { AwsV4Signer } from "aws4fetch";
+
+// src/version.ts
+var VERSION = true ? "4.0.0-beta.48" : "0.0.0-test";
+
+// src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
   return async (input, init) => {
     var _a, _b;
@@ -1609,6 +1609,130 @@ function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
   };
 }
 
+// src/reranking/bedrock-reranking-model.ts
+import {
+  combineHeaders as combineHeaders5,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler4,
+  createJsonResponseHandler as createJsonResponseHandler4,
+  parseProviderOptions as parseProviderOptions4,
+  postJsonToApi as postJsonToApi4,
+  resolve as resolve4
+} from "@ai-sdk/provider-utils";
+
+// src/reranking/bedrock-reranking-api.ts
+import { lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z as z7 } from "zod/v4";
+var bedrockRerankingResponseSchema = lazySchema(
+  () => zodSchema(
+    z7.object({
+      results: z7.array(
+        z7.object({
+          index: z7.number(),
+          relevanceScore: z7.number()
+        })
+      ),
+      nextToken: z7.string().optional()
+    })
+  )
+);
+
+// src/reranking/bedrock-reranking-options.ts
+import { lazySchema as lazySchema2, zodSchema as zodSchema2 } from "@ai-sdk/provider-utils";
+import { z as z8 } from "zod/v4";
+var bedrockRerankingOptionsSchema = lazySchema2(
+  () => zodSchema2(
+    z8.object({
+      /**
+       * If the total number of results was greater than could fit in a response, a token is returned in the nextToken field. You can enter that token in this field to return the next batch of results.
+       */
+      nextToken: z8.string().optional(),
+      /**
+       * Additional model request fields to pass to the model.
+       */
+      additionalModelRequestFields: z8.record(z8.string(), z8.any()).optional()
+    })
+  )
+);
+
+// src/reranking/bedrock-reranking-model.ts
+var BedrockRerankingModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+    this.provider = "amazon-bedrock";
+  }
+  async doRerank({
+    documents,
+    headers,
+    query,
+    topN,
+    abortSignal,
+    providerOptions
+  }) {
+    const bedrockOptions = await parseProviderOptions4({
+      provider: "bedrock",
+      providerOptions,
+      schema: bedrockRerankingOptionsSchema
+    });
+    const {
+      value: response,
+      responseHeaders,
+      rawValue
+    } = await postJsonToApi4({
+      url: `${this.config.baseUrl()}/rerank`,
+      headers: await resolve4(
+        combineHeaders5(await resolve4(this.config.headers), headers)
+      ),
+      body: {
+        nextToken: bedrockOptions == null ? void 0 : bedrockOptions.nextToken,
+        queries: [
+          {
+            textQuery: { text: query },
+            type: "TEXT"
+          }
+        ],
+        rerankingConfiguration: {
+          bedrockRerankingConfiguration: {
+            modelConfiguration: {
+              modelArn: `arn:aws:bedrock:${this.config.region}::foundation-model/${this.modelId}`,
+              additionalModelRequestFields: bedrockOptions == null ? void 0 : bedrockOptions.additionalModelRequestFields
+            },
+            numberOfResults: topN
+          },
+          type: "BEDROCK_RERANKING_MODEL"
+        },
+        sources: documents.values.map((value) => ({
+          type: "INLINE",
+          inlineDocumentSource: documents.type === "text" ? {
+            type: "TEXT",
+            textDocument: { text: value }
+          } : {
+            type: "JSON",
+            jsonDocument: value
+          }
+        }))
+      },
+      failedResponseHandler: createJsonErrorResponseHandler4({
+        errorSchema: BedrockErrorSchema,
+        errorToMessage: (error) => `${error.type}: ${error.message}`
+      }),
+      successfulResponseHandler: createJsonResponseHandler4(
+        bedrockRerankingResponseSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal
+    });
+    return {
+      ranking: response.results,
+      response: {
+        headers: responseHeaders,
+        body: rawValue
+      }
+    };
+  }
+};
+
 // src/bedrock-provider.ts
 function createAmazonBedrock(options = {}) {
   const rawApiKey = loadOptionalSetting({
@@ -1677,7 +1801,12 @@ Original error: ${errorMessage}`
       throw error;
     }
   }, options.fetch);
-  const getBaseUrl = () => {
+  const getHeaders = () => {
+    var _a;
+    const baseHeaders = (_a = options.headers) != null ? _a : {};
+    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
+  };
+  const getBedrockRuntimeBaseUrl = () => {
     var _a, _b;
     return (_b = withoutTrailingSlash(
       (_a = options.baseURL) != null ? _a : `https://bedrock-runtime.${loadSetting({
@@ -1688,13 +1817,19 @@ Original error: ${errorMessage}`
       })}.amazonaws.com`
     )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
   };
-  const getHeaders = () => {
-    var _a;
-    const baseHeaders = (_a = options.headers) != null ? _a : {};
-    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
+  const getBedrockAgentRuntimeBaseUrl = () => {
+    var _a, _b;
+    return (_b = withoutTrailingSlash(
+      (_a = options.baseURL) != null ? _a : `https://bedrock-agent-runtime.${loadSetting({
+        settingValue: options.region,
+        settingName: "region",
+        environmentVariableName: "AWS_REGION",
+        description: "AWS region"
+      })}.amazonaws.com`
+    )) != null ? _b : `https://bedrock-agent-runtime.us-west-2.amazonaws.com`;
   };
   const createChatModel = (modelId) => new BedrockChatLanguageModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,
     fetch: fetchFunction,
     generateId
@@ -1708,12 +1843,23 @@ Original error: ${errorMessage}`
     return createChatModel(modelId);
   };
   const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,
     fetch: fetchFunction
   });
   const createImageModel = (modelId) => new BedrockImageModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
+  const createRerankingModel = (modelId) => new BedrockRerankingModel(modelId, {
+    baseUrl: getBedrockAgentRuntimeBaseUrl,
+    region: loadSetting({
+      settingValue: options.region,
+      settingName: "region",
+      environmentVariableName: "AWS_REGION",
+      description: "AWS region"
+    }),
     headers: getHeaders,
     fetch: fetchFunction
   });
@@ -1724,6 +1870,8 @@ Original error: ${errorMessage}`
   provider.textEmbeddingModel = createEmbeddingModel;
   provider.image = createImageModel;
   provider.imageModel = createImageModel;
+  provider.reranking = createRerankingModel;
+  provider.rerankingModel = createRerankingModel;
   provider.tools = anthropicTools2;
   return provider;
 }