diff --git a/tests/integration/cost_calculation/cases/bedrock_mantle.py b/tests/integration/cost_calculation/cases/bedrock_mantle.py index 3cd9150033a..2966168e672 100644 --- a/tests/integration/cost_calculation/cases/bedrock_mantle.py +++ b/tests/integration/cost_calculation/cases/bedrock_mantle.py @@ -1012,20 +1012,22 @@ BEDROCK_MANTLE_CLAUDE_OPUS_5_5_STREAM: Final = CLAUDE_OPUS_5_5_STREAM.model_copy } ) -BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT: Final = with_usage( - CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT, - "bedrock_mantle-anthropic.claude-opus-5-5-messages_input_text", - _ADAPTIVE_USAGE, - _ADAPTIVE_EXPECTED, -).model_copy( - update={ - "model": _MANTLE_CLAUDE_OPUS_MODEL, - "request": { - **CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT.request, - "thinking": {"type": "adaptive"}, - "output_config": {"effort": "medium"}, - }, - } +BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT: Final = _mantle_without_usage_fields( + with_usage( + CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT, + "bedrock_mantle-anthropic.claude-opus-5-5-messages_input_text", + _ADAPTIVE_USAGE, + _ADAPTIVE_EXPECTED, + ).model_copy( + update={ + "model": _MANTLE_CLAUDE_OPUS_MODEL, + "request": { + **CLAUDE_OPUS_5_5_MESSAGES_INPUT_TEXT.request, + "thinking": {"type": "adaptive"}, + "output_config": {"effort": "medium"}, + }, + } + ) ) BEDROCK_MANTLE_CLAUDE_OPUS_5_5_MESSAGES_STREAM: Final = CLAUDE_OPUS_5_5_MESSAGES_STREAM.model_copy( diff --git a/tests/integration/cost_calculation/cases/vertex_ai.py b/tests/integration/cost_calculation/cases/vertex_ai.py index 8590a19646b..119f73c0dfa 100644 --- a/tests/integration/cost_calculation/cases/vertex_ai.py +++ b/tests/integration/cost_calculation/cases/vertex_ai.py @@ -142,6 +142,7 @@ def _vertex_claude_usage( }, "output_tokens": output_tokens, "output_tokens_details": {"thinking_tokens": thinking_tokens}, + "speed": "standard", } @@ -230,6 +231,7 @@ def _vertex_claude_stream_response(model: str, stop_reason: str, text_deltas: tu "cache_read_input_tokens": 0, "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 0, + "speed": "standard", }, }, }, @@ -2916,6 +2918,7 @@ VERTEX_AI_CLAUDE_SONNET_5_5_INPUT_TEXT: Final = CLAUDE_SONNET_5_5_INPUT_TEXT.mod "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 412, "output_tokens_details": {"thinking_tokens": 0}, + "speed": "standard", }, }, ), @@ -2958,6 +2961,7 @@ VERTEX_AI_CLAUDE_OPUS_5_5_INPUT_TEXT: Final = CLAUDE_OPUS_5_5_INPUT_TEXT.model_c "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 412, "output_tokens_details": {"thinking_tokens": 0}, + "speed": "standard", }, }, ), @@ -3001,6 +3005,7 @@ VERTEX_AI_CLAUDE_FABLE_5_1_INPUT_TEXT: Final = CLAUDE_FABLE_5_1_INPUT_TEXT.model "cache_creation": {"ephemeral_5m_input_tokens": 0, "ephemeral_1h_input_tokens": 0}, "output_tokens": 412, "output_tokens_details": {"thinking_tokens": 0}, + "speed": "standard", }, }, ),