vectordb-bench 0.0.9__tar.gz → 0.0.10__tar.gz

{vectordb_bench-0.0.9 → vectordb_bench-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectordb-bench
-Version: 0.0.9
+Version: 0.0.10
 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-0.0.9 → vectordb_bench-0.0.10}/vectordb_bench/__init__.py

@@ -32,10 +32,10 @@ class config:
     LOAD_TIMEOUT_1536D_500K     = 2.5 * 3600 # 2.5h
     LOAD_TIMEOUT_1536D_5M       =  25 * 3600 # 25h
 
-    OPTIMIZE_TIMEOUT_DEFAULT    = 15 * 60   # 15min
-    OPTIMIZE_TIMEOUT_768D_1M    =  15 * 60   # 15min
-    OPTIMIZE_TIMEOUT_768D_10M   = 2.5 * 3600 # 2.5h
-    OPTIMIZE_TIMEOUT_768D_100M  =  25 * 3600 # 1.04d
+    OPTIMIZE_TIMEOUT_DEFAULT    = 30 * 60   # 30min
+    OPTIMIZE_TIMEOUT_768D_1M    =  30 * 60   # 30min
+    OPTIMIZE_TIMEOUT_768D_10M   = 5 * 3600 # 5h
+    OPTIMIZE_TIMEOUT_768D_100M  =  50 * 3600 # 50h
 
 
     OPTIMIZE_TIMEOUT_1536D_500K =  15 * 60   # 15min

{vectordb_bench-0.0.9 → vectordb_bench-0.0.10}/vectordb_bench/backend/clients/milvus/config.py

@@ -14,13 +14,17 @@ class MilvusIndexConfig(BaseModel):
 
     index: IndexType
     metric_type: MetricType | None = None
+    
+    @property
+    def is_gpu_index(self) -> bool:
+        return self.index in [IndexType.GPU_CAGRA, IndexType.GPU_IVF_FLAT, IndexType.GPU_IVF_PQ]
 
     def parse_metric(self) -> str:
         if not self.metric_type:
             return ""
 
-        # if self.metric_type == MetricType.COSINE:
-        #     return MetricType.L2.value
+        if self.is_gpu_index and self.metric_type == MetricType.COSINE:
+            return MetricType.L2.value
         return self.metric_type.value
 
 

{vectordb_bench-0.0.9 → vectordb_bench-0.0.10}/vectordb_bench/backend/clients/milvus/milvus.py

@@ -8,7 +8,7 @@ from typing import Iterable
 from pymilvus import Collection, utility
 from pymilvus import CollectionSchema, DataType, FieldSchema, MilvusException
 
-from ..api import VectorDB, IndexType
+from ..api import VectorDB
 from .config import MilvusIndexConfig
 
 
@@ -119,7 +119,7 @@ class Milvus(VectorDB):
             wait_index()
 
             # Skip compaction if use GPU indexType
-            if self.case_config.index in [IndexType.GPU_CAGRA, IndexType.GPU_IVF_FLAT, IndexType.GPU_IVF_PQ]:
+            if self.case_config.is_gpu_index:
                 log.debug("skip compaction for gpu index type.")
             else :
                 self.col.compact()
@@ -157,6 +157,10 @@ class Milvus(VectorDB):
 
     def need_normalize_cosine(self) -> bool:
         """Wheather this database need to normalize dataset to support COSINE"""
+        if self.case_config.is_gpu_index:
+            log.info(f"current gpu_index only supports IP / L2, cosine dataset need normalize.")
+            return True
+        
         return False
 
     def insert_embeddings(

{vectordb_bench-0.0.9 → vectordb_bench-0.0.10}/vectordb_bench.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectordb-bench
-Version: 0.0.9
+Version: 0.0.10
 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