diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index c124f1c7d8..b93ca94bed 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -25,6 +25,7 @@ from litellm.llms.base_llm.chat.transformation import BaseConfig, BaseLLMExcepti from litellm.types.llms.bedrock import * from litellm.types.llms.openai import ( AllMessageValues, + ChatCompletionAssistantMessage, ChatCompletionRedactedThinkingBlock, ChatCompletionResponseMessage, ChatCompletionSystemMessage, @@ -505,6 +506,7 @@ class AmazonConverseConfig(BaseConfig): OpenAIMessageContentListBlock, ChatCompletionUserMessage, ChatCompletionSystemMessage, + ChatCompletionAssistantMessage, ], block_type: Literal["system"], ) -> Optional[SystemContentBlock]: @@ -517,6 +519,7 @@ class AmazonConverseConfig(BaseConfig): OpenAIMessageContentListBlock, ChatCompletionUserMessage, ChatCompletionSystemMessage, + ChatCompletionAssistantMessage, ], block_type: Literal["content_block"], ) -> Optional[ContentBlock]: @@ -528,6 +531,7 @@ class AmazonConverseConfig(BaseConfig): OpenAIMessageContentListBlock, ChatCompletionUserMessage, ChatCompletionSystemMessage, + ChatCompletionAssistantMessage, ], block_type: Literal["system", "content_block"], ) -> Optional[Union[SystemContentBlock, ContentBlock]]: