scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

test: update deepseek-v4 prices in tests to match active discounted rates

Browse source
This commit is contained in:
Dotify71 2026-05-21 00:28:51 +05:30
parent 8e40186764
commit 46b0bae2ca
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/test_utils.py
View file

@ -3992,7 +3992,7 @@ def test_deepseek_v4_models_in_cost_map():
Prices sourced from https://api-docs.deepseek.com/quick_start/pricing:
- deepseek-v4-flash: $0.14/M input, $0.28/M output
- deepseek-v4-pro: $1.74/M input, $3.48/M output
- deepseek-v4-pro: $0.435/M input, $0.87/M output (75% discounted active price)
Closes https://github.com/BerriAI/litellm/issues/26709
"""
@ -4005,8 +4005,8 @@ def test_deepseek_v4_models_in_cost_map():
# --- bare model names ---
for key, expected_input, expected_output, expected_cache in [
("deepseek-v4-flash", 1.4e-07, 2.8e-07, 1.4e-08),
("deepseek-v4-pro", 1.74e-06, 3.48e-06, 1.74e-07),
("deepseek-v4-flash", 1.4e-07, 2.8e-07, 2.8e-09),
("deepseek-v4-pro", 4.35e-07, 8.7e-07, 3.625e-09),
]:
info = model_cost.get(key)
assert info is not None, f"{key} missing from model_prices_and_context_window.json"
@ -4021,8 +4021,8 @@ def test_deepseek_v4_models_in_cost_map():
# --- provider-prefixed names ---
for key, expected_input, expected_output, expected_cache in [
("deepseek/deepseek-v4-flash", 1.4e-07, 2.8e-07, 1.4e-08),
("deepseek/deepseek-v4-pro", 1.74e-06, 3.48e-06, 1.74e-07),
("deepseek/deepseek-v4-flash", 1.4e-07, 2.8e-07, 2.8e-09),
("deepseek/deepseek-v4-pro", 4.35e-07, 8.7e-07, 3.625e-09),
]:
info = model_cost.get(key)
assert info is not None, f"{key} missing from model_prices_and_context_window.json"