From 7872cb5197ddbf855ecb0c6483ce907ce2d20466 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Fri, 9 May 2025 20:34:01 +0800
Subject: [PATCH] .

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

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)