From ca6fa63362d2e3889aa40b67727a7238337a3430 Mon Sep 17 00:00:00 2001
From: Low Jian Sheng <15527690+lowjiansheng@users.noreply.github.com>
Date: Thu, 19 Jun 2025 11:16:48 +0800
Subject: [PATCH] Fix gemini 2.5 flash config (#11830)

* fix gemini 2.5 flash config

* add gemini 2.5 flash
---
 model_prices_and_context_window.json | 42 ++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 5fae9ae4dde..66d94cff7f2 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -6761,6 +6761,48 @@
         "source": "https://cloud.google.com/vertex-ai/generative-ai/pricing",
         "supports_web_search": true
     },
+    "gemini/gemini-2.5-flash": {
+        "max_tokens": 65535,
+        "max_input_tokens": 1048576,
+        "max_output_tokens": 65535,
+        "max_images_per_prompt": 3000,
+        "max_videos_per_prompt": 10,
+        "max_video_length": 1,
+        "max_audio_length_hours": 8.4,
+        "max_audio_per_prompt": 1,
+        "max_pdf_size_mb": 30,
+        "input_cost_per_audio_token": 1e-06,
+        "input_cost_per_token": 3e-07,
+        "output_cost_per_token": 2.5e-06,
+        "output_cost_per_reasoning_token": 2.5e-06,
+        "litellm_provider": "gemini",
+        "mode": "chat",
+        "supports_reasoning": true,
+        "supports_system_messages": true,
+        "supports_function_calling": true,
+        "supports_vision": true,
+        "supports_response_schema": true,
+        "supports_audio_output": false,
+        "supports_tool_choice": true,
+        "supported_endpoints": [
+            "/v1/chat/completions",
+            "/v1/completions",
+            "/v1/batch"
+        ],
+        "supported_modalities": [
+            "text",
+            "image",
+            "audio",
+            "video"
+        ],
+        "supported_output_modalities": [
+            "text"
+        ],
+        "source": "https://ai.google.dev/gemini-api/docs/models#gemini-2.5-flash-preview",
+        "supports_parallel_function_calling": true,
+        "supports_web_search": true,
+        "supports_url_context": true
+    },
     "gemini-2.5-flash": {
         "max_tokens": 65535,
         "max_input_tokens": 1048576,