diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py
index ed77ba1..28b5144 100644
--- a/scripts/rag_build_query.py
+++ b/scripts/rag_build_query.py
@@ -20,6 +20,8 @@ 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)