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

package/src/bedrock-provider.ts

@@ -1,10 +1,10 @@
 import { anthropicTools } from '@ai-sdk/anthropic/internal';
 import {
-  EmbeddingModelV3,
-  ImageModelV3,
-  LanguageModelV3,
-  ProviderV3,
-  RerankingModelV3,
+  EmbeddingModelV4,
+  ImageModelV4,
+  LanguageModelV4,
+  ProviderV4,
+  RerankingModelV4,
 } from '@ai-sdk/provider';
 import {
   FetchFunction,
@@ -107,50 +107,50 @@ export interface AmazonBedrockProviderSettings {
   generateId?: () => string;
 }
 
-export interface AmazonBedrockProvider extends ProviderV3 {
-  (modelId: BedrockChatModelId): LanguageModelV3;
+export interface AmazonBedrockProvider extends ProviderV4 {
+  (modelId: BedrockChatModelId): LanguageModelV4;
 
-  languageModel(modelId: BedrockChatModelId): LanguageModelV3;
+  languageModel(modelId: BedrockChatModelId): LanguageModelV4;
 
   /**
    * Creates a model for text embeddings.
    */
-  embedding(modelId: BedrockEmbeddingModelId): EmbeddingModelV3;
+  embedding(modelId: BedrockEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * Creates a model for text embeddings.
    */
-  embeddingModel(modelId: BedrockEmbeddingModelId): EmbeddingModelV3;
+  embeddingModel(modelId: BedrockEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * @deprecated Use `embedding` instead.
    */
-  textEmbedding(modelId: BedrockEmbeddingModelId): EmbeddingModelV3;
+  textEmbedding(modelId: BedrockEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * @deprecated Use `embeddingModel` instead.
    */
-  textEmbeddingModel(modelId: BedrockEmbeddingModelId): EmbeddingModelV3;
+  textEmbeddingModel(modelId: BedrockEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * Creates a model for image generation.
    */
-  image(modelId: BedrockImageModelId): ImageModelV3;
+  image(modelId: BedrockImageModelId): ImageModelV4;
 
   /**
    * Creates a model for image generation.
    */
-  imageModel(modelId: BedrockImageModelId): ImageModelV3;
+  imageModel(modelId: BedrockImageModelId): ImageModelV4;
 
   /**
    * Creates a model for reranking documents.
    */
-  reranking(modelId: BedrockRerankingModelId): RerankingModelV3;
+  reranking(modelId: BedrockRerankingModelId): RerankingModelV4;
 
   /**
    * Creates a model for reranking documents.
    */
-  rerankingModel(modelId: BedrockRerankingModelId): RerankingModelV3;
+  rerankingModel(modelId: BedrockRerankingModelId): RerankingModelV4;
 
   /**
    * Anthropic-specific tools that can be used with Anthropic models on Bedrock.
@@ -330,7 +330,7 @@ export function createAmazonBedrock(
       fetch: fetchFunction,
     });
 
-  provider.specificationVersion = 'v3' as const;
+  provider.specificationVersion = 'v4' as const;
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.embeddingModel = createEmbeddingModel;

package/src/convert-bedrock-usage.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV3Usage } from '@ai-sdk/provider';
+import { LanguageModelV4Usage } from '@ai-sdk/provider';
 
 export type BedrockUsage = {
   inputTokens: number;
@@ -10,7 +10,7 @@ export type BedrockUsage = {
 
 export function convertBedrockUsage(
   usage: BedrockUsage | undefined | null,
-): LanguageModelV3Usage {
+): LanguageModelV4Usage {
   if (usage == null) {
     return {
       inputTokens: {

package/src/convert-to-bedrock-chat-messages.ts

@@ -1,8 +1,8 @@
 import {
   JSONObject,
-  LanguageModelV3Message,
-  LanguageModelV3Prompt,
-  SharedV3ProviderMetadata,
+  LanguageModelV4Message,
+  LanguageModelV4Prompt,
+  SharedV4ProviderMetadata,
   UnsupportedFunctionalityError,
 } from '@ai-sdk/provider';
 import {
@@ -28,7 +28,7 @@ import { bedrockFilePartProviderOptions } from './bedrock-chat-options';
 import { normalizeToolCallId } from './normalize-tool-call-id';
 
 function getCachePoint(
-  providerMetadata: SharedV3ProviderMetadata | undefined,
+  providerMetadata: SharedV4ProviderMetadata | undefined,
 ): BedrockCachePoint | undefined {
   const cachePointConfig = providerMetadata?.bedrock?.cachePoint as
     | BedrockCachePoint['cachePoint']
@@ -42,7 +42,7 @@ function getCachePoint(
 }
 
 async function shouldEnableCitations(
-  providerMetadata: SharedV3ProviderMetadata | undefined,
+  providerMetadata: SharedV4ProviderMetadata | undefined,
 ): Promise<boolean> {
   const bedrockOptions = await parseProviderOptions({
     provider: 'bedrock',
@@ -54,7 +54,7 @@ async function shouldEnableCitations(
 }
 
 export async function convertToBedrockChatMessages(
-  prompt: LanguageModelV3Prompt,
+  prompt: LanguageModelV4Prompt,
   isMistral: boolean = false,
 ): Promise<{
   system: BedrockSystemMessages;
@@ -287,33 +287,41 @@ export async function convertToBedrockChatMessages(
                   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?.signature != null) {
+                  // do not trim reasoning text when a signature is present:
+                  // the signature validates the exact original bytes
+                  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?.redactedData != null) {
+                  bedrockContent.push({
+                    reasoningContent: {
+                      redactedReasoning: {
+                        data: reasoningMetadata.redactedData,
                       },
-                    });
-                  }
+                    },
+                  });
+                } else {
+                  // 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
+                  bedrockContent.push({
+                    reasoningContent: {
+                      reasoningText: {
+                        text: trimIfLast(
+                          isLastBlock,
+                          isLastMessage,
+                          isLastContentPart,
+                          part.text,
+                        ),
+                      },
+                    },
+                  });
                 }
 
                 break;
@@ -400,19 +408,19 @@ function trimIfLast(
 
 type SystemBlock = {
   type: 'system';
-  messages: Array<LanguageModelV3Message & { role: 'system' }>;
+  messages: Array<LanguageModelV4Message & { role: 'system' }>;
 };
 type AssistantBlock = {
   type: 'assistant';
-  messages: Array<LanguageModelV3Message & { role: 'assistant' }>;
+  messages: Array<LanguageModelV4Message & { role: 'assistant' }>;
 };
 type UserBlock = {
   type: 'user';
-  messages: Array<LanguageModelV3Message & { role: 'user' | 'tool' }>;
+  messages: Array<LanguageModelV4Message & { role: 'user' | 'tool' }>;
 };
 
 function groupIntoBlocks(
-  prompt: LanguageModelV3Prompt,
+  prompt: LanguageModelV4Prompt,
 ): Array<SystemBlock | AssistantBlock | UserBlock> {
   const blocks: Array<SystemBlock | AssistantBlock | UserBlock> = [];
   let currentBlock: SystemBlock | AssistantBlock | UserBlock | undefined =

package/src/map-bedrock-finish-reason.ts

@@ -1,10 +1,10 @@
-import { LanguageModelV3FinishReason } from '@ai-sdk/provider';
+import { LanguageModelV4FinishReason } from '@ai-sdk/provider';
 import { BedrockStopReason } from './bedrock-api-types';
 
 export function mapBedrockFinishReason(
   finishReason: BedrockStopReason,
   isJsonResponseFromTool?: boolean,
-): LanguageModelV3FinishReason['unified'] {
+): LanguageModelV4FinishReason['unified'] {
   switch (finishReason) {
     case 'stop_sequence':
     case 'end_turn':

package/src/reranking/bedrock-reranking-model.ts

@@ -1,4 +1,4 @@
-import { RerankingModelV3 } from '@ai-sdk/provider';
+import { RerankingModelV4 } from '@ai-sdk/provider';
 import {
   FetchFunction,
   Resolvable,
@@ -26,10 +26,10 @@ type BedrockRerankingConfig = {
   fetch?: FetchFunction;
 };
 
-type DoRerankResponse = Awaited<ReturnType<RerankingModelV3['doRerank']>>;
+type DoRerankResponse = Awaited<ReturnType<RerankingModelV4['doRerank']>>;
 
-export class BedrockRerankingModel implements RerankingModelV3 {
-  readonly specificationVersion = 'v3';
+export class BedrockRerankingModel implements RerankingModelV4 {
+  readonly specificationVersion = 'v4';
   readonly provider = 'amazon-bedrock';
 
   constructor(
@@ -44,7 +44,7 @@ export class BedrockRerankingModel implements RerankingModelV3 {
     topN,
     abortSignal,
     providerOptions,
-  }: Parameters<RerankingModelV3['doRerank']>[0]): Promise<DoRerankResponse> {
+  }: Parameters<RerankingModelV4['doRerank']>[0]): Promise<DoRerankResponse> {
     const bedrockOptions = await parseProviderOptions({
       provider: 'bedrock',
       providerOptions,