yuchengwei 2 mesi fa
parent
commit
317f9c4bc1
1 ha cambiato i file con 3 aggiunte e 2 eliminazioni
  1. 3 2
      router/text_search.py

+ 3 - 2
router/text_search.py

@@ -5,6 +5,7 @@ from service.trunks_service import TrunksService
 from utils.text_splitter import TextSplitter
 from utils.vector_distance import VectorDistance
 from model.response import StandardResponse
+from utils.vectorizer import Vectorizer
 import logging
 
 logger = logging.getLogger(__name__)
@@ -36,10 +37,10 @@ async def search_text(request: TextSearchRequest):
                 # 如果有缓存结果,计算向量距离
                 min_distance = float('inf')
                 best_result = None
-                sentence_vector = trunks_service.get_vector(sentence)
+                sentence_vector = Vectorizer.get_embedding(sentence)
                 
                 for cached_result in cached_results:
-                    content_vector = trunks_service.get_vector(cached_result['content']); 
+                    content_vector = Vectorizer.get_embedding(cached_result['content'])
                     distance = VectorDistance.calculate_distance(sentence_vector, content_vector)
                     if distance < min_distance:
                         min_distance = distance