fixes for tests

Author: David Motsonashvili

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/ImagenModel.kt

@@ -72,7 +72,6 @@ internal constructor(
    */
   public suspend fun generateImages(prompt: String): ImagenGenerationResponse<ImagenInlineImage> =
     try {
-      controller.generateImage(constructRequest(prompt, null, generationConfig)).toPublicInline()
       controller
         .generateImage(constructRequest(prompt, null, generationConfig))
         .validate()

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/APIController.kt

@@ -60,12 +60,15 @@ import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.tasks.await
 import kotlinx.coroutines.withTimeout
+import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.json.Json
 
+@OptIn(ExperimentalSerializationApi::class)
 internal val JSON = Json {
   ignoreUnknownKeys = true
   prettyPrint = false
   isLenient = true
+  explicitNulls = false
 }
 
 /**

firebase-vertexai/src/test/java/com/google/firebase/vertexai/UnarySnapshotTests.kt

@@ -503,6 +503,14 @@ internal class UnarySnapshotTests {
       }
     }
 
+  @Test
+  fun `generateImages should return when some images are filtered -- base64`() =
+    goldenUnaryFile("unary-failure-generate-images-base64-some-filtered.json") {
+      withTimeout(testTimeout) {
+        imagenModel.generateImages("prompt").images.isEmpty() shouldBe false
+      }
+    }
+
   @Test
   fun `generateImages should throw when prompt blocked`() =
     goldenUnaryFile(