@ai-sdk/amazon-bedrock 3.0.0-canary.1 → 3.0.0-canary.3

package/dist/index.mjs

@@ -8,8 +8,7 @@ import {
 
 // src/bedrock-chat-language-model.ts
 import {
-  InvalidArgumentError,
-  UnsupportedFunctionalityError as UnsupportedFunctionalityError3
+  InvalidArgumentError
 } from "@ai-sdk/provider";
 import {
   combineHeaders,
@@ -123,9 +122,11 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
 import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
-function prepareTools(mode) {
-  var _a;
-  const tools = ((_a = mode.tools) == null ? void 0 : _a.length) ? mode.tools : void 0;
+function prepareTools({
+  tools,
+  toolChoice
+}) {
+  tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
   if (tools == null) {
     return {
       toolConfig: { tools: void 0, toolChoice: void 0 },
@@ -149,7 +150,6 @@ function prepareTools(mode) {
       });
     }
   }
-  const toolChoice = mode.toolChoice;
   if (toolChoice == null) {
     return {
       toolConfig: { tools: bedrockTools, toolChoice: void 0 },
@@ -184,7 +184,7 @@ function prepareTools(mode) {
     default: {
       const _exhaustiveCheck = type;
       throw new UnsupportedFunctionalityError({
-        functionality: `Unsupported tool choice type: ${_exhaustiveCheck}`
+        functionality: `tool choice type: ${_exhaustiveCheck}`
       });
     }
   }
@@ -194,17 +194,14 @@ function prepareTools(mode) {
 import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
-import {
-  convertUint8ArrayToBase64,
-  createIdGenerator
-} from "@ai-sdk/provider-utils";
+import { createIdGenerator } from "@ai-sdk/provider-utils";
 var generateFileId = createIdGenerator({ prefix: "file", size: 16 });
 function getCachePoint(providerMetadata) {
   var _a;
   return (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.cachePoint;
 }
 function convertToBedrockChatMessages(prompt) {
-  var _a, _b, _c, _d, _e;
+  var _a, _b, _c, _d;
   const blocks = groupIntoBlocks(prompt);
   let system = [];
   const messages = [];
@@ -221,7 +218,7 @@ function convertToBedrockChatMessages(prompt) {
         }
         for (const message of block.messages) {
           system.push({ text: message.content });
-          if (getCachePoint(message.providerMetadata)) {
+          if (getCachePoint(message.providerOptions)) {
             system.push(BEDROCK_CACHE_POINT);
           }
         }
@@ -230,7 +227,7 @@ function convertToBedrockChatMessages(prompt) {
       case "user": {
         const bedrockContent = [];
         for (const message of block.messages) {
-          const { role, content, providerMetadata } = message;
+          const { role, content, providerOptions } = message;
           switch (role) {
             case "user": {
               for (let j = 0; j < content.length; j++) {
@@ -242,43 +239,33 @@ function convertToBedrockChatMessages(prompt) {
                     });
                     break;
                   }
-                  case "image": {
-                    if (part.image instanceof URL) {
-                      throw new UnsupportedFunctionalityError2({
-                        functionality: "Image URLs in user messages"
-                      });
-                    }
-                    bedrockContent.push({
-                      image: {
-                        format: (_b = (_a = part.mimeType) == null ? void 0 : _a.split(
-                          "/"
-                        )) == null ? void 0 : _b[1],
-                        source: {
-                          bytes: convertUint8ArrayToBase64(
-                            (_c = part.image) != null ? _c : part.image
-                          )
-                        }
-                      }
-                    });
-                    break;
-                  }
                   case "file": {
                     if (part.data instanceof URL) {
                       throw new UnsupportedFunctionalityError2({
-                        functionality: "File URLs in user messages"
+                        functionality: "File URL data"
                       });
                     }
-                    bedrockContent.push({
-                      document: {
-                        format: (_e = (_d = part.mimeType) == null ? void 0 : _d.split(
-                          "/"
-                        )) == null ? void 0 : _e[1],
-                        name: generateFileId(),
-                        source: {
-                          bytes: part.data
+                    if (part.mediaType.startsWith("image/")) {
+                      const bedrockImageFormat = part.mediaType === "image/*" ? void 0 : (_b = (_a = part.mediaType) == null ? void 0 : _a.split("/")) == null ? void 0 : _b[1];
+                      bedrockContent.push({
+                        image: {
+                          format: bedrockImageFormat,
+                          source: { bytes: part.data }
                         }
-                      }
-                    });
+                      });
+                    } else {
+                      bedrockContent.push({
+                        document: {
+                          format: (_d = (_c = part.mediaType) == null ? void 0 : _c.split(
+                            "/"
+                          )) == null ? void 0 : _d[1],
+                          name: generateFileId(),
+                          source: {
+                            bytes: part.data
+                          }
+                        }
+                      });
+                    }
                     break;
                   }
                 }
@@ -295,12 +282,12 @@ function convertToBedrockChatMessages(prompt) {
                         text: part2.text
                       };
                     case "image":
-                      if (!part2.mimeType) {
+                      if (!part2.mediaType) {
                         throw new Error(
                           "Image mime type is required in tool result part content"
                         );
                       }
-                      const format = part2.mimeType.split("/")[1];
+                      const format = part2.mediaType.split("/")[1];
                       if (!isBedrockImageFormat(format)) {
                         throw new Error(
                           `Unsupported image format: ${format}`
@@ -330,7 +317,7 @@ function convertToBedrockChatMessages(prompt) {
               throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
             }
           }
-          if (getCachePoint(providerMetadata)) {
+          if (getCachePoint(providerOptions)) {
             bedrockContent.push(BEDROCK_CACHE_POINT);
           }
         }
@@ -404,7 +391,7 @@ function convertToBedrockChatMessages(prompt) {
               }
             }
           }
-          if (getCachePoint(message.providerMetadata)) {
+          if (getCachePoint(message.providerOptions)) {
             bedrockContent.push(BEDROCK_CACHE_POINT);
           }
         }
@@ -502,7 +489,6 @@ var BedrockChatLanguageModel = class {
     this.supportsImageUrls = false;
   }
   getArgs({
-    mode,
     prompt,
     maxTokens,
     temperature,
@@ -513,10 +499,11 @@ var BedrockChatLanguageModel = class {
     stopSequences,
     responseFormat,
     seed,
-    providerMetadata
+    tools,
+    toolChoice,
+    providerOptions
   }) {
     var _a, _b, _c, _d, _e, _f, _g;
-    const type = mode.type;
     const warnings = [];
     if (frequencyPenalty != null) {
       warnings.push({
@@ -551,7 +538,7 @@ var BedrockChatLanguageModel = class {
     }
     const { system, messages } = convertToBedrockChatMessages(prompt);
     const reasoningConfigOptions = BedrockReasoningConfigOptionsSchema.safeParse(
-      (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.reasoning_config
+      (_a = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _a.reasoning_config
     );
     if (!reasoningConfigOptions.success) {
       throw new InvalidArgumentError({
@@ -598,58 +585,20 @@ var BedrockChatLanguageModel = class {
         details: "topP is not supported when thinking is enabled"
       });
     }
-    const baseArgs = {
-      system,
-      additionalModelRequestFields: this.settings.additionalModelRequestFields,
-      ...Object.keys(inferenceConfig).length > 0 && {
-        inferenceConfig
+    const { toolConfig, toolWarnings } = prepareTools({ tools, toolChoice });
+    return {
+      command: {
+        system,
+        messages,
+        additionalModelRequestFields: this.settings.additionalModelRequestFields,
+        ...Object.keys(inferenceConfig).length > 0 && {
+          inferenceConfig
+        },
+        ...providerOptions == null ? void 0 : providerOptions.bedrock,
+        ...((_g = toolConfig.tools) == null ? void 0 : _g.length) ? { toolConfig } : {}
       },
-      messages,
-      ...providerMetadata == null ? void 0 : providerMetadata.bedrock
+      warnings: [...warnings, ...toolWarnings]
     };
-    switch (type) {
-      case "regular": {
-        const { toolConfig, toolWarnings } = prepareTools(mode);
-        return {
-          command: {
-            ...baseArgs,
-            ...((_g = toolConfig.tools) == null ? void 0 : _g.length) ? { toolConfig } : {}
-          },
-          warnings: [...warnings, ...toolWarnings]
-        };
-      }
-      case "object-json": {
-        throw new UnsupportedFunctionalityError3({
-          functionality: "json-mode object generation"
-        });
-      }
-      case "object-tool": {
-        return {
-          command: {
-            ...baseArgs,
-            toolConfig: {
-              tools: [
-                {
-                  toolSpec: {
-                    name: mode.tool.name,
-                    description: mode.tool.description,
-                    inputSchema: {
-                      json: mode.tool.parameters
-                    }
-                  }
-                }
-              ],
-              toolChoice: { tool: { name: mode.tool.name } }
-            }
-          },
-          warnings
-        };
-      }
-      default: {
-        const _exhaustiveCheck = type;
-        throw new Error(`Unsupported type: ${_exhaustiveCheck}`);
-      }
-    }
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;