gitea/infio-copilot
1
0
Fork 0
mirror of https://github.com/EthanMarti/infio-copilot.git synced 2026-05-13 19:11:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

更新嵌入管理器以支持 GPU 加速，调整批处理大小，优化内容处理逻辑，并添加获取数据库最大修改时间的功能以提高文件索引效率。同时修复了向量管理器中的类型问题，确保模型加载和嵌入过程的稳定性。

Browse Source
This commit is contained in:
duanfuxiang
2025-07-05 07:40:54 +08:00
parent 558e3b3fe4
commit c657a50563
7 changed files with 398 additions and 340 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/rag/embedding.ts
View File

@@ -64,7 +64,7 @@ export const getEmbeddingModel = (
// 确保模型已加载
if (!embeddingManager.modelLoaded || embeddingManager.currentModel !== settings.embeddingModelId) {
console.log(`Loading model: ${settings.embeddingModelId}`)
await embeddingManager.loadModel(settings.embeddingModelId, false)
await embeddingManager.loadModel(settings.embeddingModelId, true)
}
const results = await embeddingManager.embedBatch(texts)