From 3de5f66aeb097cc0146bbdc193554a64422f6b78 Mon Sep 17 00:00:00 2001 From: hy Date: Wed, 24 Dec 2025 16:58:42 +0800 Subject: [PATCH] =?UTF-8?q?fix(=E5=90=91=E9=87=8F=E5=8C=96):=20=E6=9B=B4?= =?UTF-8?q?=E6=96=B0=E5=B5=8C=E5=85=A5=E6=A8=A1=E5=9E=8B=E4=BE=9D=E8=B5=96?= =?UTF-8?q?=E4=BB=8Exenova=E5=88=B0huggingface?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 更新项目文档和嵌入服务代码，将向量化依赖从@xenova/transformers迁移到@huggingface/transformers 同时更新默认模型路径为huggingface官方路径 --- .trae/rules/project_rules.md | 2 +- src/main/services/vector/EmbeddingService.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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() {}