From 75f7760cd0671714250e2002dd5367df60e1f75a Mon Sep 17 00:00:00 2001
From: Abir Chakraborty <142606190+abirc8010@users.noreply.github.com>
Date: Mon, 11 Nov 2024 21:11:39 +0530
Subject: [PATCH] Fix logout issue (#666)

* set messages and avatar url to null on logout

* Fixed linting issues

* Remove unnecessary import
---
 packages/react/src/views/ChatHeader/ChatHeader.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/react/src/views/ChatHeader/ChatHeader.js b/packages/react/src/views/ChatHeader/ChatHeader.js
index 0e46b9ccd..7c43056a6 100644
--- a/packages/react/src/views/ChatHeader/ChatHeader.js
+++ b/packages/react/src/views/ChatHeader/ChatHeader.js
@@ -88,8 +88,9 @@ const ChatHeader = ({
   const dispatchToastMessage = useToastBarDispatch();
   const getMessagesAndRoles = useFetchChatData(showRoles);
   const setMessageLimit = useSettingsStore((state) => state.setMessageLimit);
-
+  const setMessages = useMessageStore((state) => state.setMessages);
   const avatarUrl = useUserStore((state) => state.avatarUrl);
+  const setUserAvatarUrl = useUserStore((state) => state.setUserAvatarUrl);
   const headerTitle = useMessageStore((state) => state.headerTitle);
   const filtered = useMessageStore((state) => state.filtered);
   const setFilter = useMessageStore((state) => state.setFilter);
@@ -128,6 +129,9 @@ const ChatHeader = ({
   const handleLogout = useCallback(async () => {
     try {
       await RCInstance.logout();
+      setMessages([]);
+      setUserAvatarUrl(null);
+      useMessageStore.setState({ isMessageLoaded: false });
     } catch (e) {
       console.error(e);
     } finally {