@ai-sdk/amazon-bedrock 4.0.175 → 4.0.177

package/dist/index.mjs

@@ -2,9 +2,8 @@
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 import {
   generateId,
-  loadOptionalSetting,
+  loadOptionalSetting as loadOptionalSetting2,
   loadSetting,
-  withoutTrailingSlash,
   withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
 
@@ -111,6 +110,13 @@ var amazonBedrockLanguageModelOptions = z.object({
 });
 
 // src/bedrock-anthropic-model-support.ts
+function isAnthropicModel({
+  modelId,
+  modelFamily,
+  reasoningBudgetTokens
+}) {
+  return modelFamily === "anthropic" || modelId.includes("anthropic") || modelId.includes(":application-inference-profile/") && reasoningBudgetTokens != null;
+}
 function supportsStrictTools(modelId) {
   return !matchesModel(modelId, MODELS_WITHOUT_STRICT_TOOL_SUPPORT);
 }
@@ -254,6 +260,8 @@ async function prepareTools({
   tools,
   toolChoice,
   modelId,
+  modelFamily,
+  reasoningBudgetTokens,
   disableParallelToolUse
 }) {
   var _a;
@@ -286,12 +294,16 @@ async function prepareTools({
       toolWarnings
     };
   }
-  const isAnthropicModel = modelId.includes("anthropic.");
+  const isAnthropicModel2 = isAnthropicModel({
+    modelId,
+    modelFamily,
+    reasoningBudgetTokens
+  });
   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 && ProviderTools.length > 0;
+  const usingAnthropicTools = isAnthropicModel2 && ProviderTools.length > 0;
   if (usingAnthropicTools) {
     const {
       toolChoice: preparedAnthropicToolChoice,
@@ -356,7 +368,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",
@@ -963,7 +975,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;
     const bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
@@ -1015,7 +1027,11 @@ var BedrockChatLanguageModel = class {
         details: "Only text and json response formats are supported."
       });
     }
-    const isAnthropicModel = this.modelId.includes("anthropic") || this.modelId.includes(":application-inference-profile/") && ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.budgetTokens) != null;
+    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 = (_d = openAIModelId == null ? void 0 : openAIModelId.startsWith("openai.gpt-oss-")) != null ? _d : false;
@@ -1038,9 +1054,9 @@ var BedrockChatLanguageModel = class {
         bedrockOptions.additionalModelRequestFields = additionalModelRequestFields;
       }
     }
-    const modelSupportsNativeStructuredOutput = supportsNativeStructuredOutput(this.modelId) && (modelSupportsStructuredOutput || isThinkingEnabled);
-    const useNativeStructuredOutput = isAnthropicModel && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && (structuredOutputMode === "outputFormat" || structuredOutputMode === "auto" && modelSupportsNativeStructuredOutput);
-    const useJsonInstructionForStructuredOutput = !useNativeStructuredOutput && structuredOutputMode !== "jsonTool" && isAnthropicModel && !supportsStrictTools(this.modelId) && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && tools != null && tools.length > 0;
+    const modelSupportsNativeStructuredOutput = supportsNativeStructuredOutput(this.modelId) && (modelSupportsStructuredOutput || isThinkingEnabled || this.config.modelFamily === "anthropic");
+    const useNativeStructuredOutput = isAnthropicModel2 && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && (structuredOutputMode === "outputFormat" || structuredOutputMode === "auto" && modelSupportsNativeStructuredOutput);
+    const useJsonInstructionForStructuredOutput = !useNativeStructuredOutput && structuredOutputMode !== "jsonTool" && isAnthropicModel2 && !supportsStrictTools(this.modelId) && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && tools != null && tools.length > 0;
     const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && !useNativeStructuredOutput && !useJsonInstructionForStructuredOutput ? {
       type: "function",
       name: "json",
@@ -1051,6 +1067,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);
@@ -1061,17 +1079,17 @@ var BedrockChatLanguageModel = class {
       };
     }
     if (betas.size > 0 || bedrockOptions.anthropicBeta) {
-      const existingBetas = (_i = bedrockOptions.anthropicBeta) != null ? _i : [];
+      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 = (_j = bedrockOptions.reasoningConfig) == null ? void 0 : _j.type;
-    const thinkingBudget = thinkingType === "enabled" ? (_k = bedrockOptions.reasoningConfig) == null ? void 0 : _k.budgetTokens : void 0;
-    const thinkingDisplay = thinkingType === "adaptive" ? (_l = bedrockOptions.reasoningConfig) == null ? void 0 : _l.display : void 0;
-    const isAnthropicThinkingEnabled = isAnthropicModel && isThinkingEnabled;
+    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 && isThinkingEnabled;
     const inferenceConfig = {
       ...maxOutputTokens != null && { maxTokens: maxOutputTokens },
       ...temperature != null && { temperature },
@@ -1102,8 +1120,8 @@ var BedrockChatLanguageModel = class {
           }
         };
       }
-    } else if (!isAnthropicModel) {
-      if (((_m = bedrockOptions.reasoningConfig) == null ? void 0 : _m.budgetTokens) != null) {
+    } else if (!isAnthropicModel2) {
+      if (((_n = bedrockOptions.reasoningConfig) == null ? void 0 : _n.budgetTokens) != null) {
         warnings.push({
           type: "unsupported",
           feature: "budgetTokens",
@@ -1118,13 +1136,13 @@ var BedrockChatLanguageModel = class {
         });
       }
     }
-    const maxReasoningEffort = (_n = bedrockOptions.reasoningConfig) == null ? void 0 : _n.maxReasoningEffort;
+    const maxReasoningEffort = (_o = bedrockOptions.reasoningConfig) == null ? void 0 : _o.maxReasoningEffort;
     if (maxReasoningEffort != null) {
-      if (isAnthropicModel) {
+      if (isAnthropicModel2) {
         bedrockOptions.additionalModelRequestFields = {
           ...bedrockOptions.additionalModelRequestFields,
           output_config: {
-            ...(_o = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _o.output_config,
+            ...(_p = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _p.output_config,
             effort: maxReasoningEffort
           }
         };
@@ -1135,7 +1153,7 @@ var BedrockChatLanguageModel = class {
         } : {
           ...bedrockOptions.additionalModelRequestFields,
           reasoning: {
-            ...(_p = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _p.reasoning,
+            ...(_q = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _q.reasoning,
             effort: maxReasoningEffort
           }
         };
@@ -1154,7 +1172,7 @@ var BedrockChatLanguageModel = class {
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         output_config: {
-          ...(_q = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _q.output_config,
+          ...(_r = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _r.output_config,
           format: {
             type: "json_schema",
             schema: sanitizeJsonSchema(responseFormat.schema)
@@ -1186,7 +1204,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 = ((_t = (_s = toolConfig.tools) == null ? void 0 : _s.length) != null ? _t : 0) > 0 || additionalTools;
     let filteredPrompt = prompt;
     if (!hasAnyTools) {
       const hasToolContent = prompt.some(
@@ -1231,7 +1249,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,
@@ -2366,7 +2384,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.175" : "0.0.0-test";
+var VERSION = true ? "4.0.177" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {
@@ -2575,9 +2593,47 @@ var BedrockRerankingModel = class {
   }
 };
 
+// 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"
   });
@@ -2617,7 +2673,7 @@ function createAmazonBedrock(options = {}) {
           environmentVariableName: "AWS_SECRET_ACCESS_KEY",
           description: "AWS secret access key"
         }),
-        sessionToken: options.accessKeyId != null && options.secretAccessKey != null ? options.sessionToken : loadOptionalSetting({
+        sessionToken: options.accessKeyId != null && options.secretAccessKey != null ? options.sessionToken : loadOptionalSetting2({
           settingValue: options.sessionToken,
           environmentVariableName: "AWS_SESSION_TOKEN"
         })
@@ -2648,41 +2704,42 @@ Original error: ${errorMessage}`
     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({
-        settingValue: options.region,
-        settingName: "region",
-        environmentVariableName: "AWS_REGION",
-        description: "AWS region"
-      })}.amazonaws.com`
-    )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
-  };
-  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, {
+  const getBedrockRuntimeBaseUrl = () => 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 getBedrockAgentRuntimeBaseUrl = () => resolveBedrockBaseURL({
+    baseURL: options.baseURL,
+    getRegion: () => loadSetting({
+      settingValue: options.region,
+      settingName: "region",
+      environmentVariableName: "AWS_REGION",
+      description: "AWS region"
+    }),
+    service: "bedrock-agent-runtime",
+    serviceEndpointUrlEnvironmentVariableName: "AWS_ENDPOINT_URL_BEDROCK_AGENT_RUNTIME"
+  });
+  const createChatModel = (modelId, settings = {}) => new BedrockChatLanguageModel(modelId, {
     baseUrl: getBedrockRuntimeBaseUrl,
     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, settings = {}) => new BedrockEmbeddingModel(modelId, {
     baseUrl: getBedrockRuntimeBaseUrl,