@ai-sdk/amazon-bedrock 4.0.0-beta.99 → 4.0.0

package/dist/index.js

@@ -107,45 +107,6 @@ var BedrockErrorSchema = import_v42.z.object({
   type: import_v42.z.string().nullish()
 });
 
-// src/convert-bedrock-usage.ts
-function convertBedrockUsage(usage) {
-  var _a, _b;
-  if (usage == null) {
-    return {
-      inputTokens: {
-        total: void 0,
-        noCache: void 0,
-        cacheRead: void 0,
-        cacheWrite: void 0
-      },
-      outputTokens: {
-        total: void 0,
-        text: void 0,
-        reasoning: void 0
-      },
-      raw: void 0
-    };
-  }
-  const inputTokens = usage.inputTokens;
-  const outputTokens = usage.outputTokens;
-  const cacheReadTokens = (_a = usage.cacheReadInputTokens) != null ? _a : 0;
-  const cacheWriteTokens = (_b = usage.cacheWriteInputTokens) != null ? _b : 0;
-  return {
-    inputTokens: {
-      total: inputTokens,
-      noCache: inputTokens - cacheReadTokens,
-      cacheRead: cacheReadTokens,
-      cacheWrite: cacheWriteTokens
-    },
-    outputTokens: {
-      total: outputTokens,
-      text: outputTokens,
-      reasoning: void 0
-    },
-    raw: usage
-  };
-}
-
 // src/bedrock-event-stream-response-handler.ts
 var import_provider = require("@ai-sdk/provider");
 var import_provider_utils = require("@ai-sdk/provider-utils");
@@ -355,6 +316,45 @@ async function prepareTools({
   };
 }
 
+// src/convert-bedrock-usage.ts
+function convertBedrockUsage(usage) {
+  var _a, _b;
+  if (usage == null) {
+    return {
+      inputTokens: {
+        total: void 0,
+        noCache: void 0,
+        cacheRead: void 0,
+        cacheWrite: void 0
+      },
+      outputTokens: {
+        total: void 0,
+        text: void 0,
+        reasoning: void 0
+      },
+      raw: void 0
+    };
+  }
+  const inputTokens = usage.inputTokens;
+  const outputTokens = usage.outputTokens;
+  const cacheReadTokens = (_a = usage.cacheReadInputTokens) != null ? _a : 0;
+  const cacheWriteTokens = (_b = usage.cacheWriteInputTokens) != null ? _b : 0;
+  return {
+    inputTokens: {
+      total: inputTokens,
+      noCache: inputTokens - cacheReadTokens,
+      cacheRead: cacheReadTokens,
+      cacheWrite: cacheWriteTokens
+    },
+    outputTokens: {
+      total: outputTokens,
+      text: outputTokens,
+      reasoning: void 0
+    },
+    raw: usage
+  };
+}
+
 // src/convert-to-bedrock-chat-messages.ts
 var import_provider3 = require("@ai-sdk/provider");
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
@@ -454,6 +454,9 @@ async function convertToBedrockChatMessages(prompt) {
             }
             case "tool": {
               for (const part of content) {
+                if (part.type === "tool-approval-response") {
+                  continue;
+                }
                 let toolResultContent;
                 const output = part.output;
                 switch (output.type) {
@@ -706,7 +709,7 @@ function mapBedrockFinishReason(finishReason, isJsonResponseFromTool) {
     case "tool_use":
       return isJsonResponseFromTool ? "stop" : "tool-calls";
     default:
-      return "unknown";
+      return "other";
   }
 }
 
@@ -919,6 +922,7 @@ var BedrockChatLanguageModel = class {
         system,
         messages,
         additionalModelRequestFields: bedrockOptions.additionalModelRequestFields,
+        additionalModelResponseFieldPaths: ["/stop_sequence"],
         ...Object.keys(inferenceConfig).length > 0 && {
           inferenceConfig
         },
@@ -936,7 +940,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, _k;
     const {
       command: args,
       warnings,
@@ -1010,23 +1014,28 @@ 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 {
       content,
-      finishReason: mapBedrockFinishReason(
-        response.stopReason,
-        isJsonResponseFromTool
-      ),
+      finishReason: {
+        unified: mapBedrockFinishReason(
+          response.stopReason,
+          isJsonResponseFromTool
+        ),
+        raw: (_k = response.stopReason) != null ? _k : void 0
+      },
       usage: convertBedrockUsage(response.usage),
       response: {
         // TODO add id, timestamp, etc
@@ -1055,10 +1064,14 @@ var BedrockChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    let finishReason = "unknown";
+    let finishReason = {
+      unified: "other",
+      raw: void 0
+    };
     let usage = void 0;
     let providerMetadata = void 0;
     let isJsonResponseFromTool = false;
+    let stopSequence = null;
     const contentBlocks = {};
     return {
       stream: response.pipeThrough(
@@ -1067,9 +1080,9 @@ 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, _l;
             function enqueueError(bedrockError) {
-              finishReason = "error";
+              finishReason = { unified: "error", raw: void 0 };
               controller.enqueue({ type: "error", error: bedrockError });
             }
             if (options.includeRawChunks) {
@@ -1097,16 +1110,20 @@ var BedrockChatLanguageModel = class {
               return;
             }
             if (value.messageStop) {
-              finishReason = mapBedrockFinishReason(
-                value.messageStop.stopReason,
-                isJsonResponseFromTool
-              );
+              finishReason = {
+                unified: mapBedrockFinishReason(
+                  value.messageStop.stopReason,
+                  isJsonResponseFromTool
+                ),
+                raw: (_a = value.messageStop.stopReason) != null ? _a : void 0
+              };
+              stopSequence = (_c = (_b = value.messageStop.additionalModelResponseFields) == null ? void 0 : _b.stop_sequence) != null ? _c : 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 = ((_d = value.metadata.usage) == null ? void 0 : _d.cacheWriteInputTokens) != null ? {
                 usage: {
                   cacheWriteInputTokens: value.metadata.usage.cacheWriteInputTokens
                 }
@@ -1123,7 +1140,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 (((_e = value.contentBlockStart) == null ? void 0 : _e.contentBlockIndex) != null && !((_g = (_f = value.contentBlockStart) == null ? void 0 : _f.start) == null ? void 0 : _g.toolUse)) {
               const blockIndex = value.contentBlockStart.contentBlockIndex;
               contentBlocks[blockIndex] = { type: "text" };
               controller.enqueue({
@@ -1131,7 +1148,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 (((_h = value.contentBlockDelta) == null ? void 0 : _h.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 +1163,7 @@ var BedrockChatLanguageModel = class {
                 delta: value.contentBlockDelta.delta.text
               });
             }
-            if (((_f = value.contentBlockStop) == null ? void 0 : _f.contentBlockIndex) != null) {
+            if (((_i = value.contentBlockStop) == null ? void 0 : _i.contentBlockIndex) != null) {
               const blockIndex = value.contentBlockStop.contentBlockIndex;
               const contentBlock = contentBlocks[blockIndex];
               if (contentBlock != null) {
@@ -1192,7 +1209,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 (((_j = value.contentBlockDelta) == null ? void 0 : _j.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 +1250,7 @@ var BedrockChatLanguageModel = class {
               }
             }
             const contentBlockStart = value.contentBlockStart;
-            if (((_h = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _h.toolUse) != null) {
+            if (((_k = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _k.toolUse) != null) {
               const toolUse = contentBlockStart.start.toolUse;
               const blockIndex = contentBlockStart.contentBlockIndex;
               const isJsonResponseTool = usesJsonResponseTool && toolUse.name === "json";
@@ -1257,7 +1274,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 = (_l = contentBlockDelta.delta.toolUse.input) != null ? _l : "";
                 if (!contentBlock.isJsonResponseTool) {
                   controller.enqueue({
                     type: "tool-input-delta",
@@ -1270,16 +1287,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 +1329,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 +1368,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 +1408,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 +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",
@@ -1555,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 (0, import_provider_utils6.postJsonToApi)({
       url: this.getUrl(this.modelId),
       headers: await (0, import_provider_utils6.resolve)(
@@ -1572,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,
@@ -1583,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 (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 +1729,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.99" : "0.0.0-test";
+var VERSION = true ? "4.0.0" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
@@ -1926,6 +2063,8 @@ Original error: ${errorMessage}`
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.embeddingModel = createEmbeddingModel;
+  provider.textEmbedding = createEmbeddingModel;
+  provider.textEmbeddingModel = createEmbeddingModel;
   provider.image = createImageModel;
   provider.imageModel = createImageModel;
   provider.reranking = createRerankingModel;