fix: rerender chat component

ChatGPTNextWeb · Nov 11, 2024 · 790adb2 · 790adb2
1 parent 38fa305
commit 790adb2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/app/components/chat.tsx b/app/components/chat.tsx
@@ -2071,6 +2071,6 @@ function _Chat() {
 
 export function Chat() {
   const chatStore = useChatStore();
-  const sessionIndex = chatStore.currentSessionIndex;
-  return <_Chat key={sessionIndex}></_Chat>;
+  const forceUpdateKey = chatStore.forceUpdateKey;
+  return <_Chat key={forceUpdateKey}></_Chat>;
 }
diff --git a/app/store/chat.ts b/app/store/chat.ts
@@ -193,6 +193,7 @@ const DEFAULT_CHAT_STATE = {
   sessions: [createEmptySession()],
   currentSessionIndex: 0,
   lastInput: "",
+  forceUpdateKey: Date.now(),
 };
 
 export const useChatStore = createPersistStore(
@@ -225,19 +226,22 @@ export const useChatStore = createPersistStore(
         set((state) => ({
           currentSessionIndex: 0,
           sessions: [newSession, ...state.sessions],
+          forceUpdateKey: Date.now(),
         }));
       },
 
       clearSessions() {
         set(() => ({
           sessions: [createEmptySession()],
           currentSessionIndex: 0,
+          forceUpdateKey: Date.now(),
         }));
       },
 
       selectSession(index: number) {
         set({
           currentSessionIndex: index,
+          forceUpdateKey: Date.now(),
         });
       },
 
@@ -262,6 +266,7 @@ export const useChatStore = createPersistStore(
           return {
             currentSessionIndex: newIndex,
             sessions: newSessions,
+            forceUpdateKey: Date.now(),
           };
         });
       },
@@ -286,6 +291,7 @@ export const useChatStore = createPersistStore(
         set((state) => ({
           currentSessionIndex: 0,
           sessions: [session].concat(state.sessions),
+          forceUpdateKey: Date.now(),
         }));
       },
 
@@ -325,6 +331,7 @@ export const useChatStore = createPersistStore(
         set(() => ({
           currentSessionIndex: nextIndex,
           sessions,
+          forceUpdateKey: Date.now(),
         }));
 
         showToast(