chore: Deflake GenAI system test test_send_message_from_text

matthew29tang · copybara-github · commit c8be56a1cb27 · 2024-06-25T16:29:19.000-07:00
PiperOrigin-RevId: 646645371
diff --git a/tests/system/vertexai/test_generative_models.py b/tests/system/vertexai/test_generative_models.py
@@ -313,7 +313,7 @@ def test_send_message_from_text(self):
         model = generative_models.GenerativeModel(GEMINI_MODEL_NAME)
         chat = model.start_chat()
         response1 = chat.send_message(
-            "I really like fantasy books.",
+            "I really like fantasy movies.",
             generation_config=generative_models.GenerationConfig(temperature=0),
         )
         assert response1.text

Original file line number	Diff line number	Diff line change
`@@ -313,7 +313,7 @@ def test_send_message_from_text(self):`
`313`	`313`	`model = generative_models.GenerativeModel(GEMINI_MODEL_NAME)`
`314`	`314`	`chat = model.start_chat()`
`315`	`315`	`response1 = chat.send_message(`
`316`		`- "I really like fantasy books.",`
	`316`	`+ "I really like fantasy movies.",`
`317`	`317`	`generation_config=generative_models.GenerationConfig(temperature=0),`
`318`	`318`	`)`
`319`	`319`	`assert response1.text`