diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index df83d0117e8..0107221327a 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -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"