re-add fixed test

Author: David Motsonashvili

firebase-ai/src/test/java/com/google/firebase/ai/DevAPIUnarySnapshotTests.kt

@@ -24,6 +24,7 @@ import com.google.firebase.ai.util.goldenDevAPIUnaryFile
 import io.kotest.assertions.throwables.shouldThrow
 import io.kotest.matchers.should
 import io.kotest.matchers.shouldBe
+import io.kotest.matchers.shouldNotBe
 import io.ktor.http.HttpStatusCode
 import kotlin.time.Duration.Companion.seconds
 import kotlinx.coroutines.withTimeout
@@ -56,6 +57,21 @@ internal class DevAPIUnarySnapshotTests {
       }
     }
 
+  @Test
+  fun `citation returns correctly`() =
+    goldenDevAPIUnaryFile("unary-success-citations.json") {
+      withTimeout(testTimeout) {
+        val response = model.generateContent("prompt")
+
+        response.candidates.isEmpty() shouldBe false
+        response.candidates.first().citationMetadata?.citations?.size shouldBe 4
+        response.candidates.first().citationMetadata?.citations?.forEach {
+          it.startIndex shouldNotBe null
+          it.endIndex shouldNotBe null
+        }
+      }
+    }
+
   @Test
   fun `response blocked for safety`() =
     goldenDevAPIUnaryFile("unary-failure-finish-reason-safety.txt") {

firebase-ai/update_responses.sh

@@ -17,7 +17,7 @@
 # This script replaces mock response files for Vertex AI unit tests with a fresh
 # clone of the shared repository of Vertex AI test data.
 
-RESPONSES_VERSION='v10.*' # The major version of mock responses to use
+RESPONSES_VERSION='v11.*' # The major version of mock responses to use
 REPO_NAME="vertexai-sdk-test-data"
 REPO_LINK="https://github.com/FirebaseExtended/$REPO_NAME.git"