fix: performance issue with atom storage persistence

web/helpers/atoms/ChatMessage.atom.ts

@@ -19,11 +19,20 @@
 /**
  * Stores all chat messages for all threads
  */
-export const chatMessages = atomWithStorage<Record<string, ThreadMessage[]>>(
-  CHAT_MESSAGE_NAME,
-  {},
-  undefined,
-  { getOnInit: true }
+export const chatMessagesStorage = atomWithStorage<
+  Record<string, ThreadMessage[]>
+>(CHAT_MESSAGE_NAME, {}, undefined, { getOnInit: true })
+
+export const cachedMessages = atom<Record<string, ThreadMessage[]>>()
+/**
+ * Retrieve chat messages for all threads
+ */
+export const chatMessages = atom(
+  (get) => get(cachedMessages) ?? get(chatMessagesStorage),
+  (_get, set, newValue: Record<string, ThreadMessage[]>) => {
+    set(cachedMessages, newValue)
+    ;(() => set(chatMessagesStorage, newValue))()
+  }
 )
 
 /**
@@ -45,7 +54,7 @@
   if (!activeThreadId) return []
   const messages = get(chatMessages)[activeThreadId]
   if (!Array.isArray(messages)) return []
   return messages ?? []
 })
 
 export const setConvoMessagesAtom = atom(
@@ -69,31 +78,31 @@
 export const addOldMessagesAtom = atom(
   null,
   (get, set, newMessages: ThreadMessage[]) => {
     const currentConvoId = get(getActiveThreadIdAtom)
     if (!currentConvoId) return
 
     const currentMessages = get(chatMessages)[currentConvoId] ?? []
     const updatedMessages = [...currentMessages, ...newMessages]
 
     const newData: Record<string, ThreadMessage[]> = {
       ...get(chatMessages),
     }
     newData[currentConvoId] = updatedMessages
     set(chatMessages, newData)
   }
 )
 
 export const addNewMessageAtom = atom(
   null,
   (get, set, newMessage: ThreadMessage) => {
     const currentMessages = get(chatMessages)[newMessage.thread_id] ?? []
     const updatedMessages = [...currentMessages, newMessage]
 
     const newData: Record<string, ThreadMessage[]> = {
       ...get(chatMessages),
     }
     newData[newMessage.thread_id] = updatedMessages
     set(chatMessages, newData)
 
     // Update thread last message
     if (newMessage.content.length)