diff --git a/tests/test_litellm/test_cost_calculator.py b/tests/test_litellm/test_cost_calculator.py index db3e6511db2..18ab8a2a07a 100644 --- a/tests/test_litellm/test_cost_calculator.py +++ b/tests/test_litellm/test_cost_calculator.py @@ -2073,6 +2073,7 @@ def test_gemini_3_1_flash_lite_pricing(): assert model_info["input_cost_per_token"] == 4.5e-07 assert model_info["input_cost_per_audio_token"] == 9e-07 assert model_info["output_cost_per_token"] == 2.7e-06 + assert model_info["output_cost_per_reasoning_token"] == 2.7e-06 assert model_info["cache_read_input_token_cost"] == 4.5e-08 assert model_info["max_input_tokens"] == 1048576