fix: message state

Libertai · Sep 16, 2024 · e6981ed · e6981ed
1 parent f62b441
commit e6981ed
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/pages/Chat.vue b/src/pages/Chat.vue
@@ -267,13 +267,15 @@ async function generatePersonaMessage() {
     )) {
       const stopped = output.stopped || shouldStopGeneration.value;
       let content = output.content;
+
+      // Update the local state include updates
+      response.content = content;
+      response.stopped = stopped;
+
       if (stopped) {
         shouldStopGeneration.value = false;
         break;
       }
-      // Update the local state include updates
-      response.content = content;
-      response.stopped = stopped;
 
       chatRef.value.messages = [...chatRef.value.messages];
       // Scroll to the bottom of the chat