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

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @ai-sdk/amazon-bedrock
 
+## 4.0.0-beta.102
+
+### Patch Changes
+
+- 9ab6ebe: feat(provider/amazon-bedrock): expose stop_sequence in provider metadata
+
+  The Bedrock provider now exposes the specific stop sequence that triggered generation to halt via `providerMetadata.bedrock.stopSequence`. This is implemented by:
+
+  - Requesting `/stop_sequence` via `additionalModelResponseFieldPaths` in the API call
+  - Parsing the value from `additionalModelResponseFields.stop_sequence` in both generate and stream responses
+  - Exposing it as `stopSequence` in the provider metadata (returns `null` when no stop sequence was matched)
+
+- 9ab6ebe: Add stop sequence support for amazon bedrock provider
+
 ## 4.0.0-beta.101
 
 ### Patch Changes

package/dist/index.js

@@ -919,6 +919,7 @@ var BedrockChatLanguageModel = class {
         system,
         messages,
         additionalModelRequestFields: bedrockOptions.additionalModelRequestFields,
+        additionalModelResponseFieldPaths: ["/stop_sequence"],
         ...Object.keys(inferenceConfig).length > 0 && {
           inferenceConfig
         },
@@ -936,7 +937,7 @@ var BedrockChatLanguageModel = class {
     return (0, import_provider_utils4.combineHeaders)(await (0, import_provider_utils4.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,
@@ -1010,15 +1011,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 {
@@ -1059,6 +1062,7 @@ var BedrockChatLanguageModel = class {
     let usage = void 0;
     let providerMetadata = void 0;
     let isJsonResponseFromTool = false;
+    let stopSequence = null;
     const contentBlocks = {};
     return {
       stream: response.pipeThrough(
@@ -1067,7 +1071,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 });
@@ -1101,12 +1105,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
                 }
@@ -1123,7 +1128,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({
@@ -1131,7 +1136,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" };
@@ -1146,7 +1151,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) {
@@ -1192,7 +1197,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) {
@@ -1233,7 +1238,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";
@@ -1257,7 +1262,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",
@@ -1270,16 +1275,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
                   }
                 };
               }
@@ -1306,6 +1317,9 @@ var BedrockStopReasonSchema = import_v43.z.union([
   import_v43.z.enum(BEDROCK_STOP_REASONS),
   import_v43.z.string()
 ]);
+var BedrockAdditionalModelResponseFieldsSchema = import_v43.z.object({
+  stop_sequence: import_v43.z.string().optional()
+}).catchall(import_v43.z.unknown());
 var BedrockToolUseSchema = import_v43.z.object({
   toolUseId: import_v43.z.string(),
   name: import_v43.z.string(),
@@ -1342,6 +1356,7 @@ var BedrockResponseSchema = import_v43.z.object({
     })
   }),
   stopReason: BedrockStopReasonSchema,
+  additionalModelResponseFields: BedrockAdditionalModelResponseFieldsSchema.nullish(),
   trace: import_v43.z.unknown().nullish(),
   usage: import_v43.z.object({
     inputTokens: import_v43.z.number(),
@@ -1381,7 +1396,7 @@ var BedrockStreamSchema = import_v43.z.object({
   }).nullish(),
   internalServerException: import_v43.z.record(import_v43.z.string(), import_v43.z.unknown()).nullish(),
   messageStop: import_v43.z.object({
-    additionalModelResponseFields: import_v43.z.record(import_v43.z.string(), import_v43.z.unknown()).nullish(),
+    additionalModelResponseFields: BedrockAdditionalModelResponseFieldsSchema.nullish(),
     stopReason: BedrockStopReasonSchema
   }).nullish(),
   metadata: import_v43.z.object({
@@ -1523,31 +1538,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",
@@ -1555,7 +1649,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 (0, import_provider_utils6.postJsonToApi)({
       url: this.getUrl(this.modelId),
       headers: await (0, import_provider_utils6.resolve)(
@@ -1572,6 +1666,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,
@@ -1583,8 +1689,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 (0, import_provider_utils6.convertUint8ArrayToBase64)(file.data);
+  }
+  return file.data;
+}
 var bedrockImageResponseSchema = import_v46.z.object({
-  images: import_v46.z.array(import_v46.z.string())
+  // Normal successful response
+  images: import_v46.z.array(import_v46.z.string()).optional(),
+  // Moderation response fields
+  id: import_v46.z.string().optional(),
+  status: import_v46.z.string().optional(),
+  result: import_v46.z.unknown().optional(),
+  progress: import_v46.z.unknown().optional(),
+  details: import_v46.z.record(import_v46.z.string(), import_v46.z.unknown()).optional(),
+  preview: import_v46.z.unknown().optional()
 });
 
 // src/bedrock-sigv4-fetch.ts
@@ -1592,7 +1717,7 @@ var import_provider_utils7 = require("@ai-sdk/provider-utils");
 var import_aws4fetch = require("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) {