@ai-sdk/amazon-bedrock 5.0.0-beta.2 → 5.0.0-beta.20

package/dist/index.mjs

@@ -13,10 +13,14 @@ import {
   combineHeaders,
   createJsonErrorResponseHandler,
   createJsonResponseHandler,
+  isCustomReasoning,
+  mapReasoningToProviderBudget,
+  mapReasoningToProviderEffort,
   parseProviderOptions as parseProviderOptions2,
   postJsonToApi,
   resolve
 } from "@ai-sdk/provider-utils";
+import { getModelCapabilities } from "@ai-sdk/anthropic/internal";
 import { z as z3 } from "zod/v4";
 
 // src/bedrock-api-types.ts
@@ -83,7 +87,17 @@ var amazonBedrockLanguageModelOptions = z.object({
   /**
    * Anthropic beta features to enable
    */
-  anthropicBeta: z.array(z.string()).optional()
+  anthropicBeta: z.array(z.string()).optional(),
+  /**
+   * Service tier for the request.
+   * @see https://docs.aws.amazon.com/bedrock/latest/userguide/service-tiers-inference.html
+   *
+   * - 'reserved': Uses provisioned throughput capacity
+   * - 'priority': Prioritizes low-latency inference when capacity is available
+   * - 'default': Standard on-demand tier
+   * - 'flex': Lower-cost tier for flexible latency workloads
+   */
+  serviceTier: z.enum(["reserved", "priority", "default", "flex"]).optional()
 });
 
 // src/bedrock-error.ts
@@ -240,7 +254,8 @@ async function prepareTools({
     } = await prepareAnthropicTools({
       tools: ProviderTools,
       toolChoice,
-      supportsStructuredOutput: false
+      supportsStructuredOutput: false,
+      supportsStrictTools: false
     });
     toolWarnings.push(...anthropicToolWarnings);
     anthropicBetas.forEach((beta) => betas.add(beta));
@@ -592,33 +607,36 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
                   providerOptions: part.providerOptions,
                   schema: bedrockReasoningMetadataSchema
                 });
-                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
-                        }
+                if ((reasoningMetadata == null ? void 0 : reasoningMetadata.signature) != null) {
+                  bedrockContent.push({
+                    reasoningContent: {
+                      reasoningText: {
+                        text: part.text,
+                        signature: reasoningMetadata.signature
                       }
-                    });
-                  } else if (reasoningMetadata.redactedData != null) {
-                    bedrockContent.push({
-                      reasoningContent: {
-                        redactedReasoning: {
-                          data: reasoningMetadata.redactedData
-                        }
+                    }
+                  });
+                } else if ((reasoningMetadata == null ? void 0 : reasoningMetadata.redactedData) != null) {
+                  bedrockContent.push({
+                    reasoningContent: {
+                      redactedReasoning: {
+                        data: reasoningMetadata.redactedData
                       }
-                    });
-                  }
+                    }
+                  });
+                } else {
+                  bedrockContent.push({
+                    reasoningContent: {
+                      reasoningText: {
+                        text: trimIfLast(
+                          isLastBlock,
+                          isLastMessage,
+                          isLastContentPart,
+                          part.text
+                        )
+                      }
+                    }
+                  });
                 }
                 break;
               }
@@ -749,7 +767,7 @@ var BedrockChatLanguageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v3";
+    this.specificationVersion = "v4";
     this.provider = "amazon-bedrock";
     this.supportedUrls = {
       // no supported urls for bedrock
@@ -768,10 +786,11 @@ var BedrockChatLanguageModel = class {
     seed,
     tools,
     toolChoice,
+    reasoning,
     providerOptions
   }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
-    const bedrockOptions = (_a = await parseProviderOptions2({
+    let bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
       schema: amazonBedrockLanguageModelOptions
@@ -818,6 +837,14 @@ var BedrockChatLanguageModel = class {
       });
     }
     const isAnthropicModel = this.modelId.includes("anthropic");
+    const isOpenAIModel = this.modelId.startsWith("openai.");
+    bedrockOptions = resolveBedrockReasoningConfig({
+      reasoning,
+      bedrockOptions,
+      warnings,
+      isAnthropicModel,
+      modelId: this.modelId
+    });
     const isThinkingEnabled = ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.type) === "enabled" || ((_c = bedrockOptions.reasoningConfig) == null ? void 0 : _c.type) === "adaptive";
     const useNativeStructuredOutput = isAnthropicModel && isThinkingEnabled && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null;
     const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && !useNativeStructuredOutput ? {
@@ -895,7 +922,6 @@ var BedrockChatLanguageModel = class {
       }
     }
     const maxReasoningEffort = (_h = bedrockOptions.reasoningConfig) == null ? void 0 : _h.maxReasoningEffort;
-    const isOpenAIModel = this.modelId.startsWith("openai.");
     if (maxReasoningEffort != null) {
       if (isAnthropicModel) {
         bedrockOptions.additionalModelRequestFields = {
@@ -991,6 +1017,7 @@ var BedrockChatLanguageModel = class {
     const {
       reasoningConfig: _,
       additionalModelRequestFields: __,
+      serviceTier: ___,
       ...filteredBedrockOptions
     } = (providerOptions == null ? void 0 : providerOptions.bedrock) || {};
     const additionalModelResponseFieldPaths = isAnthropicModel ? ["/delta/stop_sequence"] : void 0;
@@ -1005,6 +1032,11 @@ var BedrockChatLanguageModel = class {
         ...Object.keys(inferenceConfig).length > 0 && {
           inferenceConfig
         },
+        ...bedrockOptions.serviceTier != null && {
+          serviceTier: {
+            type: bedrockOptions.serviceTier
+          }
+        },
         ...filteredBedrockOptions,
         ...toolConfig.tools !== void 0 && toolConfig.tools.length > 0 ? { toolConfig } : {}
       },
@@ -1554,6 +1586,62 @@ var bedrockReasoningMetadataSchema = z3.object({
   signature: z3.string().optional(),
   redactedData: z3.string().optional()
 });
+var bedrockReasoningEffortMap = {
+  minimal: "low",
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: "max"
+};
+function resolveBedrockReasoningConfig({
+  reasoning,
+  bedrockOptions,
+  warnings,
+  isAnthropicModel,
+  modelId
+}) {
+  if (!isCustomReasoning(reasoning) || bedrockOptions.reasoningConfig != null) {
+    return bedrockOptions;
+  }
+  const result = { ...bedrockOptions };
+  if (isAnthropicModel) {
+    const capabilities = getModelCapabilities(modelId);
+    if (reasoning === "none") {
+      result.reasoningConfig = { type: "disabled" };
+    } else if (capabilities.supportsAdaptiveThinking) {
+      const effort = mapReasoningToProviderEffort({
+        reasoning,
+        effortMap: bedrockReasoningEffortMap,
+        warnings
+      });
+      result.reasoningConfig = {
+        type: "adaptive",
+        maxReasoningEffort: effort
+      };
+    } else {
+      const budgetTokens = mapReasoningToProviderBudget({
+        reasoning,
+        maxOutputTokens: capabilities.maxOutputTokens,
+        maxReasoningBudget: capabilities.maxOutputTokens,
+        warnings
+      });
+      if (budgetTokens != null) {
+        result.reasoningConfig = {
+          type: "enabled",
+          budgetTokens
+        };
+      }
+    }
+  } else if (reasoning !== "none") {
+    const effort = mapReasoningToProviderEffort({
+      reasoning,
+      effortMap: bedrockReasoningEffortMap,
+      warnings
+    });
+    result.reasoningConfig = { maxReasoningEffort: effort };
+  }
+  return result;
+}
 
 // src/bedrock-embedding-model.ts
 import {
@@ -1626,7 +1714,7 @@ var BedrockEmbeddingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v3";
+    this.specificationVersion = "v4";
     this.provider = "amazon-bedrock";
     this.maxEmbeddingsPerCall = 1;
     this.supportsParallelCalls = true;
@@ -1766,7 +1854,7 @@ var BedrockImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v3";
+    this.specificationVersion = "v4";
     this.provider = "amazon-bedrock";
   }
   get maxImagesPerCall() {
@@ -1969,7 +2057,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.2" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.20" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
@@ -2102,7 +2190,7 @@ var BedrockRerankingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v3";
+    this.specificationVersion = "v4";
     this.provider = "amazon-bedrock";
   }
   async doRerank({
@@ -2306,7 +2394,7 @@ Original error: ${errorMessage}`
     headers: getHeaders,
     fetch: fetchFunction
   });
-  provider.specificationVersion = "v3";
+  provider.specificationVersion = "v4";
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.embeddingModel = createEmbeddingModel;