🗨️ fix: Prevent Resetting Title to 'New Chat' on Follow-Up Message (#1870)

* fix: prevent reseting title to 'New Chat' on follow up message

* chore(useSSE): remove empty line

client/src/hooks/SSE/useSSE.ts

@@ -24,7 +24,12 @@ import type {
   TSubmission,
   ConversationData,
 } from 'librechat-data-provider';
-import { addConversation, deleteConversation, updateConversation } from '~/utils';
+import {
+  addConversation,
+  deleteConversation,
+  updateConversation,
+  getConversationById,
+} from '~/utils';
 import { useGenTitleMutation } from '~/data-provider';
 import useContentHandler from './useContentHandler';
 import { useAuthContext } from '../AuthContext';
@@ -231,13 +236,21 @@ export default function useSSE(submission: TSubmission | null, index = 0) {
       ]);
     }
 
-    const { conversationId } = message;
+    const { conversationId, parentMessageId } = message;
 
     let update = {} as TConversation;
     setConversation((prevState) => {
+      let title = prevState?.title;
+      if (parentMessageId !== Constants.NO_PARENT && title?.toLowerCase()?.includes('new chat')) {
+        const convos = queryClient.getQueryData<ConversationData>([QueryKeys.allConversations]);
+        const cachedConvo = getConversationById(convos, conversationId);
+        title = cachedConvo?.title;
+      }
+
       update = tConvoUpdateSchema.parse({
         ...prevState,
         conversationId,
+        title,
       }) as TConversation;
 
       setStorage(update);
@@ -248,7 +261,7 @@ export default function useSSE(submission: TSubmission | null, index = 0) {
       if (!convoData) {
         return convoData;
       }
-      if (message.parentMessageId === Constants.NO_PARENT) {
+      if (parentMessageId === Constants.NO_PARENT) {
         return addConversation(convoData, update);
       } else {
         return updateConversation(convoData, update);

client/src/utils/convos.ts

@@ -3,7 +3,6 @@ import {
   isToday,
   isWithinInterval,
   subDays,
-  getMonth,
   getYear,
   startOfDay,
   startOfYear,
@@ -162,3 +161,20 @@ export const deleteConversation = (
 
   return newData;
 };
+
+export const getConversationById = (
+  data: ConversationData | undefined,
+  conversationId: string | null,
+): TConversation | undefined => {
+  if (!data || !conversationId) {
+    return undefined;
+  }
+
+  for (const page of data.pages) {
+    const conversation = page.conversations.find((c) => c.conversationId === conversationId);
+    if (conversation) {
+      return conversation;
+    }
+  }
+  return undefined;
+};