diff --git a/src/main/services/KnowledgeService.ts b/src/main/services/KnowledgeService.ts index 95634a7d..ccbf2bb3 100644 --- a/src/main/services/KnowledgeService.ts +++ b/src/main/services/KnowledgeService.ts @@ -178,6 +178,17 @@ class KnowledgeService { const fileContent = fs.readFileSync(file.path, 'utf-8') + if (['.html'].includes(file.ext)) { + return await ragApplication.addLoader( + new WebLoader({ + urlOrContent: fileContent, + chunkSize: base.chunkSize, + chunkOverlap: base.chunkOverlap + }) as any, + forceReload + ) + } + return await ragApplication.addLoader( new TextLoader({ text: fileContent, chunkSize: base.chunkSize, chunkOverlap: base.chunkOverlap }), forceReload diff --git a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx index 857be2ea..6474e08e 100644 --- a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx +++ b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx @@ -33,7 +33,7 @@ interface KnowledgeContentProps { selectedBase: KnowledgeBase } -const fileTypes = ['.pdf', '.docx', '.pptx', '.xlsx', '.txt', '.md'] +const fileTypes = ['.pdf', '.docx', '.pptx', '.xlsx', '.txt', '.md', '.html'] const KnowledgeContent: FC = ({ selectedBase }) => { const { t } = useTranslation()