@ai-sdk/amazon-bedrock 3.0.28 → 3.0.30

package/dist/index.mjs

@@ -8,7 +8,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.28" : "0.0.0-test";
+var VERSION = true ? "3.0.30" : "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 }
+                          }
                         }
                       });
                     }