scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix(ci): sync cost map schema and intended schema with balanced keys

Browse source
This commit is contained in:
shrey kharbanda 2026-09-24 05:43:03 +00:00
parent 472f8d035a
commit 17c0dd4a0a
3 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
ci_cd/generate_model_prices_schema.py
View file

@ -228,8 +228,6 @@ COST_DESCRIPTIONS: dict[str, str] = {
def cost_description(key: str) -> Optional[str]:
if key in COST_DESCRIPTIONS:
return COST_DESCRIPTIONS[key]
if key.endswith("_balanced"):
return "Balanced service-tier rate for the same-named base field."
if key.endswith("_flex"):
return "Flex service-tier rate for the same-named base field."
if key.endswith("_priority"):

9
model_prices_and_context_window.schema.json
View file

@ -202,8 +202,7 @@
},
"cache_read_input_token_cost_balanced": {
"type": "number",
"minimum": 0,
"description": "Balanced service-tier rate for the same-named base field."
"minimum": 0
},
"cache_read_input_token_cost_batches": {
"type": "number",
@ -383,8 +382,7 @@
},
"input_cost_per_token_balanced": {
"type": "number",
"minimum": 0,
"description": "Balanced service-tier rate for the same-named base field."
"minimum": 0
},
"input_cost_per_token_batches": {
"type": "number",
@ -724,8 +722,7 @@
},
"output_cost_per_token_balanced": {
"type": "number",
"minimum": 0,
"description": "Balanced service-tier rate for the same-named base field."
"minimum": 0
},
"output_cost_per_token_batches": {
"type": "number",

3
tests/test_litellm/test_utils.py
View file

@ -793,10 +793,12 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"input_cost_per_token_above_256k_tokens": {"type": "number"},
"input_cost_per_token_above_272k_tokens": {"type": "number"},
"input_cost_per_token_above_512k_tokens": {"type": "number"},
"cache_read_input_token_cost_balanced": {"type": "number"},
"cache_read_input_token_cost_flex": {"type": "number"},
"cache_read_input_token_cost_priority": {"type": "number"},
"cache_read_input_token_cost_above_200k_tokens_priority": {"type": "number"},
"cache_read_input_token_cost_above_272k_tokens_priority": {"type": "number"},
"input_cost_per_token_balanced": {"type": "number"},
"input_cost_per_token_flex": {"type": "number"},
"input_cost_per_token_priority": {"type": "number"},
"input_cost_per_token_above_200k_tokens_priority": {"type": "number"},
@ -804,6 +806,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"input_cost_per_token_above_272k_tokens_batches": {"type": "number"},
"input_cost_per_token_above_272k_tokens_flex": {"type": "number"},
"input_cost_per_audio_token_priority": {"type": "number"},
"output_cost_per_token_balanced": {"type": "number"},
"output_cost_per_token_flex": {"type": "number"},
"output_cost_per_token_priority": {"type": "number"},
"output_cost_per_token_above_200k_tokens_priority": {"type": "number"},