diff --git a/litellm/llms/bedrock/count_tokens/transformation.py b/litellm/llms/bedrock/count_tokens/transformation.py index 91ffcdcf44..d46ed3aa45 100644 --- a/litellm/llms/bedrock/count_tokens/transformation.py +++ b/litellm/llms/bedrock/count_tokens/transformation.py @@ -92,7 +92,7 @@ class BedrockCountTokensConfig(BaseAWSLLM): system_messages.append({"text": message.get("content", "")}) else: # Transform message content to Bedrock format - transformed_message = {"role": message.get("role"), "content": []} + transformed_message: Dict[str, Any] = {"role": message.get("role"), "content": []} # Handle content - ensure it's in the correct array format content = message.get("content", "")