feat: GenAI - Added the GenerationResponse.prompt_feedback property

PiperOrigin-RevId: 625506526

Author: Ark-kun

tests/unit/vertexai/test_generative_models.py

@@ -500,6 +500,8 @@ def test_chat_send_message_response_blocked_errors(
             chat.send_message("Please block with block_reason=OTHER.")
 
         assert e.match("Blocked for testing")
+        # Checking that the original response and the block reason are accessible.
+        assert e.value.responses[0].prompt_feedback.block_reason.name == "OTHER"
         # Checking that history did not get updated
         assert len(chat.history) == 2
 

vertexai/generative_models/_generative_models.py

@@ -1642,6 +1642,12 @@ def text(self) -> str:
             raise ValueError("Response has no candidates (and no text).")
         return self.candidates[0].text
 
+    @property
+    def prompt_feedback(
+        self,
+    ) -> gapic_prediction_service_types.GenerateContentResponse.PromptFeedback:
+        return self._raw_response.prompt_feedback
+
 
 class Candidate:
     """A response candidate generated by the model."""