test: Update java-vertexai IT models (Gemini 1.x) (#11550)

copybara-service[bot] · vertex-sdk-bot · web-flow · commit 8b8d3fa98d37 · 2025-05-12T10:41:56.000-04:00
Fixes #11542 Includes reformat using the new com.spotify.fmt:fmt-maven-plugin plugin PiperOrigin-RevId: 756608894 Co-authored-by: A Vertex SDK engineer <vertex-sdk-bot@google.com>
diff --git a/java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITChatSessionIntegrationTest.java b/java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITChatSessionIntegrationTest.java
@@ -46,7 +46,7 @@
 @RunWith(JUnit4.class)
 public class ITChatSessionIntegrationTest {
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
-  private static final String MODEL_NAME_TEXT = "gemini-pro";
+  private static final String MODEL_NAME_TEXT = "gemini-1.5-pro-002";
   private static final String LOCATION = "us-central1";
   private static final Logger logger =
       Logger.getLogger(ITGenerativeModelIntegrationTest.class.getName());
diff --git a/java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITGenerativeModelIntegrationTest.java b/java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITGenerativeModelIntegrationTest.java
@@ -51,9 +51,9 @@
 @RunWith(JUnit4.class)
 public class ITGenerativeModelIntegrationTest {
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
-  private static final String MODEL_NAME_TEXT = "gemini-pro";
-  private static final String MODEL_NAME_MULTIMODAL = "gemini-pro-vision";
-  private static final String MODEL_NAME_LATEST_GEMINI = "gemini-1.5-pro-preview-0409";
+  private static final String MODEL_NAME_TEXT = "gemini-1.5-pro-002";
+  private static final String MODEL_NAME_MULTIMODAL = "gemini-1.5-pro-002";
+  private static final String MODEL_NAME_LATEST_GEMINI = "gemini-1.5-pro-002";
   private static final String LOCATION = "us-central1";
 
   // Tested content
