simplify unnecessary display logic (onyx-dot-app#2406)

UiPath · Oct 2, 2024 · a3b7b66 · a3b7b66
1 parent b2fda14
commit a3b7b66
Showing 1 changed file with 14 additions and 18 deletions.
diff --git a/web/src/app/chat/ChatPage.tsx b/web/src/app/chat/ChatPage.tsx
@@ -1965,21 +1965,16 @@ export function ChatPage({
                               const parentMessage = message.parentMessageId
                                 ? messageMap.get(message.parentMessageId)
                                 : null;
-
-                              if (
-                                (currentSessionRegenerationState?.regenerating &&
-                                  message.messageId >
-                                    currentSessionRegenerationState?.finalMessageIndex!) ||
-                                (currentSessionChatState == "loading" &&
-                                  ((i >= messageHistory.length - 2 &&
-                                    message.type == "user") ||
-                                    (i >= messageHistory.length - 1 &&
-                                      !currentSessionRegenerationState?.regenerating)))
-                              ) {
-                                return <></>;
-                              }
-
                               if (message.type === "user") {
+                                if (
+                                  (currentSessionChatState == "loading" &&
+                                    i == messageHistory.length - 1) ||
+                                  (currentSessionRegenerationState?.regenerating &&
+                                    message.messageId >=
+                                      currentSessionRegenerationState?.finalMessageIndex!)
+                                ) {
+                                  return <></>;
+                                }
                                 return (
                                   <div
                                     id={`message-${message.messageId}`}
@@ -2052,10 +2047,11 @@ export function ChatPage({
                                     : null;
 
                                 if (
-                                  currentSessionRegenerationState?.regenerating &&
-                                  currentSessionChatState == "loading" &&
-                                  (i == messageHistory.length - 1 ||
-                                    currentSessionRegenerationState?.regenerating)
+                                  (currentSessionChatState == "loading" &&
+                                    i > messageHistory.length - 1) ||
+                                  (currentSessionRegenerationState?.regenerating &&
+                                    message.messageId >
+                                      currentSessionRegenerationState?.finalMessageIndex!)
                                 ) {
                                   return <></>;
                                 }