From 325e6cd65465c2ed26acf0abc3e132327aa93f53 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 18 Oct 2024 13:33:40 +0200 Subject: [PATCH] refactor: Rewrite search query Optimize search query for PGVector Refactor #COG-170 --- .../infrastructure/databases/vector/pgvector/PGVectorAdapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py index f77d1ee20..c592a5b7a 100644 --- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py @@ -164,7 +164,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface): try: PGVectorDataPoint = Table(collection_name, Base.metadata, autoload_with=self.engine) - closest_items = await session.execute(select(PGVectorDataPoint, PGVectorDataPoint.c.vector.cosine_distance(query_vector).label('similarity')).order_by(PGVectorDataPoint.c.vector.cosine_distance(query_vector)).limit(limit)) + closest_items = await session.execute(select(PGVectorDataPoint, PGVectorDataPoint.c.vector.cosine_distance(query_vector).label('similarity')).order_by('similarity').limit(limit)) vector_list = [] # Extract distances and find min/max for normalization