diff --git a/tests/integration/translation/chat_completions/bases/bedrock_invoke.py b/tests/integration/translation/chat_completions/bases/bedrock_invoke.py index d86ce2f861f..6356bbc07e0 100644 --- a/tests/integration/translation/chat_completions/bases/bedrock_invoke.py +++ b/tests/integration/translation/chat_completions/bases/bedrock_invoke.py @@ -175,10 +175,11 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( }, mock_provider_response={ "model": "claude-sonnet-4-6", - "id": "msg_bdrk_011CfjukowJGNdRPkmrmvmEU", + "id": "msg_bdrk_4ky3vxhsv4ko5oyipjlcejlaxydydvgsfe4tnjdng5wsfzlzalca", "type": "message", "role": "assistant", "content": [{"type": "text", "text": "Hello!"}], + "container": None, "stop_reason": "end_turn", "stop_sequence": None, "stop_details": None, @@ -188,6 +189,7 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( "cache_read_input_tokens": 0, "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 5, + "service_tier": "standard", }, }, expected_litellm_response={ @@ -220,6 +222,7 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( }, "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0, + "service_tier": "standard", }, }, ) diff --git a/tests/integration/translation/messages/bases/bedrock_invoke.py b/tests/integration/translation/messages/bases/bedrock_invoke.py index 7ec8b19ea46..e89e3888402 100644 --- a/tests/integration/translation/messages/bases/bedrock_invoke.py +++ b/tests/integration/translation/messages/bases/bedrock_invoke.py @@ -141,10 +141,11 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( }, mock_provider_response={ "model": "claude-sonnet-4-6", - "id": "msg_bdrk_011CfjukiGyrRwzy4SYisy1x", + "id": "msg_bdrk_72khh2guasoaflwckvmpzldy67lxs4h745h7l7zujbyeizbpvxfq", "type": "message", "role": "assistant", - "content": [{"type": "text", "text": "Hello! How can I help you?"}], + "content": [{"type": "text", "text": "Hello!"}], + "container": None, "stop_reason": "end_turn", "stop_sequence": None, "stop_details": None, @@ -153,15 +154,17 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0, "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, - "output_tokens": 11, + "output_tokens": 5, + "service_tier": "standard", }, }, expected_litellm_response={ "model": "bedrock/invoke/us.anthropic.claude-sonnet-4-6", - "id": "msg_bdrk_011CfjukiGyrRwzy4SYisy1x", + "id": "msg_bdrk_72khh2guasoaflwckvmpzldy67lxs4h745h7l7zujbyeizbpvxfq", "type": "message", "role": "assistant", - "content": [{"type": "text", "text": "Hello! How can I help you?"}], + "content": [{"type": "text", "text": "Hello!"}], + "container": None, "stop_reason": "end_turn", "stop_sequence": None, "stop_details": None, @@ -170,7 +173,8 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0, "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, - "output_tokens": 11, + "output_tokens": 5, + "service_tier": "standard", }, }, ) diff --git a/tests/integration/translation/responses/bases/bedrock_invoke.py b/tests/integration/translation/responses/bases/bedrock_invoke.py index 829e0231335..7d031da2725 100644 --- a/tests/integration/translation/responses/bases/bedrock_invoke.py +++ b/tests/integration/translation/responses/bases/bedrock_invoke.py @@ -203,10 +203,11 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( }, mock_provider_response={ "model": "claude-sonnet-4-6", - "id": "msg_bdrk_011CfjukowJGNdRPkmrmvmEU", + "id": "msg_bdrk_zvuu2yb6ecoxss7yoi6aiednkwkcuz4s5ttchbfzobevewukiauq", "type": "message", "role": "assistant", "content": [{"type": "text", "text": "Hello!"}], + "container": None, "stop_reason": "end_turn", "stop_sequence": None, "stop_details": None, @@ -216,6 +217,7 @@ CLAUDE_SONNET_4_6_TEST_CASE: Final = TranslationTestCase( "cache_read_input_tokens": 0, "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 5, + "service_tier": "standard", }, }, expected_litellm_response={