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	3240	async onStartStreaming(text) {
3233	3241	ifconst continueOnReasoning = !!(this.type === 'continue' && this.promptReasoning.prefixReasoning) {;
	3242	+ if (continueOnReasoning) {
3234	3243	this.reasoningHandler.initContinue(this.promptReasoning);
3235	3244	}
3236	3245

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

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