diff --git a/engine/clients/qdrant/search.py b/engine/clients/qdrant/search.py index 613212cd..7f94de78 100644 --- a/engine/clients/qdrant/search.py +++ b/engine/clients/qdrant/search.py @@ -51,9 +51,9 @@ 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", {})), @@ -61,4 +61,4 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]: 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]