Selaa lähdekoodia

打标取distance值小于1的数据

yuchengwei 3 kuukautta sitten
vanhempi
commit
8b683259ab
1 muutettua tiedostoa jossa 6 lisäystä ja 1 poistoa
  1. 6 1
      router/text_search.py

+ 6 - 1
router/text_search.py

@@ -34,6 +34,11 @@ async def search_text(request: TextSearchRequest):
             
             # 处理搜索结果
             for result in search_results:
+                # 获取distance值,如果大于等于1则跳过
+                distance = result.get("distance", 1.0)
+                if distance >= 1:
+                    continue
+                
                 # 添加引用标记
                 result_sentence = sentence + f"^[{reference_index}]^"
                 result_sentences.append(result_sentence)
@@ -44,7 +49,7 @@ async def search_text(request: TextSearchRequest):
                     "content": result["content"],
                     "file_path": result.get("file_path", ""),
                     "title": result.get("title", ""),
-                    "distance": result.get("distance", "")
+                    "distance": distance
                 }
                 all_references.append(reference)
                 reference_index += 1