@ai-sdk/amazon-bedrock 3.1.0-beta.9 → 4.0.0-beta.100

package/dist/index.mjs

@@ -1,4 +1,5 @@
 // src/bedrock-provider.ts
+import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 import {
   generateId,
   loadOptionalSetting,
@@ -7,12 +8,6 @@ import {
   withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
 
-// src/version.ts
-var VERSION = true ? "3.1.0-beta.9" : "0.0.0-test";
-
-// src/bedrock-provider.ts
-import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
-
 // src/bedrock-chat-language-model.ts
 import {
   combineHeaders,
@@ -60,6 +55,18 @@ var BEDROCK_DOCUMENT_MIME_TYPES = {
 
 // src/bedrock-chat-options.ts
 import { z } from "zod/v4";
+var bedrockFilePartProviderOptions = z.object({
+  /**
+   * Citation configuration for this document.
+   * When enabled, this document will generate citations in the response.
+   */
+  citations: z.object({
+    /**
+     * Enable citations for this document
+     */
+    enabled: z.boolean()
+  }).optional()
+});
 var bedrockProviderOptions = z.object({
   /**
    * Additional inference parameters that the model supports,
@@ -69,8 +76,13 @@ var bedrockProviderOptions = z.object({
   additionalModelRequestFields: z.record(z.string(), z.any()).optional(),
   reasoningConfig: z.object({
     type: z.union([z.literal("enabled"), z.literal("disabled")]).optional(),
-    budgetTokens: z.number().optional()
-  }).optional()
+    budgetTokens: z.number().optional(),
+    maxReasoningEffort: z.enum(["low", "medium", "high"]).optional()
+  }).optional(),
+  /**
+   * Anthropic beta features to enable
+   */
+  anthropicBeta: z.array(z.string()).optional()
 });
 
 // src/bedrock-error.ts
@@ -80,6 +92,45 @@ var BedrockErrorSchema = z2.object({
   type: z2.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
 import { EmptyResponseBodyError } from "@ai-sdk/provider";
 import {
@@ -164,11 +215,12 @@ import {
   anthropicTools,
   prepareTools as prepareAnthropicTools
 } from "@ai-sdk/anthropic/internal";
-function prepareTools({
+async function prepareTools({
   tools,
   toolChoice,
   modelId
 }) {
+  var _a;
   const toolWarnings = [];
   const betas = /* @__PURE__ */ new Set();
   if (tools == null || tools.length === 0) {
@@ -180,10 +232,10 @@ function prepareTools({
     };
   }
   const supportedTools = tools.filter((tool) => {
-    if (tool.type === "provider-defined" && tool.id === "anthropic.web_search_20250305") {
+    if (tool.type === "provider" && tool.id === "anthropic.web_search_20250305") {
       toolWarnings.push({
-        type: "unsupported-tool",
-        tool,
+        type: "unsupported",
+        feature: "web_search_20250305 tool",
         details: "The web_search_20250305 tool is not supported on Amazon Bedrock."
       });
       return false;
@@ -199,18 +251,16 @@ function prepareTools({
     };
   }
   const isAnthropicModel = modelId.includes("anthropic.");
-  const providerDefinedTools = supportedTools.filter(
-    (t) => t.type === "provider-defined"
-  );
+  const ProviderTools = supportedTools.filter((t) => t.type === "provider");
   const functionTools = supportedTools.filter((t) => t.type === "function");
   let additionalTools = void 0;
   const bedrockTools = [];
-  const usingAnthropicTools = isAnthropicModel && providerDefinedTools.length > 0;
+  const usingAnthropicTools = isAnthropicModel && ProviderTools.length > 0;
   if (usingAnthropicTools) {
     if (functionTools.length > 0) {
       toolWarnings.push({
-        type: "unsupported-setting",
-        setting: "tools",
+        type: "unsupported",
+        feature: "mixing Anthropic provider-defined tools and standard function tools",
         details: "Mixed Anthropic provider-defined tools and standard function tools are not supported in a single call to Bedrock. Only Anthropic tools will be used."
       });
     }
@@ -218,9 +268,10 @@ function prepareTools({
       toolChoice: preparedAnthropicToolChoice,
       toolWarnings: anthropicToolWarnings,
       betas: anthropicBetas
-    } = prepareAnthropicTools({
-      tools: providerDefinedTools,
-      toolChoice
+    } = await prepareAnthropicTools({
+      tools: ProviderTools,
+      toolChoice,
+      supportsStructuredOutput: false
     });
     toolWarnings.push(...anthropicToolWarnings);
     anthropicBetas.forEach((beta) => betas.add(beta));
@@ -229,7 +280,7 @@ function prepareTools({
         tool_choice: preparedAnthropicToolChoice
       };
     }
-    for (const tool of providerDefinedTools) {
+    for (const tool of ProviderTools) {
       const toolFactory = Object.values(anthropicTools).find((factory) => {
         const instance = factory({});
         return instance.id === tool.id;
@@ -240,24 +291,24 @@ function prepareTools({
           toolSpec: {
             name: tool.name,
             inputSchema: {
-              json: asSchema(fullToolDefinition.inputSchema).jsonSchema
+              json: await asSchema(fullToolDefinition.inputSchema).jsonSchema
             }
           }
         });
       } else {
-        toolWarnings.push({ type: "unsupported-tool", tool });
+        toolWarnings.push({ type: "unsupported", feature: "tool ${tool.id}" });
       }
     }
   } else {
-    for (const tool of providerDefinedTools) {
-      toolWarnings.push({ type: "unsupported-tool", tool });
+    for (const tool of ProviderTools) {
+      toolWarnings.push({ type: "unsupported", feature: `tool ${tool.id}` });
     }
   }
   for (const tool of functionTools) {
     bedrockTools.push({
       toolSpec: {
         name: tool.name,
-        description: tool.description,
+        ...((_a = tool.description) == null ? void 0 : _a.trim()) !== "" ? { description: tool.description } : {},
         inputSchema: {
           json: tool.inputSchema
         }
@@ -307,7 +358,17 @@ function getCachePoint(providerMetadata) {
   var _a;
   return (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.cachePoint;
 }
+async function shouldEnableCitations(providerMetadata) {
+  var _a, _b;
+  const bedrockOptions = await parseProviderOptions({
+    provider: "bedrock",
+    providerOptions: providerMetadata,
+    schema: bedrockFilePartProviderOptions
+  });
+  return (_b = (_a = bedrockOptions == null ? void 0 : bedrockOptions.citations) == null ? void 0 : _a.enabled) != null ? _b : false;
+}
 async function convertToBedrockChatMessages(prompt) {
+  var _a, _b;
   const blocks = groupIntoBlocks(prompt);
   let system = [];
   const messages = [];
@@ -367,11 +428,17 @@ async function convertToBedrockChatMessages(prompt) {
                           message: "File mime type is required in user message part content"
                         });
                       }
+                      const enableCitations = await shouldEnableCitations(
+                        part.providerOptions
+                      );
                       bedrockContent.push({
                         document: {
                           format: getBedrockDocumentFormat(part.mediaType),
-                          name: generateDocumentName(),
-                          source: { bytes: convertToBase64(part.data) }
+                          name: (_a = part.filename) != null ? _a : generateDocumentName(),
+                          source: { bytes: convertToBase64(part.data) },
+                          ...enableCitations && {
+                            citations: { enabled: true }
+                          }
                         }
                       });
                     }
@@ -391,7 +458,7 @@ async function convertToBedrockChatMessages(prompt) {
                       switch (contentPart.type) {
                         case "text":
                           return { text: contentPart.text };
-                        case "media":
+                        case "image-data":
                           if (!contentPart.mediaType.startsWith("image/")) {
                             throw new UnsupportedFunctionalityError2({
                               functionality: `media type: ${contentPart.mediaType}`
@@ -406,6 +473,11 @@ async function convertToBedrockChatMessages(prompt) {
                               source: { bytes: contentPart.data }
                             }
                           };
+                        default: {
+                          throw new UnsupportedFunctionalityError2({
+                            functionality: `unsupported tool content part type: ${contentPart.type}`
+                          });
+                        }
                       }
                     });
                     break;
@@ -414,6 +486,11 @@ async function convertToBedrockChatMessages(prompt) {
                   case "error-text":
                     toolResultContent = [{ text: output.value }];
                     break;
+                  case "execution-denied":
+                    toolResultContent = [
+                      { text: (_b = output.reason) != null ? _b : "Tool execution denied." }
+                    ];
+                    break;
                   case "json":
                   case "error-json":
                   default:
@@ -612,7 +689,7 @@ function groupIntoBlocks(prompt) {
 }
 
 // src/map-bedrock-finish-reason.ts
-function mapBedrockFinishReason(finishReason) {
+function mapBedrockFinishReason(finishReason, isJsonResponseFromTool) {
   switch (finishReason) {
     case "stop_sequence":
     case "end_turn":
@@ -623,7 +700,7 @@ function mapBedrockFinishReason(finishReason) {
     case "guardrail_intervened":
       return "content-filter";
     case "tool_use":
-      return "tool-calls";
+      return isJsonResponseFromTool ? "stop" : "tool-calls";
     default:
       return "unknown";
   }
@@ -655,7 +732,7 @@ var BedrockChatLanguageModel = class {
     toolChoice,
     providerOptions
   }) {
-    var _a, _b, _c, _d, _e, _f;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
     const bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
@@ -664,46 +741,53 @@ var BedrockChatLanguageModel = class {
     const warnings = [];
     if (frequencyPenalty != null) {
       warnings.push({
-        type: "unsupported-setting",
-        setting: "frequencyPenalty"
+        type: "unsupported",
+        feature: "frequencyPenalty"
       });
     }
     if (presencePenalty != null) {
       warnings.push({
-        type: "unsupported-setting",
-        setting: "presencePenalty"
+        type: "unsupported",
+        feature: "presencePenalty"
       });
     }
     if (seed != null) {
       warnings.push({
-        type: "unsupported-setting",
-        setting: "seed"
+        type: "unsupported",
+        feature: "seed"
+      });
+    }
+    if (temperature != null && temperature > 1) {
+      warnings.push({
+        type: "unsupported",
+        feature: "temperature",
+        details: `${temperature} exceeds bedrock maximum of 1.0. clamped to 1.0`
+      });
+      temperature = 1;
+    } else if (temperature != null && temperature < 0) {
+      warnings.push({
+        type: "unsupported",
+        feature: "temperature",
+        details: `${temperature} is below bedrock minimum of 0. clamped to 0`
       });
+      temperature = 0;
     }
     if (responseFormat != null && responseFormat.type !== "text" && responseFormat.type !== "json") {
       warnings.push({
-        type: "unsupported-setting",
-        setting: "responseFormat",
+        type: "unsupported",
+        feature: "responseFormat",
         details: "Only text and json response formats are supported."
       });
     }
-    if (tools != null && (responseFormat == null ? void 0 : responseFormat.type) === "json") {
-      if (tools.length > 0) {
-        warnings.push({
-          type: "other",
-          message: "JSON response format does not support tools. The provided tools are ignored."
-        });
-      }
-    }
     const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null ? {
       type: "function",
       name: "json",
       description: "Respond with a JSON object.",
       inputSchema: responseFormat.schema
     } : void 0;
-    const { toolConfig, additionalTools, toolWarnings, betas } = prepareTools({
-      tools: jsonResponseTool ? [jsonResponseTool, ...tools != null ? tools : []] : tools,
-      toolChoice: jsonResponseTool != null ? { type: "tool", toolName: jsonResponseTool.name } : toolChoice,
+    const { toolConfig, additionalTools, toolWarnings, betas } = await prepareTools({
+      tools: jsonResponseTool ? [...tools != null ? tools : [], jsonResponseTool] : tools,
+      toolChoice: jsonResponseTool != null ? { type: "required" } : toolChoice,
       modelId: this.modelId
     });
     warnings.push(...toolWarnings);
@@ -713,8 +797,18 @@ var BedrockChatLanguageModel = class {
         ...additionalTools
       };
     }
-    const isThinking = ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.type) === "enabled";
-    const thinkingBudget = (_c = bedrockOptions.reasoningConfig) == null ? void 0 : _c.budgetTokens;
+    if (betas.size > 0 || bedrockOptions.anthropicBeta) {
+      const existingBetas = (_b = bedrockOptions.anthropicBeta) != null ? _b : [];
+      const mergedBetas = betas.size > 0 ? [...existingBetas, ...Array.from(betas)] : existingBetas;
+      bedrockOptions.additionalModelRequestFields = {
+        ...bedrockOptions.additionalModelRequestFields,
+        anthropic_beta: mergedBetas
+      };
+    }
+    const isAnthropicModel = this.modelId.includes("anthropic");
+    const isThinkingRequested = ((_c = bedrockOptions.reasoningConfig) == null ? void 0 : _c.type) === "enabled";
+    const thinkingBudget = (_d = bedrockOptions.reasoningConfig) == null ? void 0 : _d.budgetTokens;
+    const isAnthropicThinkingEnabled = isAnthropicModel && isThinkingRequested;
     const inferenceConfig = {
       ...maxOutputTokens != null && { maxTokens: maxOutputTokens },
       ...temperature != null && { temperature },
@@ -722,7 +816,7 @@ var BedrockChatLanguageModel = class {
       ...topK != null && { topK },
       ...stopSequences != null && { stopSequences }
     };
-    if (isThinking && thinkingBudget != null) {
+    if (isAnthropicThinkingEnabled && thinkingBudget != null) {
       if (inferenceConfig.maxTokens != null) {
         inferenceConfig.maxTokens += thinkingBudget;
       } else {
@@ -731,36 +825,60 @@ var BedrockChatLanguageModel = class {
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         thinking: {
-          type: (_d = bedrockOptions.reasoningConfig) == null ? void 0 : _d.type,
+          type: (_e = bedrockOptions.reasoningConfig) == null ? void 0 : _e.type,
           budget_tokens: thinkingBudget
         }
       };
+    } else if (!isAnthropicModel && thinkingBudget != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "budgetTokens",
+        details: "budgetTokens applies only to Anthropic models on Bedrock and will be ignored for this model."
+      });
+    }
+    const maxReasoningEffort = (_f = bedrockOptions.reasoningConfig) == null ? void 0 : _f.maxReasoningEffort;
+    if (maxReasoningEffort != null && !isAnthropicModel) {
+      bedrockOptions.additionalModelRequestFields = {
+        ...bedrockOptions.additionalModelRequestFields,
+        reasoningConfig: {
+          ...((_g = bedrockOptions.reasoningConfig) == null ? void 0 : _g.type) != null && {
+            type: bedrockOptions.reasoningConfig.type
+          },
+          maxReasoningEffort
+        }
+      };
+    } else if (maxReasoningEffort != null && isAnthropicModel) {
+      warnings.push({
+        type: "unsupported",
+        feature: "maxReasoningEffort",
+        details: "maxReasoningEffort applies only to Amazon Nova models on Bedrock and will be ignored for this model."
+      });
     }
-    if (isThinking && inferenceConfig.temperature != null) {
+    if (isAnthropicThinkingEnabled && inferenceConfig.temperature != null) {
       delete inferenceConfig.temperature;
       warnings.push({
-        type: "unsupported-setting",
-        setting: "temperature",
+        type: "unsupported",
+        feature: "temperature",
         details: "temperature is not supported when thinking is enabled"
       });
     }
-    if (isThinking && inferenceConfig.topP != null) {
+    if (isAnthropicThinkingEnabled && inferenceConfig.topP != null) {
       delete inferenceConfig.topP;
       warnings.push({
-        type: "unsupported-setting",
-        setting: "topP",
+        type: "unsupported",
+        feature: "topP",
         details: "topP is not supported when thinking is enabled"
       });
     }
-    if (isThinking && inferenceConfig.topK != null) {
+    if (isAnthropicThinkingEnabled && inferenceConfig.topK != null) {
       delete inferenceConfig.topK;
       warnings.push({
-        type: "unsupported-setting",
-        setting: "topK",
+        type: "unsupported",
+        feature: "topK",
         details: "topK is not supported when thinking is enabled"
       });
     }
-    const hasAnyTools = ((_f = (_e = toolConfig.tools) == null ? void 0 : _e.length) != null ? _f : 0) > 0 || additionalTools;
+    const hasAnyTools = ((_i = (_h = toolConfig.tools) == null ? void 0 : _h.length) != null ? _i : 0) > 0 || additionalTools;
     let filteredPrompt = prompt;
     if (!hasAnyTools) {
       const hasToolContent = prompt.some(
@@ -780,14 +898,18 @@ var BedrockChatLanguageModel = class {
           (message) => message.role === "system" || message.content.length > 0
         );
         warnings.push({
-          type: "unsupported-setting",
-          setting: "toolContent",
+          type: "unsupported",
+          feature: "toolContent",
           details: "Tool calls and results removed from conversation because Bedrock does not support tool content without active tools."
         });
       }
     }
     const { system, messages } = await convertToBedrockChatMessages(filteredPrompt);
-    const { reasoningConfig: _, ...filteredBedrockOptions } = (providerOptions == null ? void 0 : providerOptions.bedrock) || {};
+    const {
+      reasoningConfig: _,
+      additionalModelRequestFields: __,
+      ...filteredBedrockOptions
+    } = (providerOptions == null ? void 0 : providerOptions.bedrock) || {};
     return {
       command: {
         system,
@@ -805,27 +927,21 @@ var BedrockChatLanguageModel = class {
     };
   }
   async getHeaders({
-    betas,
     headers
   }) {
-    return combineHeaders(
-      await resolve(this.config.headers),
-      betas.size > 0 ? { "anthropic-beta": Array.from(betas).join(",") } : {},
-      headers
-    );
+    return combineHeaders(await resolve(this.config.headers), headers);
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const {
       command: args,
       warnings,
-      usesJsonResponseTool,
-      betas
+      usesJsonResponseTool
     } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
-      headers: await this.getHeaders({ betas, headers: options.headers }),
+      headers: await this.getHeaders({ headers: options.headers }),
       body: args,
       failedResponseHandler: createJsonErrorResponseHandler({
         errorSchema: BedrockErrorSchema,
@@ -841,11 +957,10 @@ var BedrockChatLanguageModel = class {
       fetch: this.config.fetch
     });
     const content = [];
+    let isJsonResponseFromTool = false;
     for (const part of response.output.message.content) {
       if (part.text) {
-        if (!usesJsonResponseTool) {
-          content.push({ type: "text", text: part.text });
-        }
+        content.push({ type: "text", text: part.text });
       }
       if (part.reasoningContent) {
         if ("reasoningText" in part.reasoningContent) {
@@ -874,21 +989,24 @@ var BedrockChatLanguageModel = class {
         }
       }
       if (part.toolUse) {
-        content.push(
-          // when a json response tool is used, the tool call becomes the text:
-          usesJsonResponseTool ? {
+        const isJsonResponseTool = usesJsonResponseTool && part.toolUse.name === "json";
+        if (isJsonResponseTool) {
+          isJsonResponseFromTool = true;
+          content.push({
             type: "text",
             text: JSON.stringify(part.toolUse.input)
-          } : {
+          });
+        } else {
+          content.push({
             type: "tool-call",
             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()}`,
-            input: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : "")
-          }
-        );
+            input: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : {})
+          });
+        }
       }
     }
-    const providerMetadata = response.trace || response.usage || usesJsonResponseTool ? {
+    const providerMetadata = response.trace || response.usage || isJsonResponseFromTool ? {
       bedrock: {
         ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
         ...((_h = response.usage) == null ? void 0 : _h.cacheWriteInputTokens) != null && {
@@ -896,20 +1014,16 @@ var BedrockChatLanguageModel = class {
             cacheWriteInputTokens: response.usage.cacheWriteInputTokens
           }
         },
-        ...usesJsonResponseTool && { isJsonResponseFromTool: true }
+        ...isJsonResponseFromTool && { isJsonResponseFromTool: true }
       }
     } : void 0;
     return {
       content,
       finishReason: mapBedrockFinishReason(
-        response.stopReason
+        response.stopReason,
+        isJsonResponseFromTool
       ),
-      usage: {
-        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
-      },
+      usage: convertBedrockUsage(response.usage),
       response: {
         // TODO add id, timestamp, etc
         headers: responseHeaders
@@ -922,13 +1036,12 @@ var BedrockChatLanguageModel = class {
     const {
       command: args,
       warnings,
-      usesJsonResponseTool,
-      betas
+      usesJsonResponseTool
     } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse-stream`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
-      headers: await this.getHeaders({ betas, headers: options.headers }),
+      headers: await this.getHeaders({ headers: options.headers }),
       body: args,
       failedResponseHandler: createJsonErrorResponseHandler({
         errorSchema: BedrockErrorSchema,
@@ -939,12 +1052,9 @@ var BedrockChatLanguageModel = class {
       fetch: this.config.fetch
     });
     let finishReason = "unknown";
-    const usage = {
-      inputTokens: void 0,
-      outputTokens: void 0,
-      totalTokens: void 0
-    };
+    let usage = void 0;
     let providerMetadata = void 0;
+    let isJsonResponseFromTool = false;
     const contentBlocks = {};
     return {
       stream: response.pipeThrough(
@@ -953,7 +1063,7 @@ var BedrockChatLanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i;
             function enqueueError(bedrockError) {
               finishReason = "error";
               controller.enqueue({ type: "error", error: bedrockError });
@@ -984,15 +1094,15 @@ var BedrockChatLanguageModel = class {
             }
             if (value.messageStop) {
               finishReason = mapBedrockFinishReason(
-                value.messageStop.stopReason
+                value.messageStop.stopReason,
+                isJsonResponseFromTool
               );
             }
             if (value.metadata) {
-              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 ? {
+              if (value.metadata.usage) {
+                usage = value.metadata.usage;
+              }
+              const cacheUsage = ((_a = value.metadata.usage) == null ? void 0 : _a.cacheWriteInputTokens) != null ? {
                 usage: {
                   cacheWriteInputTokens: value.metadata.usage.cacheWriteInputTokens
                 }
@@ -1000,19 +1110,16 @@ var BedrockChatLanguageModel = class {
               const trace = value.metadata.trace ? {
                 trace: value.metadata.trace
               } : void 0;
-              if (cacheUsage || trace || usesJsonResponseTool) {
+              if (cacheUsage || trace) {
                 providerMetadata = {
                   bedrock: {
                     ...cacheUsage,
-                    ...trace,
-                    ...usesJsonResponseTool && {
-                      isJsonResponseFromTool: true
-                    }
+                    ...trace
                   }
                 };
               }
             }
-            if (((_j = value.contentBlockStart) == null ? void 0 : _j.contentBlockIndex) != null && !((_l = (_k = value.contentBlockStart) == null ? void 0 : _k.start) == null ? void 0 : _l.toolUse)) {
+            if (((_b = value.contentBlockStart) == null ? void 0 : _b.contentBlockIndex) != null && !((_d = (_c = value.contentBlockStart) == null ? void 0 : _c.start) == null ? void 0 : _d.toolUse)) {
               const blockIndex = value.contentBlockStart.contentBlockIndex;
               contentBlocks[blockIndex] = { type: "text" };
               controller.enqueue({
@@ -1020,26 +1127,22 @@ var BedrockChatLanguageModel = class {
                 id: String(blockIndex)
               });
             }
-            if (((_m = value.contentBlockDelta) == null ? void 0 : _m.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
+            if (((_e = value.contentBlockDelta) == null ? void 0 : _e.delta) && "text" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.text) {
               const blockIndex = value.contentBlockDelta.contentBlockIndex || 0;
               if (contentBlocks[blockIndex] == null) {
                 contentBlocks[blockIndex] = { type: "text" };
-                if (!usesJsonResponseTool) {
-                  controller.enqueue({
-                    type: "text-start",
-                    id: String(blockIndex)
-                  });
-                }
-              }
-              if (!usesJsonResponseTool) {
                 controller.enqueue({
-                  type: "text-delta",
-                  id: String(blockIndex),
-                  delta: value.contentBlockDelta.delta.text
+                  type: "text-start",
+                  id: String(blockIndex)
                 });
               }
+              controller.enqueue({
+                type: "text-delta",
+                id: String(blockIndex),
+                delta: value.contentBlockDelta.delta.text
+              });
             }
-            if (((_n = value.contentBlockStop) == null ? void 0 : _n.contentBlockIndex) != null) {
+            if (((_f = value.contentBlockStop) == null ? void 0 : _f.contentBlockIndex) != null) {
               const blockIndex = value.contentBlockStop.contentBlockIndex;
               const contentBlock = contentBlocks[blockIndex];
               if (contentBlock != null) {
@@ -1049,14 +1152,13 @@ var BedrockChatLanguageModel = class {
                     id: String(blockIndex)
                   });
                 } else if (contentBlock.type === "text") {
-                  if (!usesJsonResponseTool) {
-                    controller.enqueue({
-                      type: "text-end",
-                      id: String(blockIndex)
-                    });
-                  }
+                  controller.enqueue({
+                    type: "text-end",
+                    id: String(blockIndex)
+                  });
                 } else if (contentBlock.type === "tool-call") {
-                  if (usesJsonResponseTool) {
+                  if (contentBlock.isJsonResponseTool) {
+                    isJsonResponseFromTool = true;
                     controller.enqueue({
                       type: "text-start",
                       id: String(blockIndex)
@@ -1079,14 +1181,14 @@ var BedrockChatLanguageModel = class {
                       type: "tool-call",
                       toolCallId: contentBlock.toolCallId,
                       toolName: contentBlock.toolName,
-                      input: contentBlock.jsonText
+                      input: contentBlock.jsonText === "" ? "{}" : contentBlock.jsonText
                     });
                   }
                 }
                 delete contentBlocks[blockIndex];
               }
             }
-            if (((_o = value.contentBlockDelta) == null ? void 0 : _o.delta) && "reasoningContent" in value.contentBlockDelta.delta && value.contentBlockDelta.delta.reasoningContent) {
+            if (((_g = value.contentBlockDelta) == null ? void 0 : _g.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) {
@@ -1127,16 +1229,18 @@ var BedrockChatLanguageModel = class {
               }
             }
             const contentBlockStart = value.contentBlockStart;
-            if (((_p = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _p.toolUse) != null) {
+            if (((_h = contentBlockStart == null ? void 0 : contentBlockStart.start) == null ? void 0 : _h.toolUse) != null) {
               const toolUse = contentBlockStart.start.toolUse;
               const blockIndex = contentBlockStart.contentBlockIndex;
+              const isJsonResponseTool = usesJsonResponseTool && toolUse.name === "json";
               contentBlocks[blockIndex] = {
                 type: "tool-call",
                 toolCallId: toolUse.toolUseId,
                 toolName: toolUse.name,
-                jsonText: ""
+                jsonText: "",
+                isJsonResponseTool
               };
-              if (!usesJsonResponseTool) {
+              if (!isJsonResponseTool) {
                 controller.enqueue({
                   type: "tool-input-start",
                   id: toolUse.toolUseId,
@@ -1149,8 +1253,8 @@ var BedrockChatLanguageModel = class {
               const blockIndex = contentBlockDelta.contentBlockIndex;
               const contentBlock = contentBlocks[blockIndex];
               if ((contentBlock == null ? void 0 : contentBlock.type) === "tool-call") {
-                const delta = (_q = contentBlockDelta.delta.toolUse.input) != null ? _q : "";
-                if (!usesJsonResponseTool) {
+                const delta = (_i = contentBlockDelta.delta.toolUse.input) != null ? _i : "";
+                if (!contentBlock.isJsonResponseTool) {
                   controller.enqueue({
                     type: "tool-input-delta",
                     id: contentBlock.toolCallId,
@@ -1162,10 +1266,24 @@ var BedrockChatLanguageModel = class {
             }
           },
           flush(controller) {
+            if (isJsonResponseFromTool) {
+              if (providerMetadata) {
+                providerMetadata.bedrock = {
+                  ...providerMetadata.bedrock,
+                  isJsonResponseFromTool: true
+                };
+              } else {
+                providerMetadata = {
+                  bedrock: {
+                    isJsonResponseFromTool: true
+                  }
+                };
+              }
+            }
             controller.enqueue({
               type: "finish",
               finishReason,
-              usage,
+              usage: convertBedrockUsage(usage),
               ...providerMetadata && { providerMetadata }
             });
           }
@@ -1366,6 +1484,7 @@ var BedrockEmbeddingModel = class {
       abortSignal
     });
     return {
+      warnings: [],
       embeddings: [response.embedding],
       usage: { tokens: response.inputTextTokenCount }
     };
@@ -1442,8 +1561,8 @@ var BedrockImageModel = class {
     };
     if (aspectRatio != void 0) {
       warnings.push({
-        type: "unsupported-setting",
-        setting: "aspectRatio",
+        type: "unsupported",
+        feature: "aspectRatio",
         details: "This model does not support aspect ratio. Use `size` instead."
       });
     }
@@ -1479,42 +1598,26 @@ var bedrockImageResponseSchema = z6.object({
   images: z6.array(z6.string())
 });
 
-// src/headers-utils.ts
-function extractHeaders(headers) {
-  let originalHeaders = {};
-  if (headers) {
-    if (headers instanceof Headers) {
-      originalHeaders = convertHeadersToRecord(headers);
-    } else if (Array.isArray(headers)) {
-      for (const [k, v] of headers) {
-        originalHeaders[k.toLowerCase()] = v;
-      }
-    } else {
-      originalHeaders = Object.fromEntries(
-        Object.entries(headers).map(([k, v]) => [k.toLowerCase(), v])
-      );
-    }
-  }
-  return originalHeaders;
-}
-function convertHeadersToRecord(headers) {
-  return Object.fromEntries([...headers]);
-}
-
 // src/bedrock-sigv4-fetch.ts
 import {
   combineHeaders as combineHeaders4,
+  normalizeHeaders,
   withUserAgentSuffix,
   getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 import { AwsV4Signer } from "aws4fetch";
+
+// src/version.ts
+var VERSION = true ? "4.0.0-beta.100" : "0.0.0-test";
+
+// src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
   return async (input, init) => {
     var _a, _b;
     const request = input instanceof Request ? input : void 0;
     const originalHeaders = combineHeaders4(
-      extractHeaders(request == null ? void 0 : request.headers),
-      extractHeaders(init == null ? void 0 : init.headers)
+      normalizeHeaders(request == null ? void 0 : request.headers),
+      normalizeHeaders(init == null ? void 0 : init.headers)
     );
     const headersWithUserAgent = withUserAgentSuffix(
       originalHeaders,
@@ -1550,7 +1653,7 @@ function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
       service: "bedrock"
     });
     const signingResult = await signer.sign();
-    const signedHeaders = convertHeadersToRecord(signingResult.headers);
+    const signedHeaders = normalizeHeaders(signingResult.headers);
     const combinedHeaders = combineHeaders4(headersWithUserAgent, signedHeaders);
     return fetch(input, {
       ...init,
@@ -1572,7 +1675,7 @@ function prepareBodyString(body) {
 }
 function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
   return async (input, init) => {
-    const originalHeaders = extractHeaders(init == null ? void 0 : init.headers);
+    const originalHeaders = normalizeHeaders(init == null ? void 0 : init.headers);
     const headersWithUserAgent = withUserAgentSuffix(
       originalHeaders,
       `ai-sdk/amazon-bedrock/${VERSION}`,
@@ -1588,6 +1691,130 @@ function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
   };
 }
 
+// src/reranking/bedrock-reranking-model.ts
+import {
+  combineHeaders as combineHeaders5,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler4,
+  createJsonResponseHandler as createJsonResponseHandler4,
+  parseProviderOptions as parseProviderOptions4,
+  postJsonToApi as postJsonToApi4,
+  resolve as resolve4
+} from "@ai-sdk/provider-utils";
+
+// src/reranking/bedrock-reranking-api.ts
+import { lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z as z7 } from "zod/v4";
+var bedrockRerankingResponseSchema = lazySchema(
+  () => zodSchema(
+    z7.object({
+      results: z7.array(
+        z7.object({
+          index: z7.number(),
+          relevanceScore: z7.number()
+        })
+      ),
+      nextToken: z7.string().optional()
+    })
+  )
+);
+
+// src/reranking/bedrock-reranking-options.ts
+import { lazySchema as lazySchema2, zodSchema as zodSchema2 } from "@ai-sdk/provider-utils";
+import { z as z8 } from "zod/v4";
+var bedrockRerankingOptionsSchema = lazySchema2(
+  () => zodSchema2(
+    z8.object({
+      /**
+       * If the total number of results was greater than could fit in a response, a token is returned in the nextToken field. You can enter that token in this field to return the next batch of results.
+       */
+      nextToken: z8.string().optional(),
+      /**
+       * Additional model request fields to pass to the model.
+       */
+      additionalModelRequestFields: z8.record(z8.string(), z8.any()).optional()
+    })
+  )
+);
+
+// src/reranking/bedrock-reranking-model.ts
+var BedrockRerankingModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+    this.provider = "amazon-bedrock";
+  }
+  async doRerank({
+    documents,
+    headers,
+    query,
+    topN,
+    abortSignal,
+    providerOptions
+  }) {
+    const bedrockOptions = await parseProviderOptions4({
+      provider: "bedrock",
+      providerOptions,
+      schema: bedrockRerankingOptionsSchema
+    });
+    const {
+      value: response,
+      responseHeaders,
+      rawValue
+    } = await postJsonToApi4({
+      url: `${this.config.baseUrl()}/rerank`,
+      headers: await resolve4(
+        combineHeaders5(await resolve4(this.config.headers), headers)
+      ),
+      body: {
+        nextToken: bedrockOptions == null ? void 0 : bedrockOptions.nextToken,
+        queries: [
+          {
+            textQuery: { text: query },
+            type: "TEXT"
+          }
+        ],
+        rerankingConfiguration: {
+          bedrockRerankingConfiguration: {
+            modelConfiguration: {
+              modelArn: `arn:aws:bedrock:${this.config.region}::foundation-model/${this.modelId}`,
+              additionalModelRequestFields: bedrockOptions == null ? void 0 : bedrockOptions.additionalModelRequestFields
+            },
+            numberOfResults: topN
+          },
+          type: "BEDROCK_RERANKING_MODEL"
+        },
+        sources: documents.values.map((value) => ({
+          type: "INLINE",
+          inlineDocumentSource: documents.type === "text" ? {
+            type: "TEXT",
+            textDocument: { text: value }
+          } : {
+            type: "JSON",
+            jsonDocument: value
+          }
+        }))
+      },
+      failedResponseHandler: createJsonErrorResponseHandler4({
+        errorSchema: BedrockErrorSchema,
+        errorToMessage: (error) => `${error.type}: ${error.message}`
+      }),
+      successfulResponseHandler: createJsonResponseHandler4(
+        bedrockRerankingResponseSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal
+    });
+    return {
+      ranking: response.results,
+      response: {
+        headers: responseHeaders,
+        body: rawValue
+      }
+    };
+  }
+};
+
 // src/bedrock-provider.ts
 function createAmazonBedrock(options = {}) {
   const rawApiKey = loadOptionalSetting({
@@ -1656,7 +1883,12 @@ Original error: ${errorMessage}`
       throw error;
     }
   }, options.fetch);
-  const getBaseUrl = () => {
+  const getHeaders = () => {
+    var _a;
+    const baseHeaders = (_a = options.headers) != null ? _a : {};
+    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
+  };
+  const getBedrockRuntimeBaseUrl = () => {
     var _a, _b;
     return (_b = withoutTrailingSlash(
       (_a = options.baseURL) != null ? _a : `https://bedrock-runtime.${loadSetting({
@@ -1667,13 +1899,19 @@ Original error: ${errorMessage}`
       })}.amazonaws.com`
     )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
   };
-  const getHeaders = () => {
-    var _a;
-    const baseHeaders = (_a = options.headers) != null ? _a : {};
-    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
+  const getBedrockAgentRuntimeBaseUrl = () => {
+    var _a, _b;
+    return (_b = withoutTrailingSlash(
+      (_a = options.baseURL) != null ? _a : `https://bedrock-agent-runtime.${loadSetting({
+        settingValue: options.region,
+        settingName: "region",
+        environmentVariableName: "AWS_REGION",
+        description: "AWS region"
+      })}.amazonaws.com`
+    )) != null ? _b : `https://bedrock-agent-runtime.us-west-2.amazonaws.com`;
   };
   const createChatModel = (modelId) => new BedrockChatLanguageModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,
     fetch: fetchFunction,
     generateId
@@ -1687,21 +1925,36 @@ Original error: ${errorMessage}`
     return createChatModel(modelId);
   };
   const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,
     fetch: fetchFunction
   });
   const createImageModel = (modelId) => new BedrockImageModel(modelId, {
-    baseUrl: getBaseUrl,
+    baseUrl: getBedrockRuntimeBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
+  const createRerankingModel = (modelId) => new BedrockRerankingModel(modelId, {
+    baseUrl: getBedrockAgentRuntimeBaseUrl,
+    region: loadSetting({
+      settingValue: options.region,
+      settingName: "region",
+      environmentVariableName: "AWS_REGION",
+      description: "AWS region"
+    }),
     headers: getHeaders,
     fetch: fetchFunction
   });
+  provider.specificationVersion = "v3";
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
+  provider.embeddingModel = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;
   provider.textEmbeddingModel = createEmbeddingModel;
   provider.image = createImageModel;
   provider.imageModel = createImageModel;
+  provider.reranking = createRerankingModel;
+  provider.rerankingModel = createRerankingModel;
   provider.tools = anthropicTools2;
   return provider;
 }