scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix(model_prices): set databricks kimi k3 output ceiling to vendor max

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-08-26 22:23:35 +00:00
parent 04593cfe48
commit e4cdeeac58
3 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/model_prices_and_context_window_backup.json
View file

@ -15469,8 +15469,8 @@
"input_dbu_cost_per_token": 4.2857e-05,
"litellm_provider": "databricks",
"max_input_tokens": 1000000,
"max_output_tokens": 131072,
"max_tokens": 131072,
"max_output_tokens": 1048576,
"max_tokens": 1048576,
"metadata": {
"notes": "Input/output cost per token is dbu cost * $0.070. Number provided for reference, '*_dbu_cost_per_token' used in actual calculation."
},

4
model_prices_and_context_window.json
View file

@ -15469,8 +15469,8 @@
"input_dbu_cost_per_token": 4.2857e-05,
"litellm_provider": "databricks",
"max_input_tokens": 1000000,
"max_output_tokens": 131072,
"max_tokens": 131072,
"max_output_tokens": 1048576,
"max_tokens": 1048576,
"metadata": {
"notes": "Input/output cost per token is dbu cost * $0.070. Number provided for reference, '*_dbu_cost_per_token' used in actual calculation."
},

5
tests/test_litellm/llms/databricks/test_databricks_cost_calculator.py
View file

@ -233,6 +233,11 @@ def test_million_token_context_models_price_and_size_at_published_values(
assert info["supports_prompt_caching"] is True
def test_output_ceilings_match_what_each_vendor_publishes(local_model_cost_map: None) -> None:
assert _model_info("databricks/databricks-kimi-k3")["max_output_tokens"] == 1048576
assert _model_info("databricks/databricks-glm-5-2")["max_output_tokens"] == 131072
def test_kimi_k3_accepts_images_while_glm_5_2_is_text_only(local_model_cost_map: None) -> None:
assert _model_info("databricks/databricks-kimi-k3")["supports_vision"] is True
assert _model_info("databricks/databricks-glm-5-2")["supports_vision"] is False