diff --git a/.trae/rules/project_rules.md b/.trae/rules/project_rules.md index 3a8243de..b8f6e123 100644 --- a/.trae/rules/project_rules.md +++ b/.trae/rules/project_rules.md @@ -22,7 +22,7 @@ NCurator-Desktop 是一个基于 Electron-Vite 构建的本地知识库搜索应 ### 后端 / 逻辑 (Node.js / 主进程) - **文档解析**: `langchain` (JS 版本) 用于解析 PDF 和 DOCX。 - **文本分片**: `langchain` 分片器。 -- **向量化**: `@xenova/transformers` (Transformers.js) 用于本地嵌入生成。 +- **向量化**: `@huggingface/transformers` (Transformers.js) 用于本地嵌入生成。 - **向量数据库**: `lancedb` 用于本地向量存储。 - **关键词搜索**: `flexsearch` 用于倒排索引和关键词匹配。 - **重排序**: 自定义实现的 RRF (Reciprocal Rank Fusion) 算法。 diff --git a/src/main/services/vector/EmbeddingService.ts b/src/main/services/vector/EmbeddingService.ts index bff93489..2244b150 100644 --- a/src/main/services/vector/EmbeddingService.ts +++ b/src/main/services/vector/EmbeddingService.ts @@ -10,7 +10,7 @@ env.allowLocalModels = true export class EmbeddingService { private static instance: EmbeddingService private extractor: any = null - private modelName: string = 'Xenova/jina-embeddings-v2-base-zh' // Using a Chinese model as default for now + private modelName: string = 'jinaai/jina-embeddings-v2-base-zh' // Using a Chinese model as default for now private constructor() {}