.

2025-05-09 20:34:01 +08:00 · 2025-05-09 20:34:01 +08:00 · 7872cb5197
parent 467f85fc1b
commit 7872cb5197
1 changed files with 2 additions and 1 deletions
--- a/scripts/rag_build_query.py
+++ b/scripts/rag_build_query.py
@ -14,6 +14,7 @@ from scripts.permissions import get_user_allowed_indexes
 import faiss
 from typing import List
 import asyncio
+from sentence_transformers import SentenceTransformer

 USER_INDEX_PATH = "index_data"
 USER_DOC_PATH = "docs"