make collection name configurable

Author: lantingchiang

vectordb_bench/backend/clients/zilliz_cloud/cli.py

@@ -47,6 +47,17 @@ class ZillizTypedDict(CommonTypedDict):
             show_default=True,
         ),
     ]
+    collection_name: Annotated[
+        str,
+        click.option(
+            "--collection-name",
+            type=str,
+            help="Collection name for Zilliz",
+            required=False,
+            default="ZillizCloudVDBBench",
+            show_default=True,
+        ),
+    ]
 
 
 @cli.command()
@@ -62,6 +73,7 @@ def ZillizAutoIndex(**parameters: Unpack[ZillizTypedDict]):
             user=parameters["user_name"],
             password=SecretStr(parameters["password"]),
             num_shards=parameters["num_shards"],
+            collection_name=parameters["collection_name"],
         ),
         db_case_config=AutoIndexConfig(
             level=int(parameters["level"]) if parameters["level"] else 1,

vectordb_bench/backend/clients/zilliz_cloud/config.py

@@ -9,13 +9,15 @@ class ZillizCloudConfig(DBConfig):
     user: str
     password: SecretStr
     num_shards: int = 1
+    collection_name: str = "ZillizCloudVDBBench"
 
     def to_dict(self) -> dict:
         return {
             "uri": self.uri.get_secret_value(),
             "user": self.user,
             "password": self.password.get_secret_value(),
             "num_shards": self.num_shards,
+            "collection_name": self.collection_name,
         }