From 79ed4e35674668d84f1aa0874ac7a6eb1e26eb0c Mon Sep 17 00:00:00 2001
From: "sp.wack" <83104063+amanape@users.noreply.github.com>
Date: Wed, 13 Nov 2024 15:38:30 +0200
Subject: [PATCH] fix(frontend): Recover full message history if exists (#4961)

---
 frontend/src/components/event-handler.tsx   | 3 ---
 frontend/src/context/ws-client-provider.tsx | 3 +++
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/frontend/src/components/event-handler.tsx b/frontend/src/components/event-handler.tsx
index a9e0085ae694..3ab50245e87d 100644
--- a/frontend/src/components/event-handler.tsx
+++ b/frontend/src/components/event-handler.tsx
@@ -14,7 +14,6 @@ import {
 } from "#/context/ws-client-provider";
 import { ErrorObservation } from "#/types/core/observations";
 import { addErrorMessage, addUserMessage } from "#/state/chatSlice";
-import { handleAssistantMessage } from "#/services/actions";
 import {
   getCloneRepoCommand,
   getGitHubTokenCommand,
@@ -112,9 +111,7 @@ export function EventHandler({ children }: React.PropsWithChildren) {
           message: event.message,
         }),
       );
-      return;
     }
-    handleAssistantMessage(event);
   }, [events.length]);
 
   React.useEffect(() => {
diff --git a/frontend/src/context/ws-client-provider.tsx b/frontend/src/context/ws-client-provider.tsx
index 97aad6ca6061..4d00fd040e48 100644
--- a/frontend/src/context/ws-client-provider.tsx
+++ b/frontend/src/context/ws-client-provider.tsx
@@ -4,6 +4,7 @@ import { Settings } from "#/services/settings";
 import ActionType from "#/types/ActionType";
 import EventLogger from "#/utils/event-logger";
 import AgentState from "#/types/AgentState";
+import { handleAssistantMessage } from "#/services/actions";
 
 const RECONNECT_RETRIES = 5;
 
@@ -80,6 +81,8 @@ export function WsClientProvider({
     ) {
       setStatus(WsClientProviderStatus.ERROR);
     }
+
+    handleAssistantMessage(event);
   }
 
   function handleClose() {