@ai-sdk/amazon-bedrock 4.0.108 → 4.0.110

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/amazon-bedrock",
-  "version": "4.0.108",
+  "version": "4.0.110",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -38,7 +38,7 @@
     "@smithy/eventstream-codec": "^4.0.1",
     "@smithy/util-utf8": "^4.0.0",
     "aws4fetch": "^1.0.20",
-    "@ai-sdk/anthropic": "3.0.79",
+    "@ai-sdk/anthropic": "3.0.80",
     "@ai-sdk/provider": "3.0.10",
     "@ai-sdk/provider-utils": "4.0.27"
   },

package/src/bedrock-api-types.ts

@@ -174,7 +174,7 @@ export interface BedrockImageBlock {
 export interface BedrockToolResultBlock {
   toolResult: {
     toolUseId: string;
-    content: Array<BedrockTextBlock | BedrockImageBlock>;
+    content: Array<BedrockTextBlock | BedrockImageBlock | BedrockDocumentBlock>;
   };
 }
 

package/src/convert-to-bedrock-chat-messages.ts

@@ -182,34 +182,69 @@ export async function convertToBedrockChatMessages(
                 const output = part.output;
                 switch (output.type) {
                   case 'content': {
-                    toolResultContent = output.value.map(contentPart => {
-                      switch (contentPart.type) {
-                        case 'text':
-                          return { text: contentPart.text };
-                        case 'image-data':
-                          if (!contentPart.mediaType.startsWith('image/')) {
+                    toolResultContent = await Promise.all(
+                      output.value.map(async contentPart => {
+                        switch (contentPart.type) {
+                          case 'text':
+                            return { text: contentPart.text };
+                          case 'image-data': {
+                            return {
+                              image: {
+                                format: getBedrockImageFormat(
+                                  contentPart.mediaType,
+                                ),
+                                source: {
+                                  bytes: convertToBase64(contentPart.data),
+                                },
+                              },
+                            };
+                          }
+                          case 'file-data': {
+                            if (!contentPart.mediaType.startsWith('image/')) {
+                              const enableCitations =
+                                await shouldEnableCitations(
+                                  contentPart.providerOptions,
+                                );
+
+                              return {
+                                document: {
+                                  format: getBedrockDocumentFormat(
+                                    contentPart.mediaType,
+                                  ),
+                                  name:
+                                    'filename' in contentPart &&
+                                    contentPart.filename
+                                      ? stripFileExtension(contentPart.filename)
+                                      : generateDocumentName(),
+                                  source: {
+                                    bytes: convertToBase64(contentPart.data),
+                                  },
+                                  ...(enableCitations && {
+                                    citations: { enabled: true },
+                                  }),
+                                },
+                              };
+                            }
+
+                            return {
+                              image: {
+                                format: getBedrockImageFormat(
+                                  contentPart.mediaType,
+                                ),
+                                source: {
+                                  bytes: convertToBase64(contentPart.data),
+                                },
+                              },
+                            };
+                          }
+                          default: {
                             throw new UnsupportedFunctionalityError({
-                              functionality: `media type: ${contentPart.mediaType}`,
+                              functionality: `unsupported tool content part type: ${contentPart.type}`,
                             });
                           }
-
-                          const format = getBedrockImageFormat(
-                            contentPart.mediaType,
-                          );
-
-                          return {
-                            image: {
-                              format,
-                              source: { bytes: contentPart.data },
-                            },
-                          };
-                        default: {
-                          throw new UnsupportedFunctionalityError({
-                            functionality: `unsupported tool content part type: ${contentPart.type}`,
-                          });
                         }
-                      }
-                    });
+                      }),
+                    );
                     break;
                   }
                   case 'text':