Skip to content

Commit fd980a6

Browse files
committed
feat: 更新 .env.template 文件路径,确保模型和保存目录的正确性;在 ChromaKB 中异步添加文档以提高性能
1 parent c3a8360 commit fd980a6

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

.env.template

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
MODEL_DIR=models
2-
SAVE_DIR=saves
1+
MODEL_DIR=./models
2+
SAVE_DIR=./saves
33

44
# region model_provider
55
# 推荐使用硅基流动免费服务

src/knowledge/implementations/chroma.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import asyncio
12
import os
23
import traceback
34
from datetime import datetime
@@ -209,7 +210,12 @@ async def add_content(self, db_id: str, items: list[str], params: dict | None) -
209210
batch_metadatas = metadatas[i : i + batch_size]
210211
batch_ids = ids[i : i + batch_size]
211212

212-
collection.add(documents=batch_documents, metadatas=batch_metadatas, ids=batch_ids)
213+
await asyncio.to_thread(
214+
collection.add,
215+
documents=batch_documents,
216+
metadatas=batch_metadatas,
217+
ids=batch_ids,
218+
)
213219

214220
batch_num = i // batch_size + 1
215221
logger.info(f"Processed batch {batch_num}/{total_batches} for {filename}")

0 commit comments

Comments
 (0)