diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py
index 28b5144..3540dd4 100644
--- 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"
@@ -21,7 +22,7 @@ USER_DOC_PATH = "docs"
 
 class CustomEmbedding(BaseEmbedding):
     model: SentenceTransformer  # ✅ 显式声明是字段
-    
+
     def __init__(self, model_name: str):
         from sentence_transformers import SentenceTransformer
         self.model = SentenceTransformer(model_name)