diff --git a/.env.template b/.env.template index 76fdeef3..dbdf5fe6 100644 --- a/.env.template +++ b/.env.template @@ -1,5 +1,5 @@ -MODEL_DIR=models -SAVE_DIR=saves +MODEL_DIR=./models +SAVE_DIR=./saves # region model_provider # 推荐使用硅基流动免费服务 diff --git a/src/knowledge/implementations/chroma.py b/src/knowledge/implementations/chroma.py index c02ce91d..7d0bdff5 100644 --- a/src/knowledge/implementations/chroma.py +++ b/src/knowledge/implementations/chroma.py @@ -1,3 +1,4 @@ +import asyncio import os import traceback from datetime import datetime @@ -209,7 +210,12 @@ class ChromaKB(KnowledgeBase): batch_metadatas = metadatas[i : i + batch_size] batch_ids = ids[i : i + batch_size] - collection.add(documents=batch_documents, metadatas=batch_metadatas, ids=batch_ids) + await asyncio.to_thread( + collection.add, + documents=batch_documents, + metadatas=batch_metadatas, + ids=batch_ids, + ) batch_num = i // batch_size + 1 logger.info(f"Processed batch {batch_num}/{total_batches} for {filename}")