From 71a914d19ec82a1d8484c648c311954ad0d096f6 Mon Sep 17 00:00:00 2001
From: Georgiy Tarasov <gtarasov.work@gmail.com>
Date: Tue, 29 Oct 2024 12:43:45 +0100
Subject: [PATCH] fix: keep message status

---
 frontend/src/scenes/max/maxLogic.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frontend/src/scenes/max/maxLogic.ts b/frontend/src/scenes/max/maxLogic.ts
index 08bb8a9555933..8a0f4e4df1119 100644
--- a/frontend/src/scenes/max/maxLogic.ts
+++ b/frontend/src/scenes/max/maxLogic.ts
@@ -158,7 +158,7 @@ export const maxLogic = kea<maxLogicType>([
                             } else if (parsedResponse) {
                                 actions.replaceMessage(newIndex, {
                                     ...parsedResponse,
-                                    status: 'loading',
+                                    status: values.thread[newIndex].status,
                                 })
                             }
                         } else if (event === AssistantEventType.Status) {