From e258217e04848e43c417bcfe7752fdaaf5440746 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Wed, 23 Sep 2026 01:38:16 +0000
Subject: [PATCH] test(vertex_ai): use mapped Gemini model in cache accounting
 tests

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../gemini/test_vertex_and_google_ai_studio_gemini.py       | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
index 2d7d27213ca..d31322c9ba4 100644
--- a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
+++ b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
@@ -6352,7 +6352,7 @@ def test_add_cache_creation_usage_preserves_cache_read_and_bills_creation_tokens
     )
     creation: VertexAICachedContentCreation = {
         "name": "cached-content",
-        "model": "gemini-2.0-flash",
+        "model": "gemini-3.8-flash",
         "total_token_count": 10000,
         "create_time": None,
         "expire_time": None,
@@ -6366,7 +6366,7 @@ def test_add_cache_creation_usage_preserves_cache_read_and_bills_creation_tokens
     assert merged_usage.prompt_tokens_details.cached_tokens == 10000
     assert merged_usage.prompt_tokens_details.cache_creation_tokens == 10000
 
-    model = "gemini-2.0-flash"
+    model = "gemini-3.8-flash"
     base_response = ModelResponse(model=model, usage=base_usage)
     merged_response = ModelResponse(model=model, usage=merged_usage)
     base_cost = litellm.completion_cost(
@@ -6388,7 +6388,7 @@ def test_add_cache_creation_usage_preserves_cache_read_and_bills_creation_tokens
 
 @pytest.mark.parametrize("include_creation", [True, False])
 def test_transform_response_applies_cache_creation_usage(include_creation):
-    model = "gemini-2.0-flash"
+    model = "gemini-3.8-flash"
     logging_obj = MagicMock()
     logging_obj.model_call_details = (
         {