chore: use the gemini-1.5-pro-001 for caching system test.

This model is the officially supported model after the feature is launched.

PiperOrigin-RevId: 642822908

Author: Zhenyi Qi

tests/system/vertexai/test_generative_models.py

@@ -34,7 +34,7 @@
 GEMINI_MODEL_NAME = "gemini-1.0-pro-002"
 GEMINI_VISION_MODEL_NAME = "gemini-1.0-pro-vision"
 GEMINI_15_MODEL_NAME = "gemini-1.5-pro-preview-0409"
-GEMINI_15_0514_MODEL_NAME = "gemini-1.5-pro-preview-0514"
+GEMINI_15_PRO_MODEL_NAME = "gemini-1.5-pro-001"
 
 
 # A dummy function for function calling
@@ -102,7 +102,7 @@ def setup_method(self):
 
     def test_generate_content_with_cached_content_from_text(self):
         cached_content = caching.CachedContent.create(
-            model_name=GEMINI_15_0514_MODEL_NAME,
+            model_name=GEMINI_15_PRO_MODEL_NAME,
             system_instruction="Please answer all the questions like a pirate.",
             contents=[
                 Content.from_dict(