From 9579e8647b125d021844be8bc871c55f5a510bb0 Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Wed, 15 Jan 2025 19:26:54 +0700
Subject: [PATCH] fix: empty response on some generation attempts (#4454)

---
 web/hooks/useSendChatMessage.ts | 42 ++++++++++++++++-----------------
 1 file changed, 21 insertions(+), 21 deletions(-)

diff --git a/web/hooks/useSendChatMessage.ts b/web/hooks/useSendChatMessage.ts
index 87a8414a2..d3c8ff142 100644
--- a/web/hooks/useSendChatMessage.ts
+++ b/web/hooks/useSendChatMessage.ts
@@ -110,7 +110,7 @@ export default function useSendChatMessage() {
     }
 
     if (toSendMessage?.content[0]?.text?.value)
-      sendChatMessage(toSendMessage.content[0].text.value, true)
+      sendChatMessage(toSendMessage.content[0].text.value, true, newConvoData)
   }
 
   const sendChatMessage = async (
@@ -176,27 +176,27 @@ export default function useSendChatMessage() {
       messages ?? currentMessages
     ).addSystemMessage(activeAssistantRef.current?.instructions)
 
+    requestBuilder.pushMessage(prompt, base64Blob, fileUpload)
+
+    // Build Thread Message to persist
+    const threadMessageBuilder = new ThreadMessageBuilder(
+      requestBuilder
+    ).pushMessage(prompt, base64Blob, fileUpload)
+
+    const newMessage = threadMessageBuilder.build()
+
+    // Update thread state
+    const updatedThread: Thread = {
+      ...activeThreadRef.current,
+      updated: newMessage.created_at,
+      metadata: {
+        ...activeThreadRef.current.metadata,
+        lastMessage: prompt,
+      },
+    }
+    updateThread(updatedThread)
+
     if (!isResend) {
-      requestBuilder.pushMessage(prompt, base64Blob, fileUpload)
-
-      // Build Thread Message to persist
-      const threadMessageBuilder = new ThreadMessageBuilder(
-        requestBuilder
-      ).pushMessage(prompt, base64Blob, fileUpload)
-
-      const newMessage = threadMessageBuilder.build()
-
-      // Update thread state
-      const updatedThread: Thread = {
-        ...activeThreadRef.current,
-        updated: newMessage.created_at,
-        metadata: {
-          ...activeThreadRef.current.metadata,
-          lastMessage: prompt,
-        },
-      }
-      updateThread(updatedThread)
-
       // Add message
       const createdMessage = await extensionManager
         .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)