@ai-sdk/amazon-bedrock 3.0.130 → 3.0.132

package/dist/index.mjs

@@ -1,14 +1,13 @@
 // src/bedrock-provider.ts
 import {
   generateId,
-  loadOptionalSetting,
+  loadOptionalSetting as loadOptionalSetting2,
   loadSetting,
-  withoutTrailingSlash,
   withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.130" : "0.0.0-test";
+var VERSION = true ? "3.0.132" : "0.0.0-test";
 
 // src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
@@ -17,7 +16,7 @@ import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 import { sanitizeJsonSchema } from "@ai-sdk/anthropic/internal";
 import {
   combineHeaders,
-  createJsonErrorResponseHandler,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler,
   parseProviderOptions as parseProviderOptions2,
   postJsonToApi,
@@ -114,12 +113,26 @@ var bedrockProviderOptions = z.object({
   serviceTier: z.enum(["reserved", "priority", "default", "flex"]).optional()
 });
 
+// src/bedrock-anthropic-model-support.ts
+function isAnthropicModel({
+  modelId,
+  modelFamily,
+  reasoningBudgetTokens
+}) {
+  return modelFamily === "anthropic" || modelId.includes("anthropic") || modelId.includes(":application-inference-profile/") && reasoningBudgetTokens != null;
+}
+
 // src/bedrock-error.ts
+import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
 import { z as z2 } from "zod/v4";
 var BedrockErrorSchema = z2.object({
   message: z2.string(),
   type: z2.string().nullish()
 });
+var bedrockFailedResponseHandler = createJsonErrorResponseHandler({
+  errorSchema: BedrockErrorSchema,
+  errorToMessage: (error) => error.type == null ? error.message : `${error.type}: ${error.message}`
+});
 
 // src/bedrock-event-stream-response-handler.ts
 import { EmptyResponseBodyError } from "@ai-sdk/provider";
@@ -232,6 +245,8 @@ async function prepareTools({
   tools,
   toolChoice,
   modelId,
+  modelFamily,
+  reasoningBudgetTokens,
   disableParallelToolUse
 }) {
   var _a;
@@ -264,14 +279,18 @@ async function prepareTools({
       toolWarnings
     };
   }
-  const isAnthropicModel = modelId.includes("anthropic.");
   const providerDefinedTools = supportedTools.filter(
     (t) => t.type === "provider-defined"
   );
+  const isAnthropicModel2 = isAnthropicModel({
+    modelId,
+    modelFamily,
+    reasoningBudgetTokens
+  });
   const functionTools = supportedTools.filter((t) => t.type === "function");
   let additionalTools = void 0;
   const bedrockTools = [];
-  const usingAnthropicTools = isAnthropicModel && providerDefinedTools.length > 0;
+  const usingAnthropicTools = isAnthropicModel2 && providerDefinedTools.length > 0;
   if (usingAnthropicTools) {
     if (functionTools.length > 0) {
       toolWarnings.push({
@@ -331,7 +350,7 @@ async function prepareTools({
       }
     });
   }
-  if (isAnthropicModel && !usingAnthropicTools && disableParallelToolUse && bedrockTools.length > 0 && (toolChoice == null ? void 0 : toolChoice.type) !== "none") {
+  if (isAnthropicModel2 && !usingAnthropicTools && disableParallelToolUse && bedrockTools.length > 0 && (toolChoice == null ? void 0 : toolChoice.type) !== "none") {
     additionalTools = {
       tool_choice: (toolChoice == null ? void 0 : toolChoice.type) === "required" ? { type: "any", disable_parallel_tool_use: true } : (toolChoice == null ? void 0 : toolChoice.type) === "tool" ? {
         type: "tool",
@@ -793,7 +812,7 @@ var BedrockChatLanguageModel = class {
     toolChoice,
     providerOptions
   }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u;
     const bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
@@ -845,12 +864,16 @@ var BedrockChatLanguageModel = class {
         details: "Only text and json response formats are supported."
       });
     }
-    const isAnthropicModel = this.modelId.includes("anthropic");
-    const openAIModelId = (_b = /^(?:[^.]+\.)?(openai\..+)$/.exec(this.modelId)) == null ? void 0 : _b[1];
+    const isAnthropicModel2 = isAnthropicModel({
+      modelId: this.modelId,
+      modelFamily: this.config.modelFamily,
+      reasoningBudgetTokens: (_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.budgetTokens
+    });
+    const openAIModelId = (_c = /^(?:[^.]+\.)?(openai\..+)$/.exec(this.modelId)) == null ? void 0 : _c[1];
     const isOpenAIModel = openAIModelId != null;
-    const isOpenAIGptOssModel = (_c = openAIModelId == null ? void 0 : openAIModelId.startsWith("openai.gpt-oss-")) != null ? _c : false;
-    const isThinkingRequested = ((_d = bedrockOptions.reasoningConfig) == null ? void 0 : _d.type) === "enabled" || ((_e = bedrockOptions.reasoningConfig) == null ? void 0 : _e.type) === "adaptive";
-    const structuredOutputMode = (_g = (_f = bedrockOptions.structuredOutputMode) != null ? _f : anthropicOptions == null ? void 0 : anthropicOptions.structuredOutputMode) != null ? _g : "auto";
+    const isOpenAIGptOssModel = (_d = openAIModelId == null ? void 0 : openAIModelId.startsWith("openai.gpt-oss-")) != null ? _d : false;
+    const isThinkingRequested = ((_e = bedrockOptions.reasoningConfig) == null ? void 0 : _e.type) === "enabled" || ((_f = bedrockOptions.reasoningConfig) == null ? void 0 : _f.type) === "adaptive";
+    const structuredOutputMode = (_h = (_g = bedrockOptions.structuredOutputMode) != null ? _g : anthropicOptions == null ? void 0 : anthropicOptions.structuredOutputMode) != null ? _h : "auto";
     if (structuredOutputMode === "jsonTool") {
       const additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields
@@ -867,7 +890,8 @@ var BedrockChatLanguageModel = class {
         bedrockOptions.additionalModelRequestFields = additionalModelRequestFields;
       }
     }
-    const useNativeStructuredOutput = isAnthropicModel && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && (structuredOutputMode === "outputFormat" || structuredOutputMode === "auto" && supportsNativeStructuredOutput(this.modelId) && isThinkingRequested);
+    const modelSupportsNativeStructuredOutput = this.config.modelFamily === "anthropic" || supportsNativeStructuredOutput(this.modelId) && isThinkingRequested;
+    const useNativeStructuredOutput = isAnthropicModel2 && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && (structuredOutputMode === "outputFormat" || structuredOutputMode === "auto" && modelSupportsNativeStructuredOutput);
     const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && !useNativeStructuredOutput ? {
       type: "function",
       name: "json",
@@ -878,6 +902,8 @@ var BedrockChatLanguageModel = class {
       tools: jsonResponseTool ? [...tools != null ? tools : [], jsonResponseTool] : tools,
       toolChoice: jsonResponseTool != null ? { type: "required" } : toolChoice,
       modelId: this.modelId,
+      modelFamily: this.config.modelFamily,
+      reasoningBudgetTokens: (_i = bedrockOptions.reasoningConfig) == null ? void 0 : _i.budgetTokens,
       disableParallelToolUse: anthropicOptions == null ? void 0 : anthropicOptions.disableParallelToolUse
     });
     warnings.push(...toolWarnings);
@@ -888,17 +914,17 @@ var BedrockChatLanguageModel = class {
       };
     }
     if (betas.size > 0 || bedrockOptions.anthropicBeta) {
-      const existingBetas = (_h = bedrockOptions.anthropicBeta) != null ? _h : [];
+      const existingBetas = (_j = bedrockOptions.anthropicBeta) != null ? _j : [];
       const mergedBetas = betas.size > 0 ? [...existingBetas, ...Array.from(betas)] : existingBetas;
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         anthropic_beta: mergedBetas
       };
     }
-    const thinkingType = (_i = bedrockOptions.reasoningConfig) == null ? void 0 : _i.type;
-    const thinkingBudget = thinkingType === "enabled" ? (_j = bedrockOptions.reasoningConfig) == null ? void 0 : _j.budgetTokens : void 0;
-    const thinkingDisplay = thinkingType === "adaptive" ? (_k = bedrockOptions.reasoningConfig) == null ? void 0 : _k.display : void 0;
-    const isAnthropicThinkingEnabled = isAnthropicModel && isThinkingRequested;
+    const thinkingType = (_k = bedrockOptions.reasoningConfig) == null ? void 0 : _k.type;
+    const thinkingBudget = thinkingType === "enabled" ? (_l = bedrockOptions.reasoningConfig) == null ? void 0 : _l.budgetTokens : void 0;
+    const thinkingDisplay = thinkingType === "adaptive" ? (_m = bedrockOptions.reasoningConfig) == null ? void 0 : _m.display : void 0;
+    const isAnthropicThinkingEnabled = isAnthropicModel2 && isThinkingRequested;
     const inferenceConfig = {
       ...maxOutputTokens != null && { maxTokens: maxOutputTokens },
       ...temperature != null && { temperature },
@@ -929,8 +955,8 @@ var BedrockChatLanguageModel = class {
           }
         };
       }
-    } else if (!isAnthropicModel) {
-      if (((_l = bedrockOptions.reasoningConfig) == null ? void 0 : _l.budgetTokens) != null) {
+    } else if (!isAnthropicModel2) {
+      if (((_n = bedrockOptions.reasoningConfig) == null ? void 0 : _n.budgetTokens) != null) {
         warnings.push({
           type: "unsupported-setting",
           setting: "budgetTokens",
@@ -945,13 +971,13 @@ var BedrockChatLanguageModel = class {
         });
       }
     }
-    const maxReasoningEffort = (_m = bedrockOptions.reasoningConfig) == null ? void 0 : _m.maxReasoningEffort;
+    const maxReasoningEffort = (_o = bedrockOptions.reasoningConfig) == null ? void 0 : _o.maxReasoningEffort;
     if (maxReasoningEffort != null) {
-      if (isAnthropicModel) {
+      if (isAnthropicModel2) {
         bedrockOptions.additionalModelRequestFields = {
           ...bedrockOptions.additionalModelRequestFields,
           output_config: {
-            ...(_n = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _n.output_config,
+            ...(_p = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _p.output_config,
             effort: maxReasoningEffort
           }
         };
@@ -962,7 +988,7 @@ var BedrockChatLanguageModel = class {
         } : {
           ...bedrockOptions.additionalModelRequestFields,
           reasoning: {
-            ...(_o = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _o.reasoning,
+            ...(_q = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _q.reasoning,
             effort: maxReasoningEffort
           }
         };
@@ -970,7 +996,7 @@ var BedrockChatLanguageModel = class {
         bedrockOptions.additionalModelRequestFields = {
           ...bedrockOptions.additionalModelRequestFields,
           reasoningConfig: {
-            ...((_p = bedrockOptions.reasoningConfig) == null ? void 0 : _p.type) != null && {
+            ...((_r = bedrockOptions.reasoningConfig) == null ? void 0 : _r.type) != null && {
               type: bedrockOptions.reasoningConfig.type
             },
             maxReasoningEffort
@@ -982,7 +1008,7 @@ var BedrockChatLanguageModel = class {
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         output_config: {
-          ...(_q = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _q.output_config,
+          ...(_s = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _s.output_config,
           format: {
             type: "json_schema",
             schema: sanitizeJsonSchema(responseFormat.schema)
@@ -1014,7 +1040,7 @@ var BedrockChatLanguageModel = class {
         details: "topK is not supported when thinking is enabled"
       });
     }
-    const hasAnyTools = ((_s = (_r = toolConfig.tools) == null ? void 0 : _r.length) != null ? _s : 0) > 0 || additionalTools;
+    const hasAnyTools = ((_u = (_t = toolConfig.tools) == null ? void 0 : _t.length) != null ? _u : 0) > 0 || additionalTools;
     let filteredPrompt = prompt;
     if (!hasAnyTools) {
       const hasToolContent = prompt.some(
@@ -1048,7 +1074,7 @@ var BedrockChatLanguageModel = class {
       structuredOutputMode: ____,
       ...filteredBedrockOptions
     } = (providerOptions == null ? void 0 : providerOptions.bedrock) || {};
-    const additionalModelResponseFieldPaths = isAnthropicModel ? ["/delta/stop_sequence"] : void 0;
+    const additionalModelResponseFieldPaths = isAnthropicModel2 ? ["/delta/stop_sequence"] : void 0;
     return {
       command: {
         system,
@@ -1090,7 +1116,7 @@ var BedrockChatLanguageModel = class {
       url,
       headers: await this.getHeaders({ headers: options.headers }),
       body: args,
-      failedResponseHandler: createJsonErrorResponseHandler({
+      failedResponseHandler: createJsonErrorResponseHandler2({
         errorSchema: BedrockErrorSchema,
         errorToMessage: (error) => {
           var _a2;
@@ -1213,10 +1239,7 @@ var BedrockChatLanguageModel = class {
       url,
       headers: await this.getHeaders({ headers: options.headers }),
       body: args,
-      failedResponseHandler: createJsonErrorResponseHandler({
-        errorSchema: BedrockErrorSchema,
-        errorToMessage: (error) => `${error.type}: ${error.message}`
-      }),
+      failedResponseHandler: bedrockFailedResponseHandler,
       successfulResponseHandler: createBedrockEventStreamResponseHandler(BedrockStreamSchema),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
@@ -1663,7 +1686,6 @@ import {
 } from "@ai-sdk/provider";
 import {
   combineHeaders as combineHeaders2,
-  createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler as createJsonResponseHandler2,
   parseProviderOptions as parseProviderOptions3,
   postJsonToApi as postJsonToApi2,
@@ -1732,10 +1754,7 @@ var BedrockEmbeddingModel = class {
         combineHeaders2(await resolve2(this.config.headers), headers)
       ),
       body: args,
-      failedResponseHandler: createJsonErrorResponseHandler2({
-        errorSchema: BedrockErrorSchema,
-        errorToMessage: (error) => `${error.type}: ${error.message}`
-      }),
+      failedResponseHandler: bedrockFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler2(
         BedrockEmbeddingResponseSchema
       ),
@@ -1756,7 +1775,6 @@ var BedrockEmbeddingResponseSchema = z6.object({
 // src/bedrock-image-model.ts
 import {
   combineHeaders as combineHeaders3,
-  createJsonErrorResponseHandler as createJsonErrorResponseHandler3,
   createJsonResponseHandler as createJsonResponseHandler3,
   postJsonToApi as postJsonToApi3,
   resolve as resolve3
@@ -1831,10 +1849,7 @@ var BedrockImageModel = class {
         combineHeaders3(await resolve3(this.config.headers), headers)
       ),
       body: args,
-      failedResponseHandler: createJsonErrorResponseHandler3({
-        errorSchema: BedrockErrorSchema,
-        errorToMessage: (error) => `${error.type}: ${error.message}`
-      }),
+      failedResponseHandler: bedrockFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler3(
         bedrockImageResponseSchema
       ),
@@ -1944,9 +1959,47 @@ function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
   };
 }
 
+// src/resolve-bedrock-base-url.ts
+import {
+  loadOptionalSetting,
+  withoutTrailingSlash
+} from "@ai-sdk/provider-utils";
+var AWS_PARTITION_DNS_SUFFIXES = [
+  { regionPrefix: "cn-", dnsSuffix: "amazonaws.com.cn" },
+  { regionPrefix: "us-iso-", dnsSuffix: "c2s.ic.gov" },
+  { regionPrefix: "us-isob-", dnsSuffix: "sc2s.sgov.gov" },
+  { regionPrefix: "eu-isoe-", dnsSuffix: "cloud.adc-e.uk" },
+  { regionPrefix: "us-isof-", dnsSuffix: "csp.hci.ic.gov" },
+  { regionPrefix: "eusc-", dnsSuffix: "amazonaws.eu" }
+];
+function resolveBedrockBaseURL({
+  baseURL,
+  getRegion,
+  service,
+  serviceEndpointUrlEnvironmentVariableName
+}) {
+  var _a, _b, _c, _d, _e;
+  const resolvedBaseURL = (_a = baseURL != null ? baseURL : loadOptionalSetting({
+    settingValue: void 0,
+    environmentVariableName: serviceEndpointUrlEnvironmentVariableName
+  })) != null ? _a : loadOptionalSetting({
+    settingValue: void 0,
+    environmentVariableName: "AWS_ENDPOINT_URL"
+  });
+  if (resolvedBaseURL != null) {
+    return (_b = withoutTrailingSlash(resolvedBaseURL)) != null ? _b : resolvedBaseURL;
+  }
+  const region = getRegion();
+  const dnsSuffix = (_d = (_c = AWS_PARTITION_DNS_SUFFIXES.find(
+    ({ regionPrefix }) => region.startsWith(regionPrefix)
+  )) == null ? void 0 : _c.dnsSuffix) != null ? _d : "amazonaws.com";
+  const generatedBaseURL = `https://${service}.${region}.${dnsSuffix}`;
+  return (_e = withoutTrailingSlash(generatedBaseURL)) != null ? _e : generatedBaseURL;
+}
+
 // src/bedrock-provider.ts
 function createAmazonBedrock(options = {}) {
-  const rawApiKey = loadOptionalSetting({
+  const rawApiKey = loadOptionalSetting2({
     settingValue: options.apiKey,
     environmentVariableName: "AWS_BEARER_TOKEN_BEDROCK"
   });
@@ -1986,7 +2039,7 @@ function createAmazonBedrock(options = {}) {
           environmentVariableName: "AWS_SECRET_ACCESS_KEY",
           description: "AWS secret access key"
         }),
-        sessionToken: loadOptionalSetting({
+        sessionToken: loadOptionalSetting2({
           settingValue: options.sessionToken,
           environmentVariableName: "AWS_SESSION_TOKEN"
         })
@@ -2012,35 +2065,36 @@ Original error: ${errorMessage}`
       throw error;
     }
   }, options.fetch);
-  const getBaseUrl = () => {
-    var _a, _b;
-    return (_b = withoutTrailingSlash(
-      (_a = options.baseURL) != null ? _a : `https://bedrock-runtime.${loadSetting({
-        settingValue: options.region,
-        settingName: "region",
-        environmentVariableName: "AWS_REGION",
-        description: "AWS region"
-      })}.amazonaws.com`
-    )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
-  };
+  const getBaseUrl = () => resolveBedrockBaseURL({
+    baseURL: options.baseURL,
+    getRegion: () => loadSetting({
+      settingValue: options.region,
+      settingName: "region",
+      environmentVariableName: "AWS_REGION",
+      description: "AWS region"
+    }),
+    service: "bedrock-runtime",
+    serviceEndpointUrlEnvironmentVariableName: "AWS_ENDPOINT_URL_BEDROCK_RUNTIME"
+  });
   const getHeaders = () => {
     var _a;
     const baseHeaders = (_a = options.headers) != null ? _a : {};
     return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
   };
-  const createChatModel = (modelId) => new BedrockChatLanguageModel(modelId, {
+  const createChatModel = (modelId, settings = {}) => new BedrockChatLanguageModel(modelId, {
     baseUrl: getBaseUrl,
     headers: getHeaders,
     fetch: fetchFunction,
-    generateId
+    generateId,
+    modelFamily: settings.modelFamily
   });
-  const provider = function(modelId) {
+  const provider = function(modelId, settings) {
     if (new.target) {
       throw new Error(
         "The Amazon Bedrock model function cannot be called with the new keyword."
       );
     }
-    return createChatModel(modelId);
+    return createChatModel(modelId, settings);
   };
   const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
     baseUrl: getBaseUrl,