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

Merge pull request #41263 from BerriAI/litellm_usgov_rows_allow_price_list_source

Browse source 
test(pricing): let synced GovCloud Bedrock rows cite the AWS price list
This commit is contained in:
kerry-berri 2026-09-15 10:41:42 -07:00 • committed by GitHub
commit b97bc10ec9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/test_bedrock_usgov_pricing.py
View file

@ -112,6 +112,11 @@ GOV_ROW_SOURCES = {
}
BEDROCK_PRICE_LIST_URL = (
"https://pricing.us-east-1.amazonaws.com/offers/v1.0/aws/AmazonBedrockFoundationModels/current/index.json"
)
def _non_pricing_fields(info):
return {k: v for k, v in info.items() if "cost" not in k and k not in ("litellm_provider", "source")}
@ -121,8 +126,10 @@ def test_usgov_rows_keep_commercial_limits_and_capabilities(model_data, gov_key)
"""A gov row differs from the commercial row it mirrors only in price and
provider: context limits, mode, and capability flags stay identical, so a
hand-copied row cannot silently drop tool calling or shrink the context window.
The only source a gov row may cite is the AWS price list, which prices the
us-gov regions itself; a commercial doc URL copied along with the row is not.
"""
gov = model_data[gov_key]
assert _non_pricing_fields(gov) == _non_pricing_fields(model_data[GOV_ROW_SOURCES[gov_key]])
assert "search_context_cost_per_query" not in gov
assert "source" not in gov
assert gov.get("source", BEDROCK_PRICE_LIST_URL) == BEDROCK_PRICE_LIST_URL