From fd980a6153b993917ae82f062703abe542d43bb6 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Sat, 11 Oct 2025 11:47:11 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0=20.env.template=20?= =?UTF-8?q?=E6=96=87=E4=BB=B6=E8=B7=AF=E5=BE=84=EF=BC=8C=E7=A1=AE=E4=BF=9D?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=92=8C=E4=BF=9D=E5=AD=98=E7=9B=AE=E5=BD=95?= =?UTF-8?q?=E7=9A=84=E6=AD=A3=E7=A1=AE=E6=80=A7=EF=BC=9B=E5=9C=A8=20Chroma?= =?UTF-8?q?KB=20=E4=B8=AD=E5=BC=82=E6=AD=A5=E6=B7=BB=E5=8A=A0=E6=96=87?= =?UTF-8?q?=E6=A1=A3=E4=BB=A5=E6=8F=90=E9=AB=98=E6=80=A7=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .env.template | 4 ++-- src/knowledge/implementations/chroma.py | 8 +++++++- 2 files changed, 9 insertions(+), 3 deletions(-) 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}")