Use gemini-pro instead of gemini-pro-vision in chat (GoogleCloudPlatform#8949)

glaforge · web-flow · commit 12e1b1ff3c93 · 2023-12-20T21:11:11.000+05:30
* Use gemini-pro instead of gemini-pro-vision in chat

* Update chat test case to use gemini-pro
diff --git a/vertexai/snippets/src/main/java/vertexai/gemini/ChatDiscussion.java b/vertexai/snippets/src/main/java/vertexai/gemini/ChatDiscussion.java
@@ -30,7 +30,7 @@ public static void main(String[] args) throws IOException {
     // TODO(developer): Replace these variables before running the sample.
     String projectId = "your-google-cloud-project-id";
     String location = "us-central1";
-    String modelName = "gemini-pro-vision";
+    String modelName = "gemini-pro";
 
     chatDiscussion(projectId, location, modelName);
   }
diff --git a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java
@@ -42,6 +42,7 @@ public class SnippetsIT {
   private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
   private static final String LOCATION = "us-central1";
   private static final String GEMINI_PRO_VISION = "gemini-pro-vision";
+  private static final String GEMINI_PRO = "gemini-pro";
   private static final int MAX_ATTEMPT_COUNT = 3;
   private static final int INITIAL_BACKOFF_MILLIS = 120000; // 2 minutes
   @Rule
@@ -113,7 +114,7 @@ public void afterEach() {
 
   @Test
   public void testChatSession() throws IOException {
-    ChatDiscussion.chatDiscussion(PROJECT_ID, LOCATION, GEMINI_PRO_VISION);
+    ChatDiscussion.chatDiscussion(PROJECT_ID, LOCATION, GEMINI_PRO);
     assertThat(out.toString()).contains("Chat Ended.");
   }
 

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ public static void main(String[] args) throws IOException {`
`30`	`30`	`// TODO(developer): Replace these variables before running the sample.`
`31`	`31`	`String projectId = "your-google-cloud-project-id";`
`32`	`32`	`String location = "us-central1";`
`33`		`- String modelName = "gemini-pro-vision";`
	`33`	`+ String modelName = "gemini-pro";`
`34`	`34`
`35`	`35`	`chatDiscussion(projectId, location, modelName);`
`36`	`36`	`}`