Run parsing at least once before rendering reasoning DOM on continue

c03da65821d729014858a4478b1f1e290fd643b3

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/script.js+13 -4 Viewed

         this.promptReasoning = promptReasoning;
+    }
-    #checkDomElements(messageId) {
+    /**
+     * Initializes DOM elements for the current message.
+     * @param {number} messageId Current message ID
+     * @param {boolean?} continueOnReasoning If continuing on reasoning
+     */
+    async #checkDomElements(messageId, continueOnReasoning = null) {
         if (this.messageDom === null || this.messageTextDom === null) {
             this.messageDom = document.querySelector(`#chat .mes[mesid="${messageId}"]`);
             this.messageTextDom = this.messageDom?.querySelector('.mes_text');
             this.messageTimerDom = this.messageDom?.querySelector('.mes_timer');
             this.messageTokenCounterDom = this.messageDom?.querySelector('.tokenCounterDisplay');
+        }
+        if (continueOnReasoning) {
+            await this.reasoningHandler.process(messageId, false, this.promptReasoning);
+        }
         this.reasoningHandler.updateDom(messageId);
+    }

@@ -3230,7 +3238,8 @@ class StreamingProcessor {
3230	}	3238	}
3231		3239
3232	async onStartStreaming(text) {	3240	async onStartStreaming(text) {
3233	if (this.type === 'continue' && this.promptReasoning.prefixReasoning) {	3241	const continueOnReasoning = !!(this.type === 'continue' && this.promptReasoning.prefixReasoning);
		3242	if (continueOnReasoning) {
3234	this.reasoningHandler.initContinue(this.promptReasoning);	3243	this.reasoningHandler.initContinue(this.promptReasoning);
3235	}	3244	}
3236		3245

         } else {
             await saveReply(this.type, text, true, '', [], '');
             messageId = chat.length - 1;
-            this.#checkDomElements(messageId);
+            await this.#checkDomElements(messageId, continueOnReasoning);
             this.markUIGenStarted();
+        }
         hideSwipeButtons();

             this.sendTextarea.dispatchEvent(new Event('input', { bubbles: true }));
         } else {
             const mesChanged = chat[messageId]['mes'] !== processedText;
-            this.#checkDomElements(messageId);
+            await this.#checkDomElements(messageId);
             this.#updateMessageBlockVisibility();
             const currentTime = new Date();
             chat[messageId]['mes'] = processedText;