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()( 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()( // 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()( 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()( sessionIndex, botMessage.id ?? messageIndex, ); - set(() => ({})); + // set(() => ({})); }, onError(error) { const isAborted = error.message.includes("aborted");