Merge pull request #110 from nahoj/patch-1

Fix streaming prompt-feedback handling

Author: owndev

pipelines/google/google_gemini.py

@@ -4,7 +4,7 @@
 author_url: https://github.com/owndev/
 project_url: https://github.com/owndev/Open-WebUI-Functions
 funding_url: https://github.com/sponsors/owndev
-version: 1.8.1
+version: 1.8.2
 required_open_webui_version: 0.6.26
 license: Apache License 2.0
 description: Highly optimized Google Gemini pipeline with advanced image generation capabilities, intelligent compression, and streamlined processing workflows.
@@ -1854,13 +1854,8 @@ async def emit_chat_event(event_type: str, data: Dict[str, Any]) -> None:
                 # Check for safety feedback or empty chunks
                 if not chunk.candidates:
                     # Check prompt feedback
-                    if (
-                        response_iterator.prompt_feedback
-                        and response_iterator.prompt_feedback.block_reason
-                    ):
-                        block_reason = (
-                            response_iterator.prompt_feedback.block_reason.name
-                        )
+                    if chunk.prompt_feedback and chunk.prompt_feedback.block_reason:
+                        block_reason = chunk.prompt_feedback.block_reason.name
                         message = f"[Blocked due to Prompt Safety: {block_reason}]"
                         await emit_chat_event(
                             "chat:finish",