scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test(bedrock_mantle): cover gpt-oss-safeguard-20b input and output rates

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-09-05 13:46:46 +00:00
parent bbf51146ca
commit 528f3ae016
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_transformation.py
View file

@ -700,6 +700,14 @@ class TestBedrockMantlePricing:
assert info["input_cost_per_token"] == pytest.approx(7e-8)
assert info["output_cost_per_token"] == pytest.approx(3e-7)
def test_gpt_oss_safeguard_20b_pricing(self, monkeypatch):
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "true")
litellm.add_known_models()
info = litellm.get_model_info("bedrock_mantle/openai.gpt-oss-safeguard-20b")
# Bedrock pricing: $0.07/M input, $0.20/M output
assert info["input_cost_per_token"] == pytest.approx(7e-8)
assert info["output_cost_per_token"] == pytest.approx(2e-7)
def test_pricing_significantly_cheaper_than_openai_native(self, monkeypatch):
"""
Verify Bedrock Mantle pricing is cheaper than OpenAI's direct API pricing.