@ai-sdk/amazon-bedrock 4.0.0-beta.101 → 4.0.0-beta.102

package/dist/index.mjs

@@ -915,6 +915,7 @@ var BedrockChatLanguageModel = class {
         system,
         messages,
         additionalModelRequestFields: bedrockOptions.additionalModelRequestFields,
+        additionalModelResponseFieldPaths: ["/stop_sequence"],
         ...Object.keys(inferenceConfig).length > 0 && {
           inferenceConfig
         },
@@ -932,7 +933,7 @@ var BedrockChatLanguageModel = class {
     return combineHeaders(await resolve(this.config.headers), headers);
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const {
       command: args,
       warnings,
@@ -1006,15 +1007,17 @@ var BedrockChatLanguageModel = class {
         }
       }
     }
-    const providerMetadata = response.trace || response.usage || isJsonResponseFromTool ? {
+    const stopSequence = (_i = (_h = response.additionalModelResponseFields) == null ? void 0 : _h.stop_sequence) != null ? _i : null;
+    const providerMetadata = response.trace || response.usage || isJsonResponseFromTool || stopSequence ? {
       bedrock: {
         ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
-        ...((_h = response.usage) == null ? void 0 : _h.cacheWriteInputTokens) != null && {
+        ...((_j = response.usage) == null ? void 0 : _j.cacheWriteInputTokens) != null && {
           usage: {
             cacheWriteInputTokens: response.usage.cacheWriteInputTokens
           }
         },
-        ...isJsonResponseFromTool && { isJsonResponseFromTool: true }
+        ...isJsonResponseFromTool && { isJsonResponseFromTool: true },
+        stopSequence
       }
     } : void 0;
     return {
@@ -1055,6 +1058,7 @@ var BedrockChatLanguageModel = class {
     let usage = void 0;
     let providerMetadata = void 0;
     let isJsonResponseFromTool = false;
+    let stopSequence = null;
     const contentBlocks = {};
     return {
       stream: response.pipeThrough(
@@ -1063,7 +1067,7 @@ var BedrockChatLanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
             function enqueueError(bedrockError) {
               finishReason = "error";
               controller.enqueue({ type: "error", error: bedrockError });
@@ -1097,12 +1101,13 @@ var BedrockChatLanguageModel = class {
                 value.messageStop.stopReason,
                 isJsonResponseFromTool
               );
+              stopSequence = (_b = (_a = value.messageStop.additionalModelResponseFields) == null ? void 0 : _a.stop_sequence) != null ? _b : null;
             }
             if (value.metadata) {
               if (value.metadata.usage) {
                 usage = value.metadata.usage;
               }
-              const cacheUsage = ((_a = value.metadata.usage) == null ? void 0 : _a.cacheWriteInputTokens) != null ? {
+              const cacheUsage = ((_c = value.metadata.usage) == null ? void 0 : _c.cacheWriteInputTokens) != null ? {
                 usage: {
                   cacheWriteInputTokens: value.metadata.usage.cacheWriteInputTokens
                 }
@@ -1119,7 +1124,7 @@ var BedrockChatLanguageModel = class {
                 };
               }
             }
-            if (((_b = value.contentBlockStart) == null ? void 0 : _b.contentBlockIndex) != null && !((_d = (_c = value.contentBlockStart) == null ? void 0 : _c.start) == null ? void 0 : _d.toolUse)) {
+            if (((_d = value.contentBlockStart) == null ? void 0 : _d.contentBlockIndex) != null && !((_f = (_e = value.contentBlockStart) == null ? void 0 : _e.start) == null ? void 0 : _f.toolUse)) {
               const blockIndex = value.contentBlockStart.contentBlockIndex;
               contentBlocks[blockIndex] = { type: "text" };
               controller.enqueue({
@@ -1127,7 +1132,7 @@ var BedrockChatLanguageModel = class {
                 id: String(blockIndex)
               });
             }
-            if (((_e = value.contentBlockDelta) == null ? void 0 : _e.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
+            if (((_g = value.contentBlockDelta) == null ? void 0 : _g.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
               const blockIndex = value.contentBlockDelta.contentBlockIndex || 0;
               if (contentBlocks[blockIndex] == null) {
                 contentBlocks[blockIndex] = { type: "text" };
@@ -1142,7 +1147,7 @@ var BedrockChatLanguageModel = class {
                 delta: value.contentBlockDelta.delta.text
               });
             }
-            if (((_f = value.contentBlockStop) == null ? void 0 : _f.contentBlockIndex) != null) {
+            if (((_h = value.contentBlockStop) == null ? void 0 : _h.contentBlockIndex) != null) {
               const blockIndex = value.contentBlockStop.contentBlockIndex;
               const contentBlock = contentBlocks[blockIndex];
               if (contentBlock != null) {
@@ -1188,7 +1193,7 @@ var BedrockChatLanguageModel = class {
                 delete contentBlocks[blockIndex];
               }
             }
-            if (((_g = value.contentBlockDelta) == null ? void 0 : _g.delta) && "reasoningContent" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.reasoningContent) {
+            if (((_i = value.contentBlockDelta) == null ? void 0 : _i.delta) && "reasoningContent" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.reasoningContent) {
               const blockIndex = value.contentBlockDelta.contentBlockIndex || 0;
               const reasoningContent = value.contentBlockDelta.delta.reasoningContent;
               if ("text" in reasoningContent && reasoningContent.text) {
@@ -1229,7 +1234,7 @@ var BedrockChatLanguageModel = class {
               }
             }
             const contentBlockStart = value.contentBlockStart;
-            if (((_h = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _h.toolUse) != null) {
+            if (((_j = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _j.toolUse) != null) {
               const toolUse = contentBlockStart.start.toolUse;
               const blockIndex = contentBlockStart.contentBlockIndex;
               const isJsonResponseTool = usesJsonResponseTool && toolUse.name === "json";
@@ -1253,7 +1258,7 @@ var BedrockChatLanguageModel = class {
               const blockIndex = contentBlockDelta.contentBlockIndex;
               const contentBlock = contentBlocks[blockIndex];
               if ((contentBlock == null ? void 0 : contentBlock.type) === "tool-call") {
-                const delta = (_i = contentBlockDelta.delta.toolUse.input) != null ? _i : "";
+                const delta = (_k = contentBlockDelta.delta.toolUse.input) != null ? _k : "";
                 if (!contentBlock.isJsonResponseTool) {
                   controller.enqueue({
                     type: "tool-input-delta",
@@ -1266,16 +1271,22 @@ var BedrockChatLanguageModel = class {
             }
           },
           flush(controller) {
-            if (isJsonResponseFromTool) {
+            if (isJsonResponseFromTool || stopSequence != null) {
               if (providerMetadata) {
                 providerMetadata.bedrock = {
                   ...providerMetadata.bedrock,
-                  isJsonResponseFromTool: true
+                  ...isJsonResponseFromTool && {
+                    isJsonResponseFromTool: true
+                  },
+                  stopSequence
                 };
               } else {
                 providerMetadata = {
                   bedrock: {
-                    isJsonResponseFromTool: true
+                    ...isJsonResponseFromTool && {
+                      isJsonResponseFromTool: true
+                    },
+                    stopSequence
                   }
                 };
               }
@@ -1302,6 +1313,9 @@ var BedrockStopReasonSchema = z3.union([
   z3.enum(BEDROCK_STOP_REASONS),
   z3.string()
 ]);
+var BedrockAdditionalModelResponseFieldsSchema = z3.object({
+  stop_sequence: z3.string().optional()
+}).catchall(z3.unknown());
 var BedrockToolUseSchema = z3.object({
   toolUseId: z3.string(),
   name: z3.string(),
@@ -1338,6 +1352,7 @@ var BedrockResponseSchema = z3.object({
     })
   }),
   stopReason: BedrockStopReasonSchema,
+  additionalModelResponseFields: BedrockAdditionalModelResponseFieldsSchema.nullish(),
   trace: z3.unknown().nullish(),
   usage: z3.object({
     inputTokens: z3.number(),
@@ -1377,7 +1392,7 @@ var BedrockStreamSchema = z3.object({
   }).nullish(),
   internalServerException: z3.record(z3.string(), z3.unknown()).nullish(),
   messageStop: z3.object({
-    additionalModelResponseFields: z3.record(z3.string(), z3.unknown()).nullish(),
+    additionalModelResponseFields: BedrockAdditionalModelResponseFieldsSchema.nullish(),
     stopReason: BedrockStopReasonSchema
   }).nullish(),
   metadata: z3.object({
@@ -1498,6 +1513,7 @@ var BedrockEmbeddingResponseSchema = z5.object({
 // src/bedrock-image-model.ts
 import {
   combineHeaders as combineHeaders3,
+  convertUint8ArrayToBase64,
   createJsonErrorResponseHandler as createJsonErrorResponseHandler3,
   createJsonResponseHandler as createJsonResponseHandler3,
   postJsonToApi as postJsonToApi3,
@@ -1534,31 +1550,110 @@ var BedrockImageModel = class {
     seed,
     providerOptions,
     headers,
-    abortSignal
+    abortSignal,
+    files,
+    mask
   }) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
     const warnings = [];
     const [width, height] = size ? size.split("x").map(Number) : [];
-    const args = {
-      taskType: "TEXT_IMAGE",
-      textToImageParams: {
-        text: prompt,
-        ...((_a = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _a.negativeText) ? {
-          negativeText: providerOptions.bedrock.negativeText
-        } : {},
-        ...((_b = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _b.style) ? {
-          style: providerOptions.bedrock.style
-        } : {}
-      },
-      imageGenerationConfig: {
-        ...width ? { width } : {},
-        ...height ? { height } : {},
-        ...seed ? { seed } : {},
-        ...n ? { numberOfImages: n } : {},
-        ...((_c = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _c.quality) ? { quality: providerOptions.bedrock.quality } : {},
-        ...((_d = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _d.cfgScale) ? { cfgScale: providerOptions.bedrock.cfgScale } : {}
-      }
+    const hasFiles = files != null && files.length > 0;
+    const imageGenerationConfig = {
+      ...width ? { width } : {},
+      ...height ? { height } : {},
+      ...seed ? { seed } : {},
+      ...n ? { numberOfImages: n } : {},
+      ...((_a = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _a.quality) ? { quality: providerOptions.bedrock.quality } : {},
+      ...((_b = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _b.cfgScale) ? { cfgScale: providerOptions.bedrock.cfgScale } : {}
     };
+    let args;
+    if (hasFiles) {
+      const hasMask = (mask == null ? void 0 : mask.type) != null;
+      const hasMaskPrompt = ((_c = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _c.maskPrompt) != null;
+      const taskType = (_e = (_d = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _d.taskType) != null ? _e : hasMask || hasMaskPrompt ? "INPAINTING" : "IMAGE_VARIATION";
+      const sourceImageBase64 = getBase64Data(files[0]);
+      switch (taskType) {
+        case "INPAINTING": {
+          const inPaintingParams = {
+            image: sourceImageBase64,
+            ...prompt ? { text: prompt } : {},
+            ...((_f = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _f.negativeText) ? { negativeText: providerOptions.bedrock.negativeText } : {}
+          };
+          if (hasMask) {
+            inPaintingParams.maskImage = getBase64Data(mask);
+          } else if (hasMaskPrompt) {
+            inPaintingParams.maskPrompt = providerOptions.bedrock.maskPrompt;
+          }
+          args = {
+            taskType: "INPAINTING",
+            inPaintingParams,
+            imageGenerationConfig
+          };
+          break;
+        }
+        case "OUTPAINTING": {
+          const outPaintingParams = {
+            image: sourceImageBase64,
+            ...prompt ? { text: prompt } : {},
+            ...((_g = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _g.negativeText) ? { negativeText: providerOptions.bedrock.negativeText } : {},
+            ...((_h = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _h.outPaintingMode) ? { outPaintingMode: providerOptions.bedrock.outPaintingMode } : {}
+          };
+          if (hasMask) {
+            outPaintingParams.maskImage = getBase64Data(mask);
+          } else if (hasMaskPrompt) {
+            outPaintingParams.maskPrompt = providerOptions.bedrock.maskPrompt;
+          }
+          args = {
+            taskType: "OUTPAINTING",
+            outPaintingParams,
+            imageGenerationConfig
+          };
+          break;
+        }
+        case "BACKGROUND_REMOVAL": {
+          args = {
+            taskType: "BACKGROUND_REMOVAL",
+            backgroundRemovalParams: {
+              image: sourceImageBase64
+            }
+          };
+          break;
+        }
+        case "IMAGE_VARIATION": {
+          const images = files.map((file) => getBase64Data(file));
+          const imageVariationParams = {
+            images,
+            ...prompt ? { text: prompt } : {},
+            ...((_i = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _i.negativeText) ? { negativeText: providerOptions.bedrock.negativeText } : {},
+            ...((_j = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _j.similarityStrength) != null ? {
+              similarityStrength: providerOptions.bedrock.similarityStrength
+            } : {}
+          };
+          args = {
+            taskType: "IMAGE_VARIATION",
+            imageVariationParams,
+            imageGenerationConfig
+          };
+          break;
+        }
+        default:
+          throw new Error(`Unsupported task type: ${taskType}`);
+      }
+    } else {
+      args = {
+        taskType: "TEXT_IMAGE",
+        textToImageParams: {
+          text: prompt,
+          ...((_k = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _k.negativeText) ? {
+            negativeText: providerOptions.bedrock.negativeText
+          } : {},
+          ...((_l = providerOptions == null ? void 0 : providerOptions.bedrock) == null ? void 0 : _l.style) ? {
+            style: providerOptions.bedrock.style
+          } : {}
+        },
+        imageGenerationConfig
+      };
+    }
     if (aspectRatio != void 0) {
       warnings.push({
         type: "unsupported",
@@ -1566,7 +1661,7 @@ var BedrockImageModel = class {
         details: "This model does not support aspect ratio. Use `size` instead."
       });
     }
-    const currentDate = (_g = (_f = (_e = this.config._internal) == null ? void 0 : _e.currentDate) == null ? void 0 : _f.call(_e)) != null ? _g : /* @__PURE__ */ new Date();
+    const currentDate = (_o = (_n = (_m = this.config._internal) == null ? void 0 : _m.currentDate) == null ? void 0 : _n.call(_m)) != null ? _o : /* @__PURE__ */ new Date();
     const { value: response, responseHeaders } = await postJsonToApi3({
       url: this.getUrl(this.modelId),
       headers: await resolve3(
@@ -1583,6 +1678,18 @@ var BedrockImageModel = class {
       abortSignal,
       fetch: this.config.fetch
     });
+    if (response.status === "Request Moderated") {
+      const moderationReasons = (_p = response.details) == null ? void 0 : _p["Moderation Reasons"];
+      const reasons = Array.isArray(moderationReasons) ? moderationReasons : ["Unknown"];
+      throw new Error(
+        `Amazon Bedrock request was moderated: ${reasons.join(", ")}`
+      );
+    }
+    if (!response.images || response.images.length === 0) {
+      throw new Error(
+        "Amazon Bedrock returned no images. " + (response.status ? `Status: ${response.status}` : "")
+      );
+    }
     return {
       images: response.images,
       warnings,
@@ -1594,8 +1701,27 @@ var BedrockImageModel = class {
     };
   }
 };
+function getBase64Data(file) {
+  if (file.type === "url") {
+    throw new Error(
+      "URL-based images are not supported for Amazon Bedrock image editing. Please provide the image data directly."
+    );
+  }
+  if (file.data instanceof Uint8Array) {
+    return convertUint8ArrayToBase64(file.data);
+  }
+  return file.data;
+}
 var bedrockImageResponseSchema = z6.object({
-  images: z6.array(z6.string())
+  // Normal successful response
+  images: z6.array(z6.string()).optional(),
+  // Moderation response fields
+  id: z6.string().optional(),
+  status: z6.string().optional(),
+  result: z6.unknown().optional(),
+  progress: z6.unknown().optional(),
+  details: z6.record(z6.string(), z6.unknown()).optional(),
+  preview: z6.unknown().optional()
 });
 
 // src/bedrock-sigv4-fetch.ts
@@ -1608,7 +1734,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.101" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.102" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {