@ai-sdk/amazon-bedrock 3.1.0-beta.1 → 3.1.0-beta.10

package/dist/index.mjs

@@ -3,8 +3,14 @@ import {
   generateId,
   loadOptionalSetting,
   loadSetting,
-  withoutTrailingSlash
+  withoutTrailingSlash,
+  withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
+
+// src/version.ts
+var VERSION = true ? "3.1.0-beta.10" : "0.0.0-test";
+
+// src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 
 // src/bedrock-chat-language-model.ts
@@ -54,6 +60,18 @@ var BEDROCK_DOCUMENT_MIME_TYPES = {
 
 // src/bedrock-chat-options.ts
 import { z } from "zod/v4";
+var bedrockFilePartProviderOptions = z.object({
+  /**
+   * Citation configuration for this document.
+   * When enabled, this document will generate citations in the response.
+   */
+  citations: z.object({
+    /**
+     * Enable citations for this document
+     */
+    enabled: z.boolean()
+  }).optional()
+});
 var bedrockProviderOptions = z.object({
   /**
    * Additional inference parameters that the model supports,
@@ -301,6 +319,15 @@ function getCachePoint(providerMetadata) {
   var _a;
   return (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.cachePoint;
 }
+async function shouldEnableCitations(providerMetadata) {
+  var _a, _b;
+  const bedrockOptions = await parseProviderOptions({
+    provider: "bedrock",
+    providerOptions: providerMetadata,
+    schema: bedrockFilePartProviderOptions
+  });
+  return (_b = (_a = bedrockOptions == null ? void 0 : bedrockOptions.citations) == null ? void 0 : _a.enabled) != null ? _b : false;
+}
 async function convertToBedrockChatMessages(prompt) {
   const blocks = groupIntoBlocks(prompt);
   let system = [];
@@ -361,11 +388,17 @@ async function convertToBedrockChatMessages(prompt) {
                           message: "File mime type is required in user message part content"
                         });
                       }
+                      const enableCitations = await shouldEnableCitations(
+                        part.providerOptions
+                      );
                       bedrockContent.push({
                         document: {
                           format: getBedrockDocumentFormat(part.mediaType),
                           name: generateDocumentName(),
-                          source: { bytes: convertToBase64(part.data) }
+                          source: { bytes: convertToBase64(part.data) },
+                          ...enableCitations && {
+                            citations: { enabled: true }
+                          }
                         }
                       });
                     }
@@ -628,7 +661,7 @@ var BedrockChatLanguageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
     this.supportedUrls = {
       // no supported urls for bedrock
@@ -1308,7 +1341,7 @@ var BedrockEmbeddingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
     this.maxEmbeddingsPerCall = 1;
     this.supportsParallelCalls = true;
@@ -1390,7 +1423,7 @@ var BedrockImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
   }
   get maxImagesPerCall() {
@@ -1502,28 +1535,35 @@ import {
   getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 import { AwsV4Signer } from "aws4fetch";
-
-// src/version.ts
-var VERSION = true ? "3.1.0-beta.1" : "0.0.0-test";
-
-// src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
   return async (input, init) => {
-    var _a;
-    const originalHeaders = extractHeaders(init == null ? void 0 : init.headers);
+    var _a, _b;
+    const request = input instanceof Request ? input : void 0;
+    const originalHeaders = combineHeaders4(
+      extractHeaders(request == null ? void 0 : request.headers),
+      extractHeaders(init == null ? void 0 : init.headers)
+    );
     const headersWithUserAgent = withUserAgentSuffix(
       originalHeaders,
       `ai-sdk/amazon-bedrock/${VERSION}`,
       getRuntimeEnvironmentUserAgent()
     );
-    if (((_a = init == null ? void 0 : init.method) == null ? void 0 : _a.toUpperCase()) !== "POST" || !(init == null ? void 0 : init.body)) {
+    let effectiveBody = (_a = init == null ? void 0 : init.body) != null ? _a : void 0;
+    if (effectiveBody === void 0 && request && request.body !== null) {
+      try {
+        effectiveBody = await request.clone().text();
+      } catch (e) {
+      }
+    }
+    const effectiveMethod = (_b = init == null ? void 0 : init.method) != null ? _b : request == null ? void 0 : request.method;
+    if ((effectiveMethod == null ? void 0 : effectiveMethod.toUpperCase()) !== "POST" || !effectiveBody) {
       return fetch(input, {
         ...init,
         headers: headersWithUserAgent
       });
     }
     const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
-    const body = prepareBodyString(init.body);
+    const body = prepareBodyString(effectiveBody);
     const credentials = await getCredentials();
     const signer = new AwsV4Signer({
       url,
@@ -1654,15 +1694,17 @@ Original error: ${errorMessage}`
       })}.amazonaws.com`
     )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
   };
-  const createChatModel = (modelId) => {
+  const getHeaders = () => {
     var _a;
-    return new BedrockChatLanguageModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction,
-      generateId
-    });
+    const baseHeaders = (_a = options.headers) != null ? _a : {};
+    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
   };
+  const createChatModel = (modelId) => new BedrockChatLanguageModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction,
+    generateId
+  });
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -1671,22 +1713,16 @@ Original error: ${errorMessage}`
     }
     return createChatModel(modelId);
   };
-  const createEmbeddingModel = (modelId) => {
-    var _a;
-    return new BedrockEmbeddingModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction
-    });
-  };
-  const createImageModel = (modelId) => {
-    var _a;
-    return new BedrockImageModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction
-    });
-  };
+  const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
+  const createImageModel = (modelId) => new BedrockImageModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;