feat: add gemini code sample and test for text-only input (GoogleCloudPlatform#9310)

* feat: add gemini code sample and test for text-only input

* address feedback

* fix test

Author: irataxy

vertexai/snippets/src/main/java/vertexai/gemini/TextInput.java

@@ -0,0 +1,70 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/*
+ * laishers. "A New Era in Cartoon History." Review of Steamboat Willie.
+ * 4 January 2001. https://www.imdb.com/review/rw0005574/?ref_=rw_urv
+ */
+
+package vertexai.gemini;
+
+// [START generativeaionvertexai_gemini_generate_from_text_input]
+import com.google.cloud.vertexai.VertexAI;
+import com.google.cloud.vertexai.api.GenerateContentResponse;
+import com.google.cloud.vertexai.generativeai.GenerativeModel;
+import com.google.cloud.vertexai.generativeai.ResponseHandler;
+import java.io.IOException;
+
+public class TextInput {
+
+  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-1.0-pro-002";
+    // Movie review from https://www.imdb.com/review/rw0005574/?ref_=rw_urv.
+    // Does the returned sentiment score match the reviewer's movie rating?
+    String textPrompt =
+        "Give a score from 1 - 10 to suggest if the following movie review is negative or positive"
+            + " (1 is most negative, 10 is most positive, 5 will be neutral). Include an"
+            + " explanation.\n"
+            + "This era was not just the dawn of sound in cartoons, but of a cartoon character"
+            + " which would go down in history as the world's most famous mouse. Yes, Mickey makes"
+            + " his debut here, in this cheery tale of life on board a steamboat. The animation is"
+            + " good for it's time, and the plot - though a little simple - is quite jolly. A true"
+            + " classic, and if you ever manage to get it on video, you won't regret it.";
+
+    String output = textInput(projectId, location, modelName, textPrompt);
+    System.out.println(output);
+  }
+
+  // Passes the provided text input to the Gemini model and returns the text-only response.
+  // For the specified textPrompt, the model returns a sentiment score for the given movie review.
+  public static String textInput(
+      String projectId, String location, String modelName, String textPrompt) throws IOException {
+    // Initialize client that will be used to send requests. This client only needs
+    // to be created once, and can be reused for multiple requests.
+    try (VertexAI vertexAI = new VertexAI(projectId, location)) {
+      String output;
+      GenerativeModel model = new GenerativeModel(modelName, vertexAI);
+
+      GenerateContentResponse response = model.generateContent(textPrompt);
+      output = ResponseHandler.getText(response);
+      return output;
+    }
+  }
+}
+// [END generativeaionvertexai_gemini_generate_from_text_input]

vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java

@@ -50,10 +50,11 @@ public class SnippetsIT {
   private static final String GEMINI_PRO_1_5 = "gemini-1.5-pro-preview-0409";
   private static final int MAX_ATTEMPT_COUNT = 3;
   private static final int INITIAL_BACKOFF_MILLIS = 120000; // 2 minutes
+
   @Rule
-  public final MultipleAttemptsRule multipleAttemptsRule = new MultipleAttemptsRule(
-      MAX_ATTEMPT_COUNT,
-      INITIAL_BACKOFF_MILLIS);
+  public final MultipleAttemptsRule multipleAttemptsRule =
+      new MultipleAttemptsRule(MAX_ATTEMPT_COUNT, INITIAL_BACKOFF_MILLIS);
+
   private final PrintStream printStream = System.out;
   private ByteArrayOutputStream bout;
 
@@ -145,10 +146,11 @@ public void testMultimodalMultiImage() throws IOException {
 
   @Test
   public void testMultimodalQuery() throws Exception {
-    String imageUri = "https://storage.googleapis.com/cloud-samples-data/vertex-ai/llm/prompts/landmark1.png";
+    String imageUri =
+        "https://storage.googleapis.com/cloud-samples-data/vertex-ai/llm/prompts/landmark1.png";
     String dataImageBase64 = Base64.getEncoder().encodeToString(readImageFile(imageUri));
-    String output = MultimodalQuery.multimodalQuery(PROJECT_ID, LOCATION, GEMINI_PRO_VISION,
-        dataImageBase64);
+    String output =
+        MultimodalQuery.multimodalQuery(PROJECT_ID, LOCATION, GEMINI_PRO_VISION, dataImageBase64);
     assertThat(output).isNotEmpty();
   }
 
@@ -180,26 +182,42 @@ public void testQuickstart() throws IOException {
 
   @Test
   public void testSingleTurnMultimodal() throws IOException {
-    String imageUri = "https://storage.googleapis.com/cloud-samples-data/vertex-ai/llm/prompts/landmark1.png";
+    String imageUri =
+        "https://storage.googleapis.com/cloud-samples-data/vertex-ai/llm/prompts/landmark1.png";
     String dataImageBase64 = Base64.getEncoder().encodeToString(readImageFile(imageUri));
-    SingleTurnMultimodal.generateContent(PROJECT_ID, LOCATION, GEMINI_PRO_VISION,
-        "What is this image", dataImageBase64);
+    SingleTurnMultimodal.generateContent(
+        PROJECT_ID, LOCATION, GEMINI_PRO_VISION, "What is this image", dataImageBase64);
     assertThat(bout.toString()).contains("Colosseum");
   }
 
   @Test
   public void testStreamingQuestions() throws Exception {
-    StreamingQuestionAnswer.streamingQuestion(PROJECT_ID, LOCATION,
-        GEMINI_PRO_VISION);
+    StreamingQuestionAnswer.streamingQuestion(PROJECT_ID, LOCATION, GEMINI_PRO_VISION);
     assertThat(bout.toString()).contains("Rayleigh scattering");
   }
 
+  @Test
+  public void testTextInput() throws Exception {
+    String textPrompt =
+        "Give a score from 1 - 10 to suggest if the following movie review is negative or positive"
+            + " (1 is most negative, 10 is most positive, 5 will be neutral). Include an"
+            + " explanation.\n"
+            + "This era was not just the dawn of sound in cartoons, but of a cartoon character"
+            + " which would go down in history as the world's most famous mouse. Yes, Mickey makes"
+            + " his debut here, in this cheery tale of life on board a steamboat. The animation is"
+            + " good for it's time, and the plot - though a little simple - is quite jolly. A true"
+            + " classic, and if you ever manage to get it on video, you won't regret it.";
+    String output = TextInput.textInput(PROJECT_ID, LOCATION, GEMINI_PRO, textPrompt);
+    assertThat(output).isNotEmpty();
+    assertThat(output).contains("positive");
+  }
+
   @Test
   public void testSafetySettings() throws Exception {
     String textPrompt = "Hello World!";
 
-    String output = WithSafetySettings.safetyCheck(PROJECT_ID, LOCATION, GEMINI_PRO_VISION,
-        textPrompt);
+    String output =
+        WithSafetySettings.safetyCheck(PROJECT_ID, LOCATION, GEMINI_PRO_VISION, textPrompt);
     assertThat(output).isNotEmpty();
     assertThat(output).contains("reasons?");
   }
@@ -208,17 +226,17 @@ public void testSafetySettings() throws Exception {
   public void testTokenCount() throws Exception {
     String textPrompt = "Why is the sky blue?";
 
-    int tokenCount = GetTokenCount.getTokenCount(PROJECT_ID, LOCATION, GEMINI_PRO_VISION,
-        textPrompt);
+    int tokenCount =
+        GetTokenCount.getTokenCount(PROJECT_ID, LOCATION, GEMINI_PRO_VISION, textPrompt);
     assertThat(tokenCount).isEqualTo(6);
   }
 
   @Test
   public void testFunctionCalling() throws Exception {
     String textPrompt = "What's the weather in Paris?";
 
-    String answer = FunctionCalling.whatsTheWeatherLike(PROJECT_ID, LOCATION, GEMINI_PRO,
-        textPrompt);
+    String answer =
+        FunctionCalling.whatsTheWeatherLike(PROJECT_ID, LOCATION, GEMINI_PRO, textPrompt);
     assertThat(answer).ignoringCase().contains("Paris");
     assertThat(answer).ignoringCase().contains("sunny");
   }