feat: remove knowledge queue

This commit is contained in:
kangfenmao
2024-12-19 13:45:11 +08:00
parent 7cb381be40
commit a0bc50b29c
26 changed files with 388 additions and 375 deletions
+2
View File
@@ -55,6 +55,8 @@ class FileStorage {
const storedFilePath = path.join(this.storageDir, file)
const storedStats = fs.statSync(storedFilePath)
console.debug('storedFilePath', storedFilePath)
if (storedStats.size === fileSize) {
const [originalHash, storedHash] = await Promise.all([
this.getFileHash(filePath),
+1 -3
View File
@@ -11,7 +11,6 @@ import { WebLoader } from '@llm-tools/embedjs-loader-web'
import { OpenAiEmbeddings } from '@llm-tools/embedjs-openai'
import { FileType, RagAppRequestParams } from '@types'
import { app } from 'electron'
import Logger from 'electron-log'
class KnowledgeService {
private storageDir = path.join(app.getPath('userData'), 'Data', 'KnowledgeBase')
@@ -27,7 +26,6 @@ class KnowledgeService {
}
private getRagApplication = async ({ id, model, apiKey, baseURL }: RagAppRequestParams): Promise<RAGApplication> => {
Logger.log('getRagApplication', { id, model, apiKey, baseURL })
return new RAGApplicationBuilder()
.setModel('NO_MODEL')
.setEmbeddingModel(
@@ -82,7 +80,7 @@ class KnowledgeService {
return await ragApplication.addLoader(new DocxLoader({ filePathOrUrl: data.path }) as any)
}
if (data.ext === '.md') {
if (data.ext === '.md' || data.ext === '.mdx') {
return await ragApplication.addLoader(new MarkdownLoader({ filePathOrUrl: data.path }) as any)
}
+4
View File
@@ -123,6 +123,10 @@ export class WindowService {
private setupWebContentsHandlers(mainWindow: BrowserWindow) {
mainWindow.webContents.on('will-navigate', (event, url) => {
if (url.includes('localhost:5173')) {
return
}
event.preventDefault()
shell.openExternal(url)
})