@ai-sdk/amazon-bedrock 3.0.27 → 3.0.29

package/dist/index.mjs

@@ -8,7 +8,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.27" : "0.0.0-test";
+var VERSION = true ? "3.0.29" : "0.0.0-test";
 
 // src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
@@ -60,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,
@@ -307,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 = [];
@@ -367,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 }
+                          }
                         }
                       });
                     }
@@ -1510,21 +1537,33 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 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,