From 8d708a3b7077f733f9e749dc60b1066fcf246f1a Mon Sep 17 00:00:00 2001 From: Will Tang Date: Fri, 20 Mar 2026 13:58:11 -0700 Subject: [PATCH 1/2] Add prompt feedback check for candidates blocked due to safety. --- agent.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/agent.py b/agent.py index f650381..aa07d32 100644 --- a/agent.py +++ b/agent.py @@ -264,6 +264,9 @@ def run_one_iteration(self) -> Literal["COMPLETE", "CONTINUE"]: return "COMPLETE" if not response.candidates: + if response.prompt_feedback and response.prompt_feedback.block_reason.name == 'SAFETY': + print(response) + raise ValueError("Response was blocked due to safety.") print("Response has no candidates!") print(response) raise ValueError("Empty response") From f61a40dffaf129b109d99a40d06d2e92fe29be60 Mon Sep 17 00:00:00 2001 From: wt17 <90870042+wt17@users.noreply.github.com> Date: Fri, 20 Mar 2026 17:31:11 -0700 Subject: [PATCH 2/2] Update agent.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- agent.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/agent.py b/agent.py index aa07d32..d263e3a 100644 --- a/agent.py +++ b/agent.py @@ -264,9 +264,8 @@ def run_one_iteration(self) -> Literal["COMPLETE", "CONTINUE"]: return "COMPLETE" if not response.candidates: - if response.prompt_feedback and response.prompt_feedback.block_reason.name == 'SAFETY': - print(response) - raise ValueError("Response was blocked due to safety.") + if response.prompt_feedback and response.prompt_feedback.block_reason == types.BlockReason.SAFETY: + raise ValueError(f"Response was blocked due to safety. Feedback: {response.prompt_feedback}") print("Response has no candidates!") print(response) raise ValueError("Empty response")