[Fix] Blank between generating message and assistant (#619)

Co-authored-by: Ian Seabock (Centific Technologies Inc) <[email protected]>
microsoft · Feb 15, 2024 · 8c4ab56 · 8c4ab56
1 parent 5f0192b
commit 8c4ab56
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 34 deletions.
diff --git a/frontend/src/pages/chat/Chat.tsx b/frontend/src/pages/chat/Chat.tsx
@@ -211,7 +211,9 @@ const Chat = () => {
                                         msg.id = result.id;
                                         msg.date = new Date().toISOString();
                                     })
-                                    setShowLoadingMessage(false);
+                                    if (result.choices[0].messages?.some(m => m.role === ASSISTANT)) {
+                                        setShowLoadingMessage(false);
+                                    }
                                     result.choices[0].messages.forEach((resultObj) => {
                                         processResultMessage(resultObj, userMessage, conversationId);
                                     })
@@ -359,7 +361,9 @@ const Chat = () => {
                                         msg.id = result.id;
                                         msg.date = new Date().toISOString();
                                     })
-                                    setShowLoadingMessage(false);
+                                    if (result.choices[0].messages?.some(m => m.role === ASSISTANT)) {
+                                        setShowLoadingMessage(false);
+                                    }
                                     result.choices[0].messages.forEach((resultObj) => {
                                         processResultMessage(resultObj, userMessage, conversationId);
                                     })