Merge d3c70e6 into 197b090

Author: rlazo

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

@@ -57,24 +57,13 @@ internal class DevAPIUnarySnapshotTests {
       }
     }
 
-  @Test
-  fun `prompt blocked for safety`() =
-    goldenDevAPIUnaryFile("unary-failure-prompt-blocked-safety.txt") {
-      withTimeout(testTimeout) {
-        shouldThrow<ResponseStoppedException> { model.generateContent("prompt") } should
-          {
-            it.response.candidates[0].finishReason shouldBe FinishReason.MAX_TOKENS
-          }
-      }
-    }
-
   @Test
   fun `response blocked for safety`() =
     goldenDevAPIUnaryFile("unary-failure-finish-reason-safety.txt") {
       withTimeout(testTimeout) {
         shouldThrow<ResponseStoppedException> { model.generateContent("prompt") } should
           {
-            it.response.candidates[0].finishReason shouldBe FinishReason.MAX_TOKENS
+            it.response.candidates[0].finishReason shouldBe FinishReason.SAFETY
           }
       }
     }

firebase-ai/src/test/java/com/google/firebase/ai/util/tests.kt

@@ -248,7 +248,7 @@ internal fun goldenDevAPIUnaryFile(
   name: String,
   httpStatusCode: HttpStatusCode = HttpStatusCode.OK,
   block: CommonTest,
-) = goldenUnaryFile("developerapi/$name", httpStatusCode, GenerativeBackend.googleAI(), block)
+) = goldenUnaryFile("googleai/$name", httpStatusCode, GenerativeBackend.googleAI(), block)
 
 /**
  * Loads a *Golden File* from the resource directory.

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='v7.*' # The major version of mock responses to use
+RESPONSES_VERSION='v10.*' # The major version of mock responses to use
 REPO_NAME="vertexai-sdk-test-data"
 REPO_LINK="https://github.com/FirebaseExtended/$REPO_NAME.git"