gemini: show msg block reason

This commit is contained in:
end-4
2024-03-23 23:45:09 +07:00
parent 842159da5a
commit 0555c945dd
+8 -4
View File
@@ -106,13 +106,17 @@ class GeminiMessage extends Service {
} }
parseSection() { parseSection() {
if(this._thinking) { if (this._thinking) {
this._thinking = false; this._thinking = false;
this._parts[0].text= ''; this._parts[0].text = '';
} }
const parsedData = JSON.parse(this._rawData); const parsedData = JSON.parse(this._rawData);
const delta = parsedData.candidates[0].content.parts[0].text; if (!parsedData.candidates)
this._parts[0].text += delta; 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.emit('delta', delta);
this.notify('content'); this.notify('content');
this._rawData = ''; this._rawData = '';