From 0eb6816081a1ee5a0e02ce6fcd8bbe36445f4f1f Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sat, 7 Nov 2026 05:57:38 +0000 Subject: [PATCH] fix(gemini): add Gemini 2.5 Pro flex cache price above 200k tokens (#50480) * fix(gemini): add Gemini 2.5 Pro flex cache price above 200k tokens * test(cost-map): allow cache_read_input_token_cost_above_200k_tokens_flex in the row schema --------- Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/model_prices_and_context_window_backup.json | 1 + model_prices_and_context_window.json | 1 + tests/unit/test_utils.py | 1 + 3 files changed, 3 insertions(+) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index 43c321f16a3..3204a00156f 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -32356,6 +32356,7 @@ "output_cost_per_token_flex": 5e-06, "supports_url_context": true, "cache_read_input_token_cost_above_200k_tokens_batches": 2.5e-07, + "cache_read_input_token_cost_above_200k_tokens_flex": 2.5e-07, "deprecation_date": "2026-12-01", "deprecation_source": "https://ai.google.dev/gemini-api/docs/deprecations", "successor_model": "gemini/gemini-3.8-flash", diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 43c321f16a3..3204a00156f 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -32356,6 +32356,7 @@ "output_cost_per_token_flex": 5e-06, "supports_url_context": true, "cache_read_input_token_cost_above_200k_tokens_batches": 2.5e-07, + "cache_read_input_token_cost_above_200k_tokens_flex": 2.5e-07, "deprecation_date": "2026-12-01", "deprecation_source": "https://ai.google.dev/gemini-api/docs/deprecations", "successor_model": "gemini/gemini-3.8-flash", diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py index 5eaabdd2c59..700307f48f5 100644 --- a/tests/unit/test_utils.py +++ b/tests/unit/test_utils.py @@ -1058,6 +1058,7 @@ _COST_MAP_ROW_SCHEMA: Final = { "cache_read_input_token_cost_balanced": {"type": "number"}, "cache_read_input_token_cost_ultrafast": {"type": "number"}, "cache_read_input_token_cost_above_200k_tokens_priority": {"type": "number"}, + "cache_read_input_token_cost_above_200k_tokens_flex": {"type": "number"}, "cache_read_input_token_cost_above_272k_tokens_priority": {"type": "number"}, "input_cost_per_token_flex": {"type": "number"}, "input_cost_per_token_priority": {"type": "number"},