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 = '';