From 50bc6f369f2c94248144e4bad87bad4e4348c392 Mon Sep 17 00:00:00 2001 From: jesus Date: Tue, 8 Sep 2026 01:05:37 +0000 Subject: [PATCH] test(bedrock_mantle): cover gpt-daybreak-blue-5.6-sol pricing and route Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- ...test_bedrock_mantle_responses_transformation.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py index 1f23d39c631..baba8e8770e 100644 --- a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py +++ b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py @@ -20,6 +20,7 @@ from botocore.exceptions import ( ) import litellm +from litellm.llms.bedrock_mantle.common_utils import mantle_base_segment from litellm.llms.bedrock_mantle.responses.transformation import ( BedrockMantleResponsesAPIConfig, ) @@ -1802,6 +1803,19 @@ class TestBedrockMantleResponsesPricing: assert info["output_cost_per_token"] == pytest.approx(8.25e-05) assert info["max_input_tokens"] == 272000 + def test_gpt_daybreak_blue_pricing_and_route(self, local_cost_map): + info = litellm.get_model_info("bedrock_mantle/openai.gpt-daybreak-blue-5.6-sol") + assert info["mode"] == "responses" + assert info["input_cost_per_token"] == pytest.approx(5.5e-06) + assert info["cache_creation_input_token_cost"] == pytest.approx(6.875e-06) + assert info["cache_read_input_token_cost"] == pytest.approx(5.5e-07) + assert info["output_cost_per_token"] == pytest.approx(3.3e-05) + assert info["input_cost_per_token_above_272k_tokens"] == pytest.approx(1.1e-05) + assert info["output_cost_per_token_above_272k_tokens"] == pytest.approx(4.95e-05) + assert info["max_input_tokens"] == 1050000 + assert info["supported_endpoints"] == ["/v1/responses"] + assert mantle_base_segment("openai.gpt-daybreak-blue-5.6-sol", litellm.model_cost) == "openai/v1" + @pytest.mark.parametrize( "model, input_cost, cache_creation_cost, cache_read_cost, output_cost", [