vectordb-bench 1.0.7__tar.gz → 1.0.8__tar.gz

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectordb-bench
-Version: 1.0.7
+Version: 1.0.8
 Summary: VectorDBBench is not just an offering of benchmark results for mainstream vector databases and cloud services, it's your go-to tool for the ultimate performance and cost-effectiveness comparison. Designed with ease-of-use in mind, VectorDBBench is devised to help users, even non-professionals, reproduce results or test new systems, making the hunt for the optimal choice amongst a plethora of cloud services and open-source vector databases a breeze.
 Author-email: XuanYang-cn <xuan.yang@zilliz.com>
 Project-URL: repository, https://github.com/zilliztech/VectorDBBench

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench/backend/clients/hologres/config.py

@@ -104,8 +104,6 @@ class HologresIndexConfig(BaseModel, DBCaseConfig):
             self.base_quantization_type = "fp32"
 
         return {
-            "min_flush_proxima_row_count": self.min_flush_proxima_row_count,
-            "min_compaction_proxima_row_count": self.min_compaction_proxima_row_count,
             "max_total_size_to_merge_mb": self.max_total_size_to_merge_mb,
             "build_thread_count": self.build_thread_count,
             "base_quantization_type": self.base_quantization_type,
@@ -113,6 +111,7 @@ class HologresIndexConfig(BaseModel, DBCaseConfig):
             "ef_construction": self.ef_construction,
             "precise_quantization_type": self.precise_quantization_type,
             "use_reorder": self.use_reorder,
+            "precise_io_type": "reader_io",
         }
 
     def searcher_params(self) -> dict:

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench/backend/clients/hologres/hologres.py

@@ -128,6 +128,18 @@ class Hologres(VectorDB):
         )
         self.conn.commit()
 
+        try:
+            log.info(f"{self.name} client purge table recycle bin: {self.table_name}")
+            self.cursor.execute(
+                sql.SQL("purge TABLE {table_name};").format(
+                    table_name=sql.Identifier(self.table_name),
+                ),
+            )
+        except Exception as e:
+            log.info(f"{self.name} client purge table {self.table_name} recycle bin failed, error: {e}, ignore.")
+        finally:
+            self.conn.commit()
+
         try:
             log.info(f"{self.name} client drop table group : {self._tg_name}")
             self.cursor.execute(sql.SQL(f"CALL HG_DROP_TABLE_GROUP('{self._tg_name}');"))
@@ -136,6 +148,14 @@ class Hologres(VectorDB):
         finally:
             self.conn.commit()
 
+        try:
+            log.info(f"{self.name} client free cache")
+            self.cursor.execute("select hg_admin_command('freecache');")
+        except Exception as e:
+            log.info(f"{self.name} client free cache failed, error: {e}, ignore.")
+        finally:
+            self.conn.commit()
+
     def optimize(self, data_size: int | None = None):
         if self.case_config.create_index_after_load:
             self._create_index()

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench/backend/clients/milvus/config.py

@@ -320,7 +320,6 @@ class GPUIVFFlatConfig(MilvusIndexConfig, DBCaseConfig):
 
 class GPUBruteForceConfig(MilvusIndexConfig, DBCaseConfig):
     limit: int = 10  # Default top-k for search
-    metric_type: str  # Metric type (e.g., 'L2', 'IP', etc.)
     index: IndexType = IndexType.GPU_BRUTE_FORCE  # Index type set to GPU_BRUTE_FORCE
 
     def index_param(self) -> dict:

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench/cli/cli.py

@@ -183,6 +183,11 @@ def get_custom_case_config(parameters: dict) -> dict:
                 "with_gt": parameters["custom_dataset_with_gt"],
             },
         }
+    elif parameters["case_type"] == "NewIntFilterPerformanceCase" :
+        custom_case_config = {
+            "dataset_with_size_type" : parameters["dataset_with_size_type"],
+            "filter_rate": parameters["filter_rate"],
+        }
     return custom_case_config
 
 
@@ -416,7 +421,24 @@ class CommonTypedDict(TypedDict):
         ),
     ]
     task_label: Annotated[str, click.option("--task-label", help="Task label")]
-
+    dataset_with_size_type: Annotated[
+        str,
+        click.option(
+            "--dataset-with-size-type",
+            help="Dataset with size type for NewIntFilterPerformanceCase, you can use Medium Cohere (768dim, 1M)|Large Cohere (768dim, 10M)|Medium Bioasq (1024dim, 1M)|Large Bioasq (1024dim, 10M)|Large OpenAI (1536dim, 5M)|Medium OpenAI (1536dim, 500K)",
+            default="Medium Cohere (768dim, 1M)",
+            show_default=True,
+        )
+    ]
+    filter_rate: Annotated[
+        float,
+        click.option(
+            "--filter-rate",
+            help="Filter rate for NewIntFilterPerformanceCase",
+            default=0.01,
+            show_default=True,
+        )
+    ]
 
 class HNSWBaseTypedDict(TypedDict):
     m: Annotated[int | None, click.option("--m", type=int, help="hnsw m")]

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench/models.py

@@ -127,6 +127,7 @@ class CaseConfigParamType(Enum):
     use_routing = "use_routing"
 
     dataset_with_size_type = "dataset_with_size_type"
+    filter_rate = "filter_rate"
     insert_rate = "insert_rate"
     search_stages = "search_stages"
     concurrencies = "concurrencies"

{vectordb_bench-1.0.7 → vectordb_bench-1.0.8}/vectordb_bench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectordb-bench
-Version: 1.0.7
+Version: 1.0.8
 Summary: VectorDBBench is not just an offering of benchmark results for mainstream vector databases and cloud services, it's your go-to tool for the ultimate performance and cost-effectiveness comparison. Designed with ease-of-use in mind, VectorDBBench is devised to help users, even non-professionals, reproduce results or test new systems, making the hunt for the optimal choice amongst a plethora of cloud services and open-source vector databases a breeze.
 Author-email: XuanYang-cn <xuan.yang@zilliz.com>
 Project-URL: repository, https://github.com/zilliztech/VectorDBBench