From e636d486f5b08fd14473f31b6c9e33ad92fe723a Mon Sep 17 00:00:00 2001
From: Jiacheng Dong <43509711+Jiacheng787@users.noreply.github.com>
Date: Thu, 15 Jun 2023 10:23:01 +0800
Subject: [PATCH] fix: updating the array using push in zustand does not
 actually trigger component updates

---
 app/store/chat.ts | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/app/store/chat.ts b/app/store/chat.ts
index f56d9b9d..455399b8 100644
--- a/app/store/chat.ts
+++ b/app/store/chat.ts
@@ -227,6 +227,7 @@ export const useChatStore = create<ChatStore>()(
 
       onNewMessage(message) {
         get().updateCurrentSession((session) => {
+          session.messages = session.messages.concat();
           session.lastUpdate = Date.now();
         });
         get().updateStat(message);
@@ -273,8 +274,9 @@ export const useChatStore = create<ChatStore>()(
 
         // save user's and bot's message
         get().updateCurrentSession((session) => {
-          session.messages.push(userMessage);
-          session.messages.push(botMessage);
+          // session.messages.push(userMessage);
+          // session.messages.push(botMessage);
+          session.messages = session.messages.concat([userMessage, botMessage]);
         });
 
         // make request
@@ -287,7 +289,10 @@ export const useChatStore = create<ChatStore>()(
             if (message) {
               botMessage.content = message;
             }
-            set(() => ({}));
+            // set(() => ({}));
+            get().updateCurrentSession((session) => {
+              session.messages = session.messages.concat();
+            });
           },
           onFinish(message) {
             botMessage.streaming = false;
@@ -299,7 +304,7 @@ export const useChatStore = create<ChatStore>()(
               sessionIndex,
               botMessage.id ?? messageIndex,
             );
-            set(() => ({}));
+            // set(() => ({}));
           },
           onError(error) {
             const isAborted = error.message.includes("aborted");