Ver Fonte

增加最低分阈值

luojunhui há 2 semanas atrás
pai
commit
93623b6083
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      applications/utils/milvus/search.py

+ 2 - 2
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] if hit.distance >= 0.6
+            for hit in hits[0]
         ]
 
 
@@ -56,7 +56,7 @@ class MilvusSearch(MilvusBase):
             expr=expr,
             output_fields=self.output_fields,
         )
-        return {"results": self.hits_to_json(response)}
+        return {"results": self.hits_to_json(response[:10])}
 
     async def search_by_strategy(
         self,