diff --git a/components/scripts/knowledge-dropdown.tsx b/components/scripts/knowledge-dropdown.tsx index 7f1f1947..06943205 100644 --- a/components/scripts/knowledge-dropdown.tsx +++ b/components/scripts/knowledge-dropdown.tsx @@ -13,11 +13,11 @@ import { MessageType } from '@/components/chat/messages'; import { GoCheckCircleFill } from 'react-icons/go'; import { deleteKnowledgeFile, lsKnowledgeFiles } from '@/actions/upload'; import { ingest } from '@/actions/knowledge/knowledge'; -import { getCookie } from '@/actions/knowledge/util'; +import { gatewayTool, getCookie } from '@/actions/knowledge/util'; import { Dirent } from 'fs'; const ScriptKnowledgeDropdown = () => { - const { workspace, selectedThreadId, program, setMessages } = + const { socket, workspace, selectedThreadId, program, setMessages } = useContext(ChatContext); const [knowledgeFiles, setKnowledgeFiles] = useState([]); @@ -41,7 +41,12 @@ const ScriptKnowledgeDropdown = () => { message: `Successfully removed knowledge ${file}`, }, ]); - setKnowledgeFiles(knowledgeFiles.filter((f) => f !== file)); + + const newKnowledgeFiles = knowledgeFiles.filter((f) => f !== file); + setKnowledgeFiles(newKnowledgeFiles); + if (newKnowledgeFiles.length === 0) { + socket?.emit('removeTool', gatewayTool()); + } } ); });