Skip to content

Commit 8b8d3fa

Browse files
test: Update java-vertexai IT models (Gemini 1.x) (#11550)
Fixes #11542 Includes reformat using the new com.spotify.fmt:fmt-maven-plugin plugin PiperOrigin-RevId: 756608894 Co-authored-by: A Vertex SDK engineer <[email protected]>
1 parent 6baae9d commit 8b8d3fa

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITChatSessionIntegrationTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
@RunWith(JUnit4.class)
4747
public class ITChatSessionIntegrationTest {
4848
private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
49-
private static final String MODEL_NAME_TEXT = "gemini-pro";
49+
private static final String MODEL_NAME_TEXT = "gemini-1.5-pro-002";
5050
private static final String LOCATION = "us-central1";
5151
private static final Logger logger =
5252
Logger.getLogger(ITGenerativeModelIntegrationTest.class.getName());

java-vertexai/google-cloud-vertexai/src/test/java/com/google/cloud/vertexai/it/ITGenerativeModelIntegrationTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,9 @@
5151
@RunWith(JUnit4.class)
5252
public class ITGenerativeModelIntegrationTest {
5353
private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
54-
private static final String MODEL_NAME_TEXT = "gemini-pro";
55-
private static final String MODEL_NAME_MULTIMODAL = "gemini-pro-vision";
56-
private static final String MODEL_NAME_LATEST_GEMINI = "gemini-1.5-pro-preview-0409";
54+
private static final String MODEL_NAME_TEXT = "gemini-1.5-pro-002";
55+
private static final String MODEL_NAME_MULTIMODAL = "gemini-1.5-pro-002";
56+
private static final String MODEL_NAME_LATEST_GEMINI = "gemini-1.5-pro-002";
5757
private static final String LOCATION = "us-central1";
5858

5959
// Tested content

0 commit comments

Comments
 (0)