@ai-sdk/amazon-bedrock 4.0.174 → 4.0.176

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";
 
@@ -2366,7 +2365,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.174" : "0.0.0-test";
+var VERSION = true ? "4.0.176" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {
@@ -2575,9 +2574,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 +2654,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,28 +2685,28 @@ 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 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) => new BedrockChatLanguageModel(modelId, {
     baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,