From be5976fcd1e7decae5006912465aa032619b2efb Mon Sep 17 00:00:00 2001
From: Joshen Lim <joshenlimek@gmail.com>
Date: Fri, 28 Mar 2025 18:51:03 +0800
Subject: [PATCH] Small fixes

---
 apps/studio/components/layouts/AppLayout/AssistantButton.tsx | 3 +++
 .../components/layouts/AppLayout/InlineEditorButton.tsx      | 5 ++++-
 .../ui/AIAssistantPanel/AIAssistantChatSelector.tsx          | 3 ++-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/apps/studio/components/layouts/AppLayout/AssistantButton.tsx b/apps/studio/components/layouts/AppLayout/AssistantButton.tsx
index 8540c6087ff..0fbdbd187dd 100644
--- a/apps/studio/components/layouts/AppLayout/AssistantButton.tsx
+++ b/apps/studio/components/layouts/AppLayout/AssistantButton.tsx
@@ -1,8 +1,10 @@
 import { useAiAssistantStateSnapshot } from 'state/ai-assistant-state'
+import { useAppStateSnapshot } from 'state/app-state'
 import { AiIconAnimation, Button } from 'ui'
 
 const AssistantButton = () => {
   const snap = useAiAssistantStateSnapshot()
+  const { setEditorPanel } = useAppStateSnapshot()
 
   return (
     <Button
@@ -12,6 +14,7 @@ const AssistantButton = () => {
       className="h-full w-full rounded-none"
       onClick={() => {
         snap.toggleAssistant()
+        setEditorPanel({ open: false })
       }}
     >
       <AiIconAnimation allowHoverEffect size={20} />
diff --git a/apps/studio/components/layouts/AppLayout/InlineEditorButton.tsx b/apps/studio/components/layouts/AppLayout/InlineEditorButton.tsx
index 71ab33dbeda..10d3fbc0838 100644
--- a/apps/studio/components/layouts/AppLayout/InlineEditorButton.tsx
+++ b/apps/studio/components/layouts/AppLayout/InlineEditorButton.tsx
@@ -1,8 +1,10 @@
+import { SqlEditor } from 'icons'
+import { useAiAssistantStateSnapshot } from 'state/ai-assistant-state'
 import { useAppStateSnapshot } from 'state/app-state'
 import { Button } from 'ui'
-import { SqlEditor } from 'icons'
 
 const InlineEditorButton = () => {
+  const { closeAssistant } = useAiAssistantStateSnapshot()
   const { setEditorPanel, editorPanel } = useAppStateSnapshot()
 
   return (
@@ -12,6 +14,7 @@ const InlineEditorButton = () => {
       id="editor-trigger"
       className="h-full w-full rounded-none text-foreground-light"
       onClick={() => {
+        closeAssistant()
         setEditorPanel({ open: !editorPanel.open })
       }}
     >
diff --git a/apps/studio/components/ui/AIAssistantPanel/AIAssistantChatSelector.tsx b/apps/studio/components/ui/AIAssistantPanel/AIAssistantChatSelector.tsx
index b434aa0293f..02818e3475c 100644
--- a/apps/studio/components/ui/AIAssistantPanel/AIAssistantChatSelector.tsx
+++ b/apps/studio/components/ui/AIAssistantPanel/AIAssistantChatSelector.tsx
@@ -32,6 +32,7 @@ export const AIAssistantChatSelector = ({ className }: AIAssistantChatSelectorPr
   const [editingChatName, setEditingChatName] = useState('')
 
   const chats = Object.entries(snap.chats)
+  console.log(snap.chats)
 
   const handleSelectChat = (id: string) => {
     snap.selectChat(id)
@@ -109,7 +110,7 @@ export const AIAssistantChatSelector = ({ className }: AIAssistantChatSelectorPr
                     value={id}
                     onSelect={() => handleSelectChat(id)}
                     className="flex items-center justify-between gap-2 py-1 w-full overflow-hidden group"
-                    keywords={[chat.name]}
+                    keywords={[chat.name ?? '']}
                   >
                     <div className="flex items-center w-full flex-1 min-w-0">
                       {editingChatId === id ? (