diff --git a/adapters/__pycache__/google_adapter.cpython-313.pyc b/adapters/__pycache__/google_adapter.cpython-313.pyc index 335fd48..5179ccb 100644 Binary files a/adapters/__pycache__/google_adapter.cpython-313.pyc and b/adapters/__pycache__/google_adapter.cpython-313.pyc differ diff --git a/adapters/google_adapter.py b/adapters/google_adapter.py index 4245c87..2b48d8f 100644 --- a/adapters/google_adapter.py +++ b/adapters/google_adapter.py @@ -105,9 +105,21 @@ class GoogleAdapter: if response.usage_metadata: token_usage = response.usage_metadata.total_token_count - - if response.parts is None and response.candidates[0].finish_reason is not None: - raise GoogleGenerationException(f"Generation blocked in cause of {response.candidates[0].finish_reason.value}") + + # Check prompt-level block (e.g. PROHIBITED_CONTENT) — no candidates in this case + if response.prompt_feedback and response.prompt_feedback.block_reason: + raise GoogleGenerationException( + f"Generation blocked at prompt level: {response.prompt_feedback.block_reason.value}" + ) + + # Check candidate-level block + if response.parts is None: + response_reason = ( + response.candidates[0].finish_reason + if response.candidates and len(response.candidates) > 0 + else "Unknown" + ) + raise GoogleGenerationException(f"Generation blocked: {response_reason}") generated_images = [] diff --git a/api/endpoints/__pycache__/assets_router.cpython-313.pyc b/api/endpoints/__pycache__/assets_router.cpython-313.pyc index cdc6a92..bd295db 100644 Binary files a/api/endpoints/__pycache__/assets_router.cpython-313.pyc and b/api/endpoints/__pycache__/assets_router.cpython-313.pyc differ