🐛 fix: Removing message from chat after complete receiving

Palm1r · Nov 25, 2024 · 36d5242 · 36d5242
1 parent 6503887
commit 36d5242
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/ChatView/ClientInterface.cpp b/ChatView/ClientInterface.cpp
@@ -148,11 +148,16 @@ void ClientInterface::handleLLMResponse(const QString &response,
                                         const QJsonObject &request,
                                         bool isComplete)
 {
-    QString messageId = request["id"].toString();
-    m_chatModel->addMessage(response.trimmed(), ChatModel::ChatRole::Assistant, messageId);
+    const auto message = response.trimmed();
 
-    if (isComplete) {
-        LOG_MESSAGE("Message completed. Final response for message " + messageId + ": " + response);
+    if (!message.isEmpty()) {
+        QString messageId = request["id"].toString();
+        m_chatModel->addMessage(message, ChatModel::ChatRole::Assistant, messageId);
+
+        if (isComplete) {
+            LOG_MESSAGE(
+                "Message completed. Final response for message " + messageId + ": " + response);
+        }
     }
 }