fix: checking for index of of current message instead of checking for…

… user messages amount
smallcloudai · Jan 28, 2025 · fc2291b · fc2291b
1 parent c002c2f
commit fc2291b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/hooks/useCheckpoints.ts b/src/hooks/useCheckpoints.ts
@@ -80,6 +80,7 @@ export const useCheckpoints = () => {
     async (checkpoints: Checkpoint[] | null, messageIndex: number) => {
       if (!checkpoints) return;
       const amountOfUserMessages = messages.filter(isUserMessage);
+      const firstUserMessage = amountOfUserMessages[0];
 
       const restoredChanges =
         await restoreChangesFromCheckpoints(checkpoints).unwrap();
@@ -88,7 +89,9 @@ export const useCheckpoints = () => {
         dispatch(setIsUndoingCheckpoints(false)),
         setLatestCheckpointResult({ ...restoredChanges, messageIndex }),
         setIsCheckpointsPopupIsVisible(true),
-        setShouldNewChatBeStarted(amountOfUserMessages.length === 1),
+        setShouldNewChatBeStarted(
+          messageIndex === messages.indexOf(firstUserMessage),
+        ),
       ];
       actions.forEach((action) => dispatch(action));
     },