From 869a37e467873b41ed4c9b50919a303070f4ee29 Mon Sep 17 00:00:00 2001
From: end-4 <97237370+end-4@users.noreply.github.com>
Date: Sat, 23 Mar 2024 23:45:09 +0700
Subject: [PATCH] gemini: show msg block reason

---
 .config/ags/services/gemini.js | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/.config/ags/services/gemini.js b/.config/ags/services/gemini.js
index 3722d0eea..ae7fe249e 100644
--- a/.config/ags/services/gemini.js
+++ b/.config/ags/services/gemini.js
@@ -106,13 +106,17 @@ class GeminiMessage extends Service {
     }
 
     parseSection() {
-        if(this._thinking) {
+        if (this._thinking) {
             this._thinking = false;
-            this._parts[0].text= '';
+            this._parts[0].text = '';
         }
         const parsedData = JSON.parse(this._rawData);
-        const delta = parsedData.candidates[0].content.parts[0].text;
-        this._parts[0].text += delta;
+        if (!parsedData.candidates)
+            this._parts[0].text += `Blocked: ${parsedData.promptFeedback.blockReason}`;
+        else {
+            const delta = parsedData.candidates[0].content.parts[0].text;
+            this._parts[0].text += delta;
+        }
         // this.emit('delta', delta);
         this.notify('content');
         this._rawData = '';