From 749d12a971a99919759e9bcd0d882e7811bb6b1a Mon Sep 17 00:00:00 2001 From: tellet-q Date: Mon, 31 Mar 2025 15:33:32 +0200 Subject: [PATCH 1/4] Use query_points, enforce with_payload=True --- engine/clients/qdrant/search.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/engine/clients/qdrant/search.py b/engine/clients/qdrant/search.py index 613212cd..1174f3ec 100644 --- a/engine/clients/qdrant/search.py +++ b/engine/clients/qdrant/search.py @@ -51,14 +51,15 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]: ) try: - res = cls.client.search( + res = cls.client.query_points( collection_name=QDRANT_COLLECTION_NAME, - query_vector=query_vector, + query=query_vector, query_filter=cls.parser.parse(query.meta_conditions), limit=top, search_params=rest.SearchParams(**cls.search_params.get("config", {})), + with_payload=True ) except Exception as ex: print(f"Something went wrong during search: {ex}") raise ex - return [(hit.id, hit.score) for hit in res] + return [(hit.id, hit.score) for hit in res.points] From 0cc50ac1e3ae46c960710e17f868571f6d2fdc03 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 31 Mar 2025 13:34:53 +0000 Subject: [PATCH 2/4] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- engine/clients/qdrant/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engine/clients/qdrant/search.py b/engine/clients/qdrant/search.py index 1174f3ec..1fc65221 100644 --- a/engine/clients/qdrant/search.py +++ b/engine/clients/qdrant/search.py @@ -57,7 +57,7 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]: query_filter=cls.parser.parse(query.meta_conditions), limit=top, search_params=rest.SearchParams(**cls.search_params.get("config", {})), - with_payload=True + with_payload=True, ) except Exception as ex: print(f"Something went wrong during search: {ex}") From 0ceee92dce2834e9d987af0fbd550eabc105a36b Mon Sep 17 00:00:00 2001 From: tellet-q Date: Mon, 31 Mar 2025 15:38:46 +0200 Subject: [PATCH 3/4] Revert with_payload=True --- engine/clients/qdrant/search.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/engine/clients/qdrant/search.py b/engine/clients/qdrant/search.py index 1fc65221..16b2e496 100644 --- a/engine/clients/qdrant/search.py +++ b/engine/clients/qdrant/search.py @@ -56,8 +56,7 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]: query=query_vector, query_filter=cls.parser.parse(query.meta_conditions), limit=top, - search_params=rest.SearchParams(**cls.search_params.get("config", {})), - with_payload=True, + search_params=rest.SearchParams(**cls.search_params.get("config", {})) ) except Exception as ex: print(f"Something went wrong during search: {ex}") From b462597a4bc1d956e7ce7cbb2ea033220a5d4aee Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 31 Mar 2025 13:39:31 +0000 Subject: [PATCH 4/4] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- engine/clients/qdrant/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engine/clients/qdrant/search.py b/engine/clients/qdrant/search.py index 16b2e496..7f94de78 100644 --- a/engine/clients/qdrant/search.py +++ b/engine/clients/qdrant/search.py @@ -56,7 +56,7 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]: query=query_vector, query_filter=cls.parser.parse(query.meta_conditions), limit=top, - search_params=rest.SearchParams(**cls.search_params.get("config", {})) + search_params=rest.SearchParams(**cls.search_params.get("config", {})), ) except Exception as ex: print(f"Something went wrong during search: {ex}")