fix: update conditional check last status message (janhq#1951)

* fix: update conditional check last status message * fix: no new message on failed message --------- Co-authored-by: Louis <[email protected]>
jesst3r · Feb 7, 2024 · 9a1b1ad · 9a1b1ad
1 parent 5890ade
commit 9a1b1ad
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 11 deletions.
diff --git a/web/containers/Providers/EventHandler.tsx b/web/containers/Providers/EventHandler.tsx
@@ -128,10 +128,10 @@ export default function EventHandler({ children }: { children: ReactNode }) {
 
       const thread = threadsRef.current?.find((e) => e.id == message.thread_id)
       if (thread) {
-        const messageContent = message.content[0]?.text.value ?? ''
+        const messageContent = message.content[0]?.text?.value
         const metadata = {
           ...thread.metadata,
-          lastMessage: messageContent,
+          ...(messageContent && { lastMessage: messageContent }),
         }
 
         updateThread({
@@ -151,7 +151,7 @@ export default function EventHandler({ children }: { children: ReactNode }) {
           ?.addNewMessage(message)
       }
     },
-    [updateMessage, updateThreadWaiting, setIsGeneratingResponse]
+    [updateMessage, updateThreadWaiting, setIsGeneratingResponse, updateThread]
   )
 
   useEffect(() => {

diff --git a/web/helpers/atoms/ChatMessage.atom.ts b/web/helpers/atoms/ChatMessage.atom.ts
@@ -70,11 +70,12 @@ export const addNewMessageAtom = atom(
     set(chatMessages, newData)
 
     // Update thread last message
-    set(
-      updateThreadStateLastMessageAtom,
-      newMessage.thread_id,
-      newMessage.content
-    )
+    if (newMessage.content.length)
+      set(
+        updateThreadStateLastMessageAtom,
+        newMessage.thread_id,
+        newMessage.content
+      )
   }
 )
 
@@ -131,7 +132,8 @@ export const updateMessageAtom = atom(
       newData[conversationId] = updatedMessages
       set(chatMessages, newData)
       // Update thread last message
-      set(updateThreadStateLastMessageAtom, conversationId, text)
+      if (text.length)
+        set(updateThreadStateLastMessageAtom, conversationId, text)
     }
   }
 )
diff --git a/web/hooks/useCreateNewThread.ts b/web/hooks/useCreateNewThread.ts
@@ -6,6 +6,7 @@ import {
   ThreadAssistantInfo,
   ThreadState,
   Model,
+  MessageStatus,
 } from '@janhq/core'
 import { atom, useAtomValue, useSetAtom } from 'jotai'
 
@@ -20,6 +21,7 @@ import useSetActiveThread from './useSetActiveThread'
 
 import { extensionManager } from '@/extension'
 
+import { getCurrentChatMessagesAtom } from '@/helpers/atoms/ChatMessage.atom'
 import {
   threadsAtom,
   threadStatesAtom,
@@ -51,6 +53,7 @@ export const useCreateNewThread = () => {
   const setFileUpload = useSetAtom(fileUploadAtom)
   const setSelectedModel = useSetAtom(selectedModelAtom)
   const setThreadModelParams = useSetAtom(setThreadModelParamsAtom)
+  const messages = useAtomValue(getCurrentChatMessagesAtom)
 
   const { recommendedModel, downloadedModels } = useRecommendedModel()
 
@@ -63,9 +66,9 @@ export const useCreateNewThread = () => {
     const defaultModel = model ?? recommendedModel ?? downloadedModels[0]
 
     // check last thread message, if there empty last message use can not create thread
-    const lastMessage = threads[0]?.metadata?.lastMessage
+    const lastMessage = threads[threads.length - 1]?.metadata?.lastMessage
 
-    if (!lastMessage && threads.length) {
+    if (!lastMessage && threads.length && !messages.length) {
       return null
     }