From 164118df81c850e5af40d9529e6ff390e2758ffb Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 18 Jun 2025 21:36:03 -0700
Subject: [PATCH] docs(litellm_logging_code_coverage.py): fix check

---
 ...odel_prices_and_context_window_backup.json | 42 +++++++++++++++++++
 .../litellm_logging_code_coverage.py          |  8 ++--
 2 files changed, 46 insertions(+), 4 deletions(-)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index 5fae9ae4dd..66d94cff7f 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.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,
diff --git a/tests/code_coverage_tests/litellm_logging_code_coverage.py b/tests/code_coverage_tests/litellm_logging_code_coverage.py
index eb927e006e..9825cfba1e 100644
--- a/tests/code_coverage_tests/litellm_logging_code_coverage.py
+++ b/tests/code_coverage_tests/litellm_logging_code_coverage.py
@@ -64,12 +64,12 @@ ignored_function_names = [
 
 
 def main():
-    # logging_file = "./litellm/litellm_core_utils/litellm_logging.py"
-    # tests_dir = "./tests/"
+    logging_file = "./litellm/litellm_core_utils/litellm_logging.py"
+    tests_dir = "./tests/"
 
     # LOCAL TESTING
-    logging_file = "../../litellm/litellm_core_utils/litellm_logging.py"
-    tests_dir = "../../tests/"
+    # logging_file = "../../litellm/litellm_core_utils/litellm_logging.py"
+    # tests_dir = "../../tests/"
 
     logging_functions = get_function_names_from_file(logging_file)
     print("logging_functions:", logging_functions)