From 1821073098e6eb888f52609cbdf6a9ff3753e710 Mon Sep 17 00:00:00 2001 From: JackLCL Date: Wed, 24 Dec 2025 18:58:38 +0800 Subject: [PATCH] feat(milvus): add nprobe parameter for SCANN index - Add nprobe search parameter for SCANN index (default: 64) - Enable nprobe UI configuration for SCANN_MILVUS index type Signed-off-by: JackLCL --- vectordb_bench/backend/clients/milvus/config.py | 6 +++++- vectordb_bench/frontend/config/dbCaseConfigs.py | 1 + 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/vectordb_bench/backend/clients/milvus/config.py b/vectordb_bench/backend/clients/milvus/config.py index 70271a600..9ffbdcece 100644 --- a/vectordb_bench/backend/clients/milvus/config.py +++ b/vectordb_bench/backend/clients/milvus/config.py @@ -417,6 +417,7 @@ def search_param(self) -> dict: class SCANNConfig(MilvusIndexConfig, DBCaseConfig): nlist: int = 1024 with_raw_data: bool = False + nprobe: int = 64 reorder_k: int | None = 100 index: IndexType = IndexType.SCANN_MILVUS @@ -433,7 +434,10 @@ def index_param(self) -> dict: def search_param(self) -> dict: return { "metric_type": self.parse_metric(), - "params": {"reorder_k": self.reorder_k}, + "params": { + "nprobe": self.nprobe, + "reorder_k": self.reorder_k, + }, } diff --git a/vectordb_bench/frontend/config/dbCaseConfigs.py b/vectordb_bench/frontend/config/dbCaseConfigs.py index 83a5bb0e0..6d27a0e3d 100644 --- a/vectordb_bench/frontend/config/dbCaseConfigs.py +++ b/vectordb_bench/frontend/config/dbCaseConfigs.py @@ -1056,6 +1056,7 @@ class CaseConfigInput(BaseModel): IndexType.IVFPQ.value, IndexType.IVFSQ8.value, IndexType.IVF_RABITQ.value, + IndexType.SCANN_MILVUS.value, IndexType.GPU_IVF_FLAT.value, IndexType.GPU_IVF_PQ.value, IndexType.GPU_BRUTE_FORCE.value,