Explorar o código

增加最低分阈值

luojunhui hai 2 semanas
pai
achega
debb504f74
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      applications/utils/milvus/search.py

+ 1 - 1
applications/utils/milvus/search.py

@@ -28,7 +28,7 @@ class MilvusBase:
                     for key, value in (hit.get("entity", {}) or {}).items()
                 },
             }
-            for hit in hits[0]
+            for hit in hits[0] if hit.distance >= 0.6
         ]