diff --git a/vectordb_bench/backend/clients/pgvector/pgvector.py b/vectordb_bench/backend/clients/pgvector/pgvector.py index 8f8244412..53bf18453 100644 --- a/vectordb_bench/backend/clients/pgvector/pgvector.py +++ b/vectordb_bench/backend/clients/pgvector/pgvector.py @@ -339,9 +339,10 @@ def search_embedding( assert self.conn is not None, "Connection is not initialized" assert self.cursor is not None, "Cursor is not initialized" + q = np.asarray(query) # TODO add filters support result = self.cursor.execute( - self._unfiltered_search, (query, k), prepare=True, binary=True + self._unfiltered_search, (q, k), prepare=True, binary=True ) return [int(i[0]) for i in result.fetchall()]