diff --git a/vectordb_bench/backend/clients/milvus/cli.py b/vectordb_bench/backend/clients/milvus/cli.py index 48ad45608..ac4ed9ca6 100644 --- a/vectordb_bench/backend/clients/milvus/cli.py +++ b/vectordb_bench/backend/clients/milvus/cli.py @@ -179,6 +179,14 @@ class MilvusRefineTypedDict(TypedDict): class MilvusHNSWPQTypedDict(CommonTypedDict, MilvusTypedDict, MilvusHNSWTypedDict, MilvusRefineTypedDict): + pq_m: Annotated[ + int, + click.option( + "--pq-m", + type=int, + required=True, + ), + ] nbits: Annotated[ int, click.option( @@ -209,6 +217,7 @@ def MilvusHNSWPQ(**parameters: Unpack[MilvusHNSWPQTypedDict]): M=parameters["m"], efConstruction=parameters["ef_construction"], ef=parameters["ef_search"], + m=parameters["pq_m"], nbits=parameters["nbits"], refine=parameters["refine"], refine_type=parameters["refine_type"],