Add blockReasonMessage to PromptFeedback

Author: rlazo

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

@@ -123,6 +123,18 @@ internal class StreamingSnapshotTests {
       }
     }
 
+  @Test
+  fun `prompt blocked for safety with message`() =
+    goldenStreamingFile("streaming-failure-prompt-blocked-safety-with-message.txt") {
+      val responses = model.generateContentStream("prompt")
+
+      withTimeout(testTimeout) {
+        val exception = shouldThrow<PromptBlockedException> { responses.collect() }
+        exception.response.promptFeedback?.blockReason shouldBe BlockReason.SAFETY
+        exception.response.promptFeedback?.blockReasonMessage shouldBe "Reasons"
+      }
+    }
+
   @Test
   fun `empty content`() =
     goldenStreamingFile("streaming-failure-empty-content.txt") {

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

@@ -167,6 +167,18 @@ internal class UnarySnapshotTests {
       }
     }
 
+  @Test
+  fun `prompt blocked for safety with message`() =
+    goldenUnaryFile("unary-failure-prompt-blocked-safety-with-message.json") {
+      withTimeout(testTimeout) {
+        shouldThrow<PromptBlockedException> { model.generateContent("prompt") } should
+          {
+            it.response.promptFeedback?.blockReason shouldBe BlockReason.SAFETY
+            it.response.promptFeedback?.blockReasonMessage shouldContain "Reasons"
+          }
+      }
+    }
+
   @Test
   fun `empty content`() =
     goldenUnaryFile("unary-failure-empty-content.json") {