diff --git a/vectordb_bench/backend/clients/pgvector/pgvector.py b/vectordb_bench/backend/clients/pgvector/pgvector.py index 5e66fd77e..1ae661041 100644 --- a/vectordb_bench/backend/clients/pgvector/pgvector.py +++ b/vectordb_bench/backend/clients/pgvector/pgvector.py @@ -341,9 +341,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()]