From 467f85fc1b34da29b1ebeff88474b978ebce0949 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Fri, 9 May 2025 20:31:25 +0800
Subject: [PATCH] .

---
 scripts/rag_build_query.py | 2 ++
 1 file changed, 2 insertions(+)

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)