From 44667beba9fa7d6fb0aa8459f5964d69055ea234 Mon Sep 17 00:00:00 2001
From: sunghyunkang1111 <sunghyunkang@microsoft.com>
Date: Fri, 6 Oct 2023 12:28:01 -0500
Subject: [PATCH] take out toggle button when closing tab

---
 src/Explorer/QueryCopilot/QueryCopilotTab.tsx | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/Explorer/QueryCopilot/QueryCopilotTab.tsx b/src/Explorer/QueryCopilot/QueryCopilotTab.tsx
index 5c93b569d..c7ce41aa6 100644
--- a/src/Explorer/QueryCopilot/QueryCopilotTab.tsx
+++ b/src/Explorer/QueryCopilot/QueryCopilotTab.tsx
@@ -66,6 +66,15 @@ export const QueryCopilotTab: React.FC<QueryCopilotProps> = ({ explorer }: Query
     useCommandBar.getState().setContextButtons(getCommandbarButtons());
   }, [query, selectedQuery, copilotActive]);
 
+  React.useEffect(() => {
+    return () => {
+      const commandbarButtons = getCommandbarButtons();
+      commandbarButtons.pop();
+      commandbarButtons.map((props: CommandButtonComponentProps) => (props.disabled = true));
+      useCommandBar.getState().setContextButtons(commandbarButtons);
+    };
+  }, []);
+
   const toggleCopilot = (toggle: boolean) => {
     setCopilotActive(toggle);
     localStorage.setItem(`${userContext.databaseAccount?.id}-queryCopilotToggleStatus`, toggle.toString());