From 90424808ab29720aaf6bbeb1d5e92165409c7aa1 Mon Sep 17 00:00:00 2001
From: Teo <cheesen.xu@gmail.com>
Date: Sun, 6 Apr 2025 18:21:51 +0800
Subject: [PATCH] refactor(Inputbar): streamline Backspace handling and update
 knowledge base management

---
 src/renderer/src/pages/home/Inputbar/Inputbar.tsx | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
index 182f532a..cb7304f9 100644
--- a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
+++ b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
@@ -478,11 +478,11 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic, topic }) =
         return newSelectedKnowledgeBases
       })
       return event.preventDefault()
+    }
 
-      if (event.key === 'Backspace' && text.trim() === '' && files.length > 0) {
-        setFiles((prev) => prev.slice(0, -1))
-        return event.preventDefault()
-      }
+    if (event.key === 'Backspace' && text.trim() === '' && files.length > 0) {
+      setFiles((prev) => prev.slice(0, -1))
+      return event.preventDefault()
     }
   }
 
@@ -757,7 +757,12 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic, topic }) =
   }
 
   const handleRemoveKnowledgeBase = (knowledgeBase: KnowledgeBase) => {
-    setSelectedKnowledgeBases(selectedKnowledgeBases.filter((kb) => kb.id !== knowledgeBase.id))
+    const newKnowledgeBases = assistant.knowledge_bases?.filter((kb) => kb.id !== knowledgeBase.id)
+    updateAssistant({
+      ...assistant,
+      knowledge_bases: newKnowledgeBases
+    })
+    setSelectedKnowledgeBases(newKnowledgeBases ?? [])
   }
 
   const toggelEnableMCP = (mcp: MCPServer) => {