diff --git a/pyproject.toml b/pyproject.toml index aafe70750..8bc83f29b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,7 +35,7 @@ dependencies = [ "psutil", "polars", "plotly", - "environs", + "environs<14.1.0", "pydantic PgVectorIndexParam: index_parameters = {"lists": self.lists} @@ -215,6 +217,8 @@ class PgVectorHNSWConfig(PgVectorIndexConfig): reranking: bool | None = None quantized_fetch_limit: int | None = None reranking_metric: str | None = None + create_index_before_load: bool | None = True + create_index_after_load: bool | None = False def index_param(self) -> PgVectorIndexParam: index_parameters = {"m": self.m, "ef_construction": self.ef_construction} diff --git a/vectordb_bench/backend/clients/pgvector/pgvector.py b/vectordb_bench/backend/clients/pgvector/pgvector.py index 4164461fb..61f030cde 100644 --- a/vectordb_bench/backend/clients/pgvector/pgvector.py +++ b/vectordb_bench/backend/clients/pgvector/pgvector.py @@ -374,11 +374,11 @@ def _create_table(self, dim: int): "CREATE TABLE IF NOT EXISTS public.{table_name} (id BIGINT PRIMARY KEY, embedding vector({dim}));", ).format(table_name=sql.Identifier(self.table_name), dim=dim), ) - self.cursor.execute( - sql.SQL( - "ALTER TABLE public.{table_name} ALTER COLUMN embedding SET STORAGE PLAIN;", - ).format(table_name=sql.Identifier(self.table_name)), - ) + # self.cursor.execute( + # sql.SQL( + # "ALTER TABLE public.{table_name} ALTER COLUMN embedding SET STORAGE PLAIN;", + # ).format(table_name=sql.Identifier(self.table_name)), + # ) self.conn.commit() except Exception as e: log.warning(f"Failed to create pgvector table: {self.table_name} error: {e}")