feat: GenAI - Added the GenerationResponse.prompt_feedback property

PiperOrigin-RevId: 625506526
googleapis · Apr 17, 2024 · efd5a72 · efd5a72
1 parent 423c764
commit efd5a72
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tests/unit/vertexai/test_generative_models.py b/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
 

diff --git a/vertexai/generative_models/_generative_models.py b/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."""