fix: empty response on some generation attempts (#4454)

janhq · Jan 15, 2025 · 9579e86 · 9579e86
1 parent 1964a9c
commit 9579e86
Showing 1 changed file with 21 additions and 21 deletions.
diff --git 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)
 
-    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)
+    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) {
       // Add message
       const createdMessage = await extensionManager
         .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)