From 900c0caffa947ea0b7a06e12affb73873ae6cd92 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sun, 2 Feb 2025 18:02:23 +0200
Subject: [PATCH] Fix class set condition

---
 public/script.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/public/script.js b/public/script.js
index b12d2d3e0..fa7ca9d1a 100644
--- a/public/script.js
+++ b/public/script.js
@@ -3266,9 +3266,9 @@ class StreamingProcessor {
                 if (this.messageReasoningDom instanceof HTMLElement) {
                     const formattedReasoning = messageFormatting(this.reasoning, '', false, false, messageId, {}, true);
                     this.messageReasoningDom.innerHTML = formattedReasoning;
-                    if (this.reasoning) {
-                        this.messageDom.classList.add('reasoning');
-                    }
+                }
+                if (this.messageDom instanceof HTMLElement) {
+                    this.messageDom.classList.add('reasoning');
                 }
             }