scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

test(bedrock_mantle): cover gpt-daybreak-blue-5.6-sol pricing and route

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
jesus 2026-09-08 01:05:37 +00:00
parent 1164efe636
commit 50bc6f369f
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py
View file

@ -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",
[