From cac1c993a9f020e42bb6a346f127fa0444abd2d4 Mon Sep 17 00:00:00 2001 From: ArnoChen Date: Fri, 14 Feb 2025 03:14:48 +0800 Subject: [PATCH] remove redundant cosine similarity filter in Qdrant query fix --- lightrag/kg/qdrant_impl.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/lightrag/kg/qdrant_impl.py b/lightrag/kg/qdrant_impl.py index 13d19806..3af76328 100644 --- a/lightrag/kg/qdrant_impl.py +++ b/lightrag/kg/qdrant_impl.py @@ -15,7 +15,6 @@ if not pm.is_installed("qdrant_client"): from qdrant_client import QdrantClient, models - config = configparser.ConfigParser() config.read("config.ini", "utf-8") @@ -138,12 +137,9 @@ class QdrantVectorDBStorage(BaseVectorStorage): query_vector=embedding[0], limit=top_k, with_payload=True, + score_threshold=self.cosine_better_than_threshold, ) + logger.debug(f"query result: {results}") - # 添加余弦相似度过滤 - filtered_results = [ - dp for dp in results if dp.score >= self.cosine_better_than_threshold - ] - return [ - {**dp.payload, "id": dp.id, "distance": dp.score} for dp in filtered_results - ] + + return [{**dp.payload, "id": dp.id, "distance": dp.score} for dp in results]