fixed prompt and test (#9430)

GoogleCloudPlatform · Aug 1, 2024 · 19adb6f · 19adb6f
1 parent 2790431
commit 19adb6f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vertexai/snippets/src/main/java/vertexai/gemini/StreamingQuestionAnswer.java b/vertexai/snippets/src/main/java/vertexai/gemini/StreamingQuestionAnswer.java
@@ -40,7 +40,7 @@ public static void streamingQuestion(String projectId, String location, String m
       GenerativeModel model = new GenerativeModel(modelName, vertexAI);
 
       // Stream the result.
-      model.generateContentStream("Why is the sky blue?")
+      model.generateContentStream("Write a story about a magic backpack.")
           .stream()
           .forEach(System.out::println);
 

diff --git a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java
@@ -190,7 +190,7 @@ public void testSingleTurnMultimodal() throws IOException {
   @Test
   public void testStreamingQuestions() throws Exception {
     StreamingQuestionAnswer.streamingQuestion(PROJECT_ID, LOCATION, GEMINI_FLASH);
-    assertThat(bout.toString()).contains("Rayleigh scattering");
+    assertThat(bout.toString()).isNotEmpty();
   }
 
   @Test