@ai-sdk/amazon-bedrock 2.2.8 → 3.0.0-alpha.1

package/dist/index.mjs

@@ -7,18 +7,15 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/bedrock-chat-language-model.ts
-import {
-  InvalidArgumentError,
-  UnsupportedFunctionalityError as UnsupportedFunctionalityError3
-} from "@ai-sdk/provider";
 import {
   combineHeaders,
   createJsonErrorResponseHandler,
   createJsonResponseHandler,
+  parseProviderOptions as parseProviderOptions2,
   postJsonToApi,
   resolve
 } from "@ai-sdk/provider-utils";
-import { z as z2 } from "zod";
+import { z as z3 } from "zod";
 
 // src/bedrock-api-types.ts
 var BEDROCK_CACHE_POINT = {
@@ -37,11 +34,26 @@ var BEDROCK_STOP_REASONS = [
   "tool_use"
 ];
 
-// src/bedrock-error.ts
+// src/bedrock-chat-options.ts
 import { z } from "zod";
-var BedrockErrorSchema = z.object({
-  message: z.string(),
-  type: z.string().nullish()
+var bedrockProviderOptions = z.object({
+  /**
+   * Additional inference parameters that the model supports,
+   * beyond the base set of inference parameters that Converse
+   * supports in the inferenceConfig field
+   */
+  additionalModelRequestFields: z.record(z.any()).optional(),
+  reasoningConfig: z.object({
+    type: z.union([z.literal("enabled"), z.literal("disabled")]).optional(),
+    budgetTokens: z.number().optional()
+  }).optional()
+});
+
+// src/bedrock-error.ts
+import { z as z2 } from "zod";
+var BedrockErrorSchema = z2.object({
+  message: z2.string(),
+  type: z2.string().nullish()
 });
 
 // src/bedrock-event-stream-response-handler.ts
@@ -65,7 +77,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
     responseHeaders,
     value: response.body.pipeThrough(
       new TransformStream({
-        transform(chunk, controller) {
+        async transform(chunk, controller) {
           var _a, _b;
           const newBuffer = new Uint8Array(buffer.length + chunk.length);
           newBuffer.set(buffer);
@@ -86,7 +98,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
               buffer = buffer.slice(totalLength);
               if (((_a = decoded.headers[":message-type"]) == null ? void 0 : _a.value) === "event") {
                 const data = textDecoder.decode(decoded.body);
-                const parsedDataResult = safeParseJSON({ text: data });
+                const parsedDataResult = await safeParseJSON({ text: data });
                 if (!parsedDataResult.success) {
                   controller.enqueue(parsedDataResult);
                   break;
@@ -95,7 +107,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
                 let wrappedData = {
                   [(_b = decoded.headers[":event-type"]) == null ? void 0 : _b.value]: parsedDataResult.value
                 };
-                const validatedWrappedData = safeValidateTypes({
+                const validatedWrappedData = await safeValidateTypes({
                   value: wrappedData,
                   schema: chunkSchema
                 });
@@ -123,9 +135,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 +163,6 @@ function prepareTools(mode) {
       });
     }
   }
-  const toolChoice = mode.toolChoice;
   if (toolChoice == null) {
     return {
       toolConfig: { tools: bedrockTools, toolChoice: void 0 },
@@ -184,7 +197,7 @@ function prepareTools(mode) {
     default: {
       const _exhaustiveCheck = type;
       throw new UnsupportedFunctionalityError({
-        functionality: `Unsupported tool choice type: ${_exhaustiveCheck}`
+        functionality: `tool choice type: ${_exhaustiveCheck}`
       });
     }
   }
@@ -195,16 +208,17 @@ import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 import {
-  convertUint8ArrayToBase64,
-  createIdGenerator
+  convertToBase64,
+  createIdGenerator,
+  parseProviderOptions
 } 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;
+async function convertToBedrockChatMessages(prompt) {
+  var _a, _b, _c, _d;
   const blocks = groupIntoBlocks(prompt);
   let system = [];
   const messages = [];
@@ -221,7 +235,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 +244,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 +256,31 @@ 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: convertToBase64(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: convertToBase64(part.data) }
+                        }
+                      });
+                    }
                     break;
                   }
                 }
@@ -295,12 +297,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 +332,7 @@ function convertToBedrockChatMessages(prompt) {
               throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
             }
           }
-          if (getCachePoint(providerMetadata)) {
+          if (getCachePoint(providerOptions)) {
             bedrockContent.push(BEDROCK_CACHE_POINT);
           }
         }
@@ -364,32 +366,39 @@ function convertToBedrockChatMessages(prompt) {
                 break;
               }
               case "reasoning": {
-                bedrockContent.push({
-                  reasoningContent: {
-                    reasoningText: {
-                      // trim the last text part if it's the last message in the block
-                      // because Bedrock does not allow trailing whitespace
-                      // in pre-filled assistant responses
-                      text: trimIfLast(
-                        isLastBlock,
-                        isLastMessage,
-                        isLastContentPart,
-                        part.text
-                      ),
-                      signature: part.signature
-                    }
-                  }
+                const reasoningMetadata = await parseProviderOptions({
+                  provider: "bedrock",
+                  providerOptions: part.providerOptions,
+                  schema: bedrockReasoningMetadataSchema
                 });
-                break;
-              }
-              case "redacted-reasoning": {
-                bedrockContent.push({
-                  reasoningContent: {
-                    redactedReasoning: {
-                      data: part.data
-                    }
+                if (reasoningMetadata != null) {
+                  if (reasoningMetadata.signature != null) {
+                    bedrockContent.push({
+                      reasoningContent: {
+                        reasoningText: {
+                          // trim the last text part if it's the last message in the block
+                          // because Bedrock does not allow trailing whitespace
+                          // in pre-filled assistant responses
+                          text: trimIfLast(
+                            isLastBlock,
+                            isLastMessage,
+                            isLastContentPart,
+                            part.text
+                          ),
+                          signature: reasoningMetadata.signature
+                        }
+                      }
+                    });
+                  } else if (reasoningMetadata.redactedData != null) {
+                    bedrockContent.push({
+                      reasoningContent: {
+                        redactedReasoning: {
+                          data: reasoningMetadata.redactedData
+                        }
+                      }
+                    });
                   }
-                });
+                }
                 break;
               }
               case "tool-call": {
@@ -404,7 +413,7 @@ function convertToBedrockChatMessages(prompt) {
               }
             }
           }
-          if (getCachePoint(message.providerMetadata)) {
+          if (getCachePoint(message.providerOptions)) {
             bedrockContent.push(BEDROCK_CACHE_POINT);
           }
         }
@@ -492,19 +501,18 @@ function mapBedrockFinishReason(finishReason) {
 
 // src/bedrock-chat-language-model.ts
 var BedrockChatLanguageModel = class {
-  constructor(modelId, settings, config) {
+  constructor(modelId, config) {
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
-    this.specificationVersion = "v1";
+    this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
-    this.defaultObjectGenerationMode = "tool";
-    this.supportsImageUrls = false;
+    this.supportedUrls = {
+      // no supported urls for bedrock
+    };
   }
-  getArgs({
-    mode,
+  async getArgs({
     prompt,
-    maxTokens,
+    maxOutputTokens,
     temperature,
     topP,
     topK,
@@ -513,10 +521,16 @@ var BedrockChatLanguageModel = class {
     stopSequences,
     responseFormat,
     seed,
-    providerMetadata
+    tools,
+    toolChoice,
+    providerOptions
   }) {
-    var _a, _b, _c, _d, _e, _f, _g;
-    const type = mode.type;
+    var _a, _b, _c, _d, _e;
+    const bedrockOptions = (_a = await parseProviderOptions2({
+      provider: "bedrock",
+      providerOptions,
+      schema: bedrockProviderOptions
+    })) != null ? _a : {};
     const warnings = [];
     if (frequencyPenalty != null) {
       warnings.push({
@@ -549,35 +563,25 @@ var BedrockChatLanguageModel = class {
         details: "JSON response format is not supported."
       });
     }
-    const { system, messages } = convertToBedrockChatMessages(prompt);
-    const reasoningConfigOptions = BedrockReasoningConfigOptionsSchema.safeParse(
-      (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.reasoning_config
-    );
-    if (!reasoningConfigOptions.success) {
-      throw new InvalidArgumentError({
-        argument: "providerOptions.bedrock.reasoning_config",
-        message: "invalid reasoning configuration options",
-        cause: reasoningConfigOptions.error
-      });
-    }
-    const isThinking = ((_b = reasoningConfigOptions.data) == null ? void 0 : _b.type) === "enabled";
-    const thinkingBudget = (_e = (_c = reasoningConfigOptions.data) == null ? void 0 : _c.budgetTokens) != null ? _e : (_d = reasoningConfigOptions.data) == null ? void 0 : _d.budget_tokens;
+    const { system, messages } = await convertToBedrockChatMessages(prompt);
+    const isThinking = ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.type) === "enabled";
+    const thinkingBudget = (_c = bedrockOptions.reasoningConfig) == null ? void 0 : _c.budgetTokens;
     const inferenceConfig = {
-      ...maxTokens != null && { maxTokens },
+      ...maxOutputTokens != null && { maxOutputTokens },
       ...temperature != null && { temperature },
       ...topP != null && { topP },
       ...stopSequences != null && { stopSequences }
     };
     if (isThinking && thinkingBudget != null) {
-      if (inferenceConfig.maxTokens != null) {
-        inferenceConfig.maxTokens += thinkingBudget;
+      if (inferenceConfig.maxOutputTokens != null) {
+        inferenceConfig.maxOutputTokens += thinkingBudget;
       } else {
-        inferenceConfig.maxTokens = thinkingBudget + 4096;
+        inferenceConfig.maxOutputTokens = thinkingBudget + 4096;
       }
-      this.settings.additionalModelRequestFields = {
-        ...this.settings.additionalModelRequestFields,
-        reasoning_config: {
-          type: (_f = reasoningConfigOptions.data) == null ? void 0 : _f.type,
+      bedrockOptions.additionalModelRequestFields = {
+        ...bedrockOptions.additionalModelRequestFields,
+        reasoningConfig: {
+          type: (_d = bedrockOptions.reasoningConfig) == null ? void 0 : _d.type,
           budget_tokens: thinkingBudget
         }
       };
@@ -598,62 +602,24 @@ 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: bedrockOptions.additionalModelRequestFields,
+        ...Object.keys(inferenceConfig).length > 0 && {
+          inferenceConfig
+        },
+        ...providerOptions == null ? void 0 : providerOptions.bedrock,
+        ...((_e = toolConfig.tools) == null ? void 0 : _e.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;
-    const { command: args, warnings } = this.getArgs(options);
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+    const { command: args, warnings } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -675,67 +641,78 @@ var BedrockChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const { messages: rawPrompt, ...rawSettings } = args;
+    const content = [];
+    for (const part of response.output.message.content) {
+      if (part.text) {
+        content.push({ type: "text", text: part.text });
+      }
+      if (part.reasoningContent) {
+        if ("reasoningText" in part.reasoningContent) {
+          const reasoning = {
+            type: "reasoning",
+            text: part.reasoningContent.reasoningText.text
+          };
+          if (part.reasoningContent.reasoningText.signature) {
+            reasoning.providerMetadata = {
+              bedrock: {
+                signature: part.reasoningContent.reasoningText.signature
+              }
+            };
+          }
+          content.push(reasoning);
+        } else if ("redactedReasoning" in part.reasoningContent) {
+          content.push({
+            type: "reasoning",
+            text: "",
+            providerMetadata: {
+              bedrock: {
+                redactedData: (_a = part.reasoningContent.redactedReasoning.data) != null ? _a : ""
+              }
+            }
+          });
+        }
+      }
+      if (part.toolUse) {
+        content.push({
+          type: "tool-call",
+          toolCallType: "function",
+          toolCallId: (_c = (_b = part.toolUse) == null ? void 0 : _b.toolUseId) != null ? _c : this.config.generateId(),
+          toolName: (_e = (_d = part.toolUse) == null ? void 0 : _d.name) != null ? _e : `tool-${this.config.generateId()}`,
+          args: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : "")
+        });
+      }
+    }
     const providerMetadata = response.trace || response.usage ? {
       bedrock: {
         ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
-        ...response.usage && {
+        ...((_h = response.usage) == null ? void 0 : _h.cacheWriteInputTokens) != null && {
           usage: {
-            cacheReadInputTokens: (_b = (_a = response.usage) == null ? void 0 : _a.cacheReadInputTokens) != null ? _b : Number.NaN,
-            cacheWriteInputTokens: (_d = (_c = response.usage) == null ? void 0 : _c.cacheWriteInputTokens) != null ? _d : Number.NaN
+            cacheWriteInputTokens: response.usage.cacheWriteInputTokens
           }
         }
       }
     } : void 0;
-    const reasoning = response.output.message.content.filter((content) => content.reasoningContent).map((content) => {
-      var _a2;
-      if (content.reasoningContent && "reasoningText" in content.reasoningContent) {
-        return {
-          type: "text",
-          text: content.reasoningContent.reasoningText.text,
-          ...content.reasoningContent.reasoningText.signature && {
-            signature: content.reasoningContent.reasoningText.signature
-          }
-        };
-      } else if (content.reasoningContent && "redactedReasoning" in content.reasoningContent) {
-        return {
-          type: "redacted",
-          data: (_a2 = content.reasoningContent.redactedReasoning.data) != null ? _a2 : ""
-        };
-      } else {
-        return void 0;
-      }
-    }).filter((item) => item !== void 0);
     return {
-      text: (_h = (_g = (_f = (_e = response.output) == null ? void 0 : _e.message) == null ? void 0 : _f.content) == null ? void 0 : _g.map((part) => {
-        var _a2;
-        return (_a2 = part.text) != null ? _a2 : "";
-      }).join("")) != null ? _h : void 0,
-      toolCalls: (_l = (_k = (_j = (_i = response.output) == null ? void 0 : _i.message) == null ? void 0 : _j.content) == null ? void 0 : _k.filter((part) => !!part.toolUse)) == null ? void 0 : _l.map((part) => {
-        var _a2, _b2, _c2, _d2, _e2, _f2;
-        return {
-          toolCallType: "function",
-          toolCallId: (_b2 = (_a2 = part.toolUse) == null ? void 0 : _a2.toolUseId) != null ? _b2 : this.config.generateId(),
-          toolName: (_d2 = (_c2 = part.toolUse) == null ? void 0 : _c2.name) != null ? _d2 : `tool-${this.config.generateId()}`,
-          args: JSON.stringify((_f2 = (_e2 = part.toolUse) == null ? void 0 : _e2.input) != null ? _f2 : "")
-        };
-      }),
+      content,
       finishReason: mapBedrockFinishReason(
         response.stopReason
       ),
       usage: {
-        promptTokens: (_n = (_m = response.usage) == null ? void 0 : _m.inputTokens) != null ? _n : Number.NaN,
-        completionTokens: (_p = (_o = response.usage) == null ? void 0 : _o.outputTokens) != null ? _p : Number.NaN
+        inputTokens: (_i = response.usage) == null ? void 0 : _i.inputTokens,
+        outputTokens: (_j = response.usage) == null ? void 0 : _j.outputTokens,
+        totalTokens: ((_k = response.usage) == null ? void 0 : _k.inputTokens) + ((_l = response.usage) == null ? void 0 : _l.outputTokens),
+        cachedInputTokens: (_n = (_m = response.usage) == null ? void 0 : _m.cacheReadInputTokens) != null ? _n : void 0
+      },
+      response: {
+        // TODO add id, timestamp, etc
+        headers: responseHeaders
       },
-      rawCall: { rawPrompt, rawSettings },
-      rawResponse: { headers: responseHeaders },
       warnings,
-      reasoning: reasoning.length > 0 ? reasoning : void 0,
       ...providerMetadata && { providerMetadata }
     };
   }
   async doStream(options) {
-    const { command: args, warnings } = this.getArgs(options);
+    const { command: args, warnings } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse-stream`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -752,19 +729,22 @@ var BedrockChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const { messages: rawPrompt, ...rawSettings } = args;
     let finishReason = "unknown";
-    let usage = {
-      promptTokens: Number.NaN,
-      completionTokens: Number.NaN
+    const usage = {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0
     };
     let providerMetadata = void 0;
     const toolCallContentBlocks = {};
     return {
       stream: response.pipeThrough(
         new TransformStream({
+          start(controller) {
+            controller.enqueue({ type: "stream-start", warnings });
+          },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
             function enqueueError(bedrockError) {
               finishReason = "error";
               controller.enqueue({ type: "error", error: bedrockError });
@@ -796,14 +776,13 @@ var BedrockChatLanguageModel = class {
               );
             }
             if (value.metadata) {
-              usage = {
-                promptTokens: (_b = (_a = value.metadata.usage) == null ? void 0 : _a.inputTokens) != null ? _b : Number.NaN,
-                completionTokens: (_d = (_c = value.metadata.usage) == null ? void 0 : _c.outputTokens) != null ? _d : Number.NaN
-              };
-              const cacheUsage = ((_e = value.metadata.usage) == null ? void 0 : _e.cacheReadInputTokens) != null || ((_f = value.metadata.usage) == null ? void 0 : _f.cacheWriteInputTokens) != null ? {
+              usage.inputTokens = (_b = (_a = value.metadata.usage) == null ? void 0 : _a.inputTokens) != null ? _b : usage.inputTokens;
+              usage.outputTokens = (_d = (_c = value.metadata.usage) == null ? void 0 : _c.outputTokens) != null ? _d : usage.outputTokens;
+              usage.totalTokens = ((_e = usage.inputTokens) != null ? _e : 0) + ((_f = usage.outputTokens) != null ? _f : 0);
+              usage.cachedInputTokens = (_h = (_g = value.metadata.usage) == null ? void 0 : _g.cacheReadInputTokens) != null ? _h : usage.cachedInputTokens;
+              const cacheUsage = ((_i = value.metadata.usage) == null ? void 0 : _i.cacheWriteInputTokens) != null ? {
                 usage: {
-                  cacheReadInputTokens: (_h = (_g = value.metadata.usage) == null ? void 0 : _g.cacheReadInputTokens) != null ? _h : Number.NaN,
-                  cacheWriteInputTokens: (_j = (_i = value.metadata.usage) == null ? void 0 : _i.cacheWriteInputTokens) != null ? _j : Number.NaN
+                  cacheWriteInputTokens: value.metadata.usage.cacheWriteInputTokens
                 }
               } : void 0;
               const trace = value.metadata.trace ? {
@@ -818,33 +797,45 @@ var BedrockChatLanguageModel = class {
                 };
               }
             }
-            if (((_k = value.contentBlockDelta) == null ? void 0 : _k.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
+            if (((_j = value.contentBlockDelta) == null ? void 0 : _j.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
               controller.enqueue({
-                type: "text-delta",
-                textDelta: value.contentBlockDelta.delta.text
+                type: "text",
+                text: value.contentBlockDelta.delta.text
               });
             }
-            if (((_l = value.contentBlockDelta) == null ? void 0 : _l.delta) && "reasoningContent" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.reasoningContent) {
+            if (((_k = value.contentBlockDelta) == null ? void 0 : _k.delta) && "reasoningContent" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.reasoningContent) {
               const reasoningContent = value.contentBlockDelta.delta.reasoningContent;
               if ("text" in reasoningContent && reasoningContent.text) {
                 controller.enqueue({
                   type: "reasoning",
-                  textDelta: reasoningContent.text
+                  text: reasoningContent.text
                 });
               } else if ("signature" in reasoningContent && reasoningContent.signature) {
                 controller.enqueue({
-                  type: "reasoning-signature",
-                  signature: reasoningContent.signature
+                  type: "reasoning",
+                  text: "",
+                  providerMetadata: {
+                    bedrock: {
+                      signature: reasoningContent.signature
+                    }
+                  }
                 });
+                controller.enqueue({ type: "reasoning-part-finish" });
               } else if ("data" in reasoningContent && reasoningContent.data) {
                 controller.enqueue({
-                  type: "redacted-reasoning",
-                  data: reasoningContent.data
+                  type: "reasoning",
+                  text: "",
+                  providerMetadata: {
+                    bedrock: {
+                      redactedData: reasoningContent.data
+                    }
+                  }
                 });
+                controller.enqueue({ type: "reasoning-part-finish" });
               }
             }
             const contentBlockStart = value.contentBlockStart;
-            if (((_m = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _m.toolUse) != null) {
+            if (((_l = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _l.toolUse) != null) {
               const toolUse = contentBlockStart.start.toolUse;
               toolCallContentBlocks[contentBlockStart.contentBlockIndex] = {
                 toolCallId: toolUse.toolUseId,
@@ -855,7 +846,7 @@ var BedrockChatLanguageModel = class {
             const contentBlockDelta = value.contentBlockDelta;
             if ((contentBlockDelta == null ? void 0 : contentBlockDelta.delta) && "toolUse" in contentBlockDelta.delta && contentBlockDelta.delta.toolUse) {
               const contentBlock = toolCallContentBlocks[contentBlockDelta.contentBlockIndex];
-              const delta = (_n = contentBlockDelta.delta.toolUse.input) != null ? _n : "";
+              const delta = (_m = contentBlockDelta.delta.toolUse.input) != null ? _m : "";
               controller.enqueue({
                 type: "tool-call-delta",
                 toolCallType: "function",
@@ -891,9 +882,8 @@ var BedrockChatLanguageModel = class {
           }
         })
       ),
-      rawCall: { rawPrompt, rawSettings },
-      rawResponse: { headers: responseHeaders },
-      warnings
+      // TODO request?
+      response: { headers: responseHeaders }
     };
   }
   getUrl(modelId) {
@@ -901,124 +891,143 @@ var BedrockChatLanguageModel = class {
     return `${this.config.baseUrl()}/model/${encodedModelId}`;
   }
 };
-var BedrockReasoningConfigOptionsSchema = z2.object({
-  type: z2.union([z2.literal("enabled"), z2.literal("disabled")]).nullish(),
-  budget_tokens: z2.number().nullish(),
-  budgetTokens: z2.number().nullish()
-}).nullish();
-var BedrockStopReasonSchema = z2.union([
-  z2.enum(BEDROCK_STOP_REASONS),
-  z2.string()
+var BedrockStopReasonSchema = z3.union([
+  z3.enum(BEDROCK_STOP_REASONS),
+  z3.string()
 ]);
-var BedrockToolUseSchema = z2.object({
-  toolUseId: z2.string(),
-  name: z2.string(),
-  input: z2.unknown()
+var BedrockToolUseSchema = z3.object({
+  toolUseId: z3.string(),
+  name: z3.string(),
+  input: z3.unknown()
 });
-var BedrockReasoningTextSchema = z2.object({
-  signature: z2.string().nullish(),
-  text: z2.string()
+var BedrockReasoningTextSchema = z3.object({
+  signature: z3.string().nullish(),
+  text: z3.string()
 });
-var BedrockRedactedReasoningSchema = z2.object({
-  data: z2.string()
+var BedrockRedactedReasoningSchema = z3.object({
+  data: z3.string()
 });
-var BedrockResponseSchema = z2.object({
-  metrics: z2.object({
-    latencyMs: z2.number()
+var BedrockResponseSchema = z3.object({
+  metrics: z3.object({
+    latencyMs: z3.number()
   }).nullish(),
-  output: z2.object({
-    message: z2.object({
-      content: z2.array(
-        z2.object({
-          text: z2.string().nullish(),
+  output: z3.object({
+    message: z3.object({
+      content: z3.array(
+        z3.object({
+          text: z3.string().nullish(),
           toolUse: BedrockToolUseSchema.nullish(),
-          reasoningContent: z2.union([
-            z2.object({
+          reasoningContent: z3.union([
+            z3.object({
               reasoningText: BedrockReasoningTextSchema
             }),
-            z2.object({
+            z3.object({
               redactedReasoning: BedrockRedactedReasoningSchema
             })
           ]).nullish()
         })
       ),
-      role: z2.string()
+      role: z3.string()
     })
   }),
   stopReason: BedrockStopReasonSchema,
-  trace: z2.unknown().nullish(),
-  usage: z2.object({
-    inputTokens: z2.number(),
-    outputTokens: z2.number(),
-    totalTokens: z2.number(),
-    cacheReadInputTokens: z2.number().nullish(),
-    cacheWriteInputTokens: z2.number().nullish()
+  trace: z3.unknown().nullish(),
+  usage: z3.object({
+    inputTokens: z3.number(),
+    outputTokens: z3.number(),
+    totalTokens: z3.number(),
+    cacheReadInputTokens: z3.number().nullish(),
+    cacheWriteInputTokens: z3.number().nullish()
   })
 });
-var BedrockStreamSchema = z2.object({
-  contentBlockDelta: z2.object({
-    contentBlockIndex: z2.number(),
-    delta: z2.union([
-      z2.object({ text: z2.string() }),
-      z2.object({ toolUse: z2.object({ input: z2.string() }) }),
-      z2.object({
-        reasoningContent: z2.object({ text: z2.string() })
+var BedrockStreamSchema = z3.object({
+  contentBlockDelta: z3.object({
+    contentBlockIndex: z3.number(),
+    delta: z3.union([
+      z3.object({ text: z3.string() }),
+      z3.object({ toolUse: z3.object({ input: z3.string() }) }),
+      z3.object({
+        reasoningContent: z3.object({ text: z3.string() })
       }),
-      z2.object({
-        reasoningContent: z2.object({
-          signature: z2.string()
+      z3.object({
+        reasoningContent: z3.object({
+          signature: z3.string()
         })
       }),
-      z2.object({
-        reasoningContent: z2.object({ data: z2.string() })
+      z3.object({
+        reasoningContent: z3.object({ data: z3.string() })
       })
     ]).nullish()
   }).nullish(),
-  contentBlockStart: z2.object({
-    contentBlockIndex: z2.number(),
-    start: z2.object({
+  contentBlockStart: z3.object({
+    contentBlockIndex: z3.number(),
+    start: z3.object({
       toolUse: BedrockToolUseSchema.nullish()
     }).nullish()
   }).nullish(),
-  contentBlockStop: z2.object({
-    contentBlockIndex: z2.number()
+  contentBlockStop: z3.object({
+    contentBlockIndex: z3.number()
   }).nullish(),
-  internalServerException: z2.record(z2.unknown()).nullish(),
-  messageStop: z2.object({
-    additionalModelResponseFields: z2.record(z2.unknown()).nullish(),
+  internalServerException: z3.record(z3.unknown()).nullish(),
+  messageStop: z3.object({
+    additionalModelResponseFields: z3.record(z3.unknown()).nullish(),
     stopReason: BedrockStopReasonSchema
   }).nullish(),
-  metadata: z2.object({
-    trace: z2.unknown().nullish(),
-    usage: z2.object({
-      cacheReadInputTokens: z2.number().nullish(),
-      cacheWriteInputTokens: z2.number().nullish(),
-      inputTokens: z2.number(),
-      outputTokens: z2.number()
+  metadata: z3.object({
+    trace: z3.unknown().nullish(),
+    usage: z3.object({
+      cacheReadInputTokens: z3.number().nullish(),
+      cacheWriteInputTokens: z3.number().nullish(),
+      inputTokens: z3.number(),
+      outputTokens: z3.number()
     }).nullish()
   }).nullish(),
-  modelStreamErrorException: z2.record(z2.unknown()).nullish(),
-  throttlingException: z2.record(z2.unknown()).nullish(),
-  validationException: z2.record(z2.unknown()).nullish()
+  modelStreamErrorException: z3.record(z3.unknown()).nullish(),
+  throttlingException: z3.record(z3.unknown()).nullish(),
+  validationException: z3.record(z3.unknown()).nullish()
+});
+var bedrockReasoningMetadataSchema = z3.object({
+  signature: z3.string().optional(),
+  redactedData: z3.string().optional()
 });
 
 // src/bedrock-embedding-model.ts
+import {
+  TooManyEmbeddingValuesForCallError
+} from "@ai-sdk/provider";
 import {
   combineHeaders as combineHeaders2,
   createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler as createJsonResponseHandler2,
+  parseProviderOptions as parseProviderOptions3,
   postJsonToApi as postJsonToApi2,
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
-import { z as z3 } from "zod";
+
+// src/bedrock-embedding-options.ts
+import { z as z4 } from "zod";
+var bedrockEmbeddingProviderOptions = z4.object({
+  /**
+  The number of dimensions the resulting output embeddings should have (defaults to 1024).
+  Only supported in amazon.titan-embed-text-v2:0.
+     */
+  dimensions: z4.union([z4.literal(1024), z4.literal(512), z4.literal(256)]).optional(),
+  /**
+  Flag indicating whether or not to normalize the output embeddings. Defaults to true
+  Only supported in amazon.titan-embed-text-v2:0.
+   */
+  normalize: z4.boolean().optional()
+});
+
+// src/bedrock-embedding-model.ts
+import { z as z5 } from "zod";
 var BedrockEmbeddingModel = class {
-  constructor(modelId, settings, config) {
+  constructor(modelId, config) {
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
-    this.specificationVersion = "v1";
+    this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
-    this.maxEmbeddingsPerCall = void 0;
+    this.maxEmbeddingsPerCall = 1;
     this.supportsParallelCalls = true;
   }
   getUrl(modelId) {
@@ -1028,50 +1037,54 @@ var BedrockEmbeddingModel = class {
   async doEmbed({
     values,
     headers,
-    abortSignal
+    abortSignal,
+    providerOptions
   }) {
-    const embedSingleText = async (inputText) => {
-      const args = {
-        inputText,
-        dimensions: this.settings.dimensions,
-        normalize: this.settings.normalize
-      };
-      const url = this.getUrl(this.modelId);
-      const { value: response } = await postJsonToApi2({
-        url,
-        headers: await resolve2(
-          combineHeaders2(await resolve2(this.config.headers), headers)
-        ),
-        body: args,
-        failedResponseHandler: createJsonErrorResponseHandler2({
-          errorSchema: BedrockErrorSchema,
-          errorToMessage: (error) => `${error.type}: ${error.message}`
-        }),
-        successfulResponseHandler: createJsonResponseHandler2(
-          BedrockEmbeddingResponseSchema
-        ),
-        fetch: this.config.fetch,
-        abortSignal
+    var _a;
+    if (values.length > this.maxEmbeddingsPerCall) {
+      throw new TooManyEmbeddingValuesForCallError({
+        provider: this.provider,
+        modelId: this.modelId,
+        maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,
+        values
       });
-      return {
-        embedding: response.embedding,
-        inputTextTokenCount: response.inputTextTokenCount
-      };
+    }
+    const bedrockOptions = (_a = await parseProviderOptions3({
+      provider: "bedrock",
+      providerOptions,
+      schema: bedrockEmbeddingProviderOptions
+    })) != null ? _a : {};
+    const args = {
+      inputText: values[0],
+      dimensions: bedrockOptions.dimensions,
+      normalize: bedrockOptions.normalize
+    };
+    const url = this.getUrl(this.modelId);
+    const { value: response } = await postJsonToApi2({
+      url,
+      headers: await resolve2(
+        combineHeaders2(await resolve2(this.config.headers), headers)
+      ),
+      body: args,
+      failedResponseHandler: createJsonErrorResponseHandler2({
+        errorSchema: BedrockErrorSchema,
+        errorToMessage: (error) => `${error.type}: ${error.message}`
+      }),
+      successfulResponseHandler: createJsonResponseHandler2(
+        BedrockEmbeddingResponseSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal
+    });
+    return {
+      embeddings: [response.embedding],
+      usage: { tokens: response.inputTextTokenCount }
     };
-    const responses = await Promise.all(values.map(embedSingleText));
-    return responses.reduce(
-      (accumulated, response) => {
-        accumulated.embeddings.push(response.embedding);
-        accumulated.usage.tokens += response.inputTextTokenCount;
-        return accumulated;
-      },
-      { embeddings: [], usage: { tokens: 0 } }
-    );
   }
 };
-var BedrockEmbeddingResponseSchema = z3.object({
-  embedding: z3.array(z3.number()),
-  inputTextTokenCount: z3.number()
+var BedrockEmbeddingResponseSchema = z5.object({
+  embedding: z5.array(z5.number()),
+  inputTextTokenCount: z5.number()
 });
 
 // src/bedrock-image-model.ts
@@ -1089,18 +1102,17 @@ var modelMaxImagesPerCall = {
 };
 
 // src/bedrock-image-model.ts
-import { z as z4 } from "zod";
+import { z as z6 } from "zod";
 var BedrockImageModel = class {
-  constructor(modelId, settings, config) {
+  constructor(modelId, config) {
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
-    this.specificationVersion = "v1";
+    this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
   }
   get maxImagesPerCall() {
-    var _a, _b;
-    return (_b = (_a = this.settings.maxImagesPerCall) != null ? _a : modelMaxImagesPerCall[this.modelId]) != null ? _b : 1;
+    var _a;
+    return (_a = modelMaxImagesPerCall[this.modelId]) != null ? _a : 1;
   }
   getUrl(modelId) {
     const encodedModelId = encodeURIComponent(modelId);
@@ -1171,8 +1183,8 @@ var BedrockImageModel = class {
     };
   }
 };
-var bedrockImageResponseSchema = z4.object({
-  images: z4.array(z4.string())
+var bedrockImageResponseSchema = z6.object({
+  images: z6.array(z6.string())
 });
 
 // src/headers-utils.ts
@@ -1297,34 +1309,34 @@ function createAmazonBedrock(options = {}) {
       })}.amazonaws.com`
     )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
   };
-  const createChatModel = (modelId, settings = {}) => {
+  const createChatModel = (modelId) => {
     var _a;
-    return new BedrockChatLanguageModel(modelId, settings, {
+    return new BedrockChatLanguageModel(modelId, {
       baseUrl: getBaseUrl,
       headers: (_a = options.headers) != null ? _a : {},
       fetch: sigv4Fetch,
       generateId
     });
   };
-  const provider = function(modelId, settings) {
+  const provider = function(modelId) {
     if (new.target) {
       throw new Error(
         "The Amazon Bedrock model function cannot be called with the new keyword."
       );
     }
-    return createChatModel(modelId, settings);
+    return createChatModel(modelId);
   };
-  const createEmbeddingModel = (modelId, settings = {}) => {
+  const createEmbeddingModel = (modelId) => {
     var _a;
-    return new BedrockEmbeddingModel(modelId, settings, {
+    return new BedrockEmbeddingModel(modelId, {
       baseUrl: getBaseUrl,
       headers: (_a = options.headers) != null ? _a : {},
       fetch: sigv4Fetch
     });
   };
-  const createImageModel = (modelId, settings = {}) => {
+  const createImageModel = (modelId) => {
     var _a;
-    return new BedrockImageModel(modelId, settings, {
+    return new BedrockImageModel(modelId, {
       baseUrl: getBaseUrl,
       headers: (_a = options.headers) != null ? _a : {},
       fetch: sigv4Fetch