Skip to content

Commit 5549545

Browse files
LoveYou3000alwayslove2013
authored andcommitted
BugFix: An error occurs when the password option is not passed.
1 parent 9411abb commit 5549545

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

  • vectordb_bench/backend/clients/milvus

vectordb_bench/backend/clients/milvus/cli.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def MilvusAutoIndex(**parameters: Unpack[MilvusAutoIndexTypedDict]):
5656
db_label=parameters["db_label"],
5757
uri=SecretStr(parameters["uri"]),
5858
user=parameters["user_name"],
59-
password=SecretStr(parameters["password"]),
59+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
6060
num_shards=int(parameters["num_shards"]),
6161
),
6262
db_case_config=AutoIndexConfig(),
@@ -75,7 +75,7 @@ def MilvusFlat(**parameters: Unpack[MilvusAutoIndexTypedDict]):
7575
db_label=parameters["db_label"],
7676
uri=SecretStr(parameters["uri"]),
7777
user=parameters["user_name"],
78-
password=SecretStr(parameters["password"]),
78+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
7979
num_shards=int(parameters["num_shards"]),
8080
),
8181
db_case_config=FLATConfig(),
@@ -123,7 +123,7 @@ def MilvusIVFFlat(**parameters: Unpack[MilvusIVFFlatTypedDict]):
123123
db_label=parameters["db_label"],
124124
uri=SecretStr(parameters["uri"]),
125125
user=parameters["user_name"],
126-
password=SecretStr(parameters["password"]),
126+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
127127
num_shards=int(parameters["num_shards"]),
128128
),
129129
db_case_config=IVFFlatConfig(
@@ -145,7 +145,7 @@ def MilvusIVFSQ8(**parameters: Unpack[MilvusIVFFlatTypedDict]):
145145
db_label=parameters["db_label"],
146146
uri=SecretStr(parameters["uri"]),
147147
user=parameters["user_name"],
148-
password=SecretStr(parameters["password"]),
148+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
149149
num_shards=int(parameters["num_shards"]),
150150
),
151151
db_case_config=IVFSQ8Config(
@@ -171,7 +171,7 @@ def MilvusDISKANN(**parameters: Unpack[MilvusDISKANNTypedDict]):
171171
db_label=parameters["db_label"],
172172
uri=SecretStr(parameters["uri"]),
173173
user=parameters["user_name"],
174-
password=SecretStr(parameters["password"]),
174+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
175175
num_shards=int(parameters["num_shards"]),
176176
),
177177
db_case_config=DISKANNConfig(
@@ -200,7 +200,7 @@ def MilvusGPUIVFFlat(**parameters: Unpack[MilvusGPUIVFTypedDict]):
200200
db_label=parameters["db_label"],
201201
uri=SecretStr(parameters["uri"]),
202202
user=parameters["user_name"],
203-
password=SecretStr(parameters["password"]),
203+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
204204
num_shards=int(parameters["num_shards"]),
205205
),
206206
db_case_config=GPUIVFFlatConfig(
@@ -235,7 +235,7 @@ def MilvusGPUBruteForce(**parameters: Unpack[MilvusGPUBruteForceTypedDict]):
235235
db_label=parameters["db_label"],
236236
uri=SecretStr(parameters["uri"]),
237237
user=parameters["user_name"],
238-
password=SecretStr(parameters["password"]),
238+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
239239
num_shards=int(parameters["num_shards"]),
240240
),
241241
db_case_config=GPUBruteForceConfig(
@@ -267,7 +267,7 @@ def MilvusGPUIVFPQ(**parameters: Unpack[MilvusGPUIVFPQTypedDict]):
267267
db_label=parameters["db_label"],
268268
uri=SecretStr(parameters["uri"]),
269269
user=parameters["user_name"],
270-
password=SecretStr(parameters["password"]),
270+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
271271
num_shards=int(parameters["num_shards"]),
272272
),
273273
db_case_config=GPUIVFPQConfig(
@@ -307,7 +307,7 @@ def MilvusGPUCAGRA(**parameters: Unpack[MilvusGPUCAGRATypedDict]):
307307
db_label=parameters["db_label"],
308308
uri=SecretStr(parameters["uri"]),
309309
user=parameters["user_name"],
310-
password=SecretStr(parameters["password"]),
310+
password=SecretStr(parameters["password"]) if parameters["password"] else None,
311311
num_shards=int(parameters["num_shards"]),
312312
),
313313
db_case_config=GPUCAGRAConfig(

0 commit comments

Comments
 (0)