Merge branch 'main' into daymon-update-sample-app

google-gemini · Jul 3, 2024 · a7fbf59 · a7fbf59
2 parents 274513b + 13b4d44
commit a7fbf59
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt b/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt
@@ -80,9 +80,9 @@ internal class UnarySnapshotTests {
       withTimeout(testTimeout) {
         val response = apiController.generateContent(textGenerateContentRequest("prompt"))
 
-        response.candidates?.first {
-          it.safetyRatings?.any { it.category == HarmCategory.UNKNOWN } ?: false
-        }
+        response.candidates?.isNullOrEmpty() shouldBe false
+        val candidate = response.candidates?.first()
+        candidate?.safetyRatings?.any { it.category == HarmCategory.UNKNOWN } shouldBe true
       }
     }