diff --git a/src/renderer/src/hooks/useKnowledge.ts b/src/renderer/src/hooks/useKnowledge.ts index 18a53324da..4de51ca1cc 100644 --- a/src/renderer/src/hooks/useKnowledge.ts +++ b/src/renderer/src/hooks/useKnowledge.ts @@ -199,10 +199,14 @@ export const useKnowledge = (baseId: string) => { } // 获取目录处理进度 - const getDirectoryProcessingPercent = (itemId: string) => { + const getDirectoryProcessingPercent = (itemId?: string) => { const [percent, setPercent] = useState(0) useEffect(() => { + if (!itemId) { + return + } + const cleanup = window.electron.ipcRenderer.on(itemId, (_, progressingPercent: number) => { setPercent(progressingPercent) }) diff --git a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx index d4ebf06278..602e748a35 100644 --- a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx +++ b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx @@ -64,6 +64,7 @@ const KnowledgeContent: FC = ({ selectedBase }) => { if (!base) { return null } + const progressingPercent = getDirectoryProcessingPercent(base?.id) const handleAddFile = () => { @@ -115,7 +116,7 @@ const KnowledgeContent: FC = ({ selectedBase }) => { inputPlaceholder: t('knowledge.url_placeholder'), inputProps: { rows: 10, - onPressEnter: () => { } + onPressEnter: () => {} } }) diff --git a/src/renderer/src/pages/knowledge/KnowledgePage.tsx b/src/renderer/src/pages/knowledge/KnowledgePage.tsx index ef2407cc46..8e9f55ab8c 100644 --- a/src/renderer/src/pages/knowledge/KnowledgePage.tsx +++ b/src/renderer/src/pages/knowledge/KnowledgePage.tsx @@ -65,6 +65,7 @@ const KnowledgePage: FC = () => { title: t('knowledge.delete_confirm'), centered: true, onOk: () => { + setSelectedBase(undefined) deleteKnowledgeBase(base.id) } })