Add CLI parameters for StreamingPerformanceCase configuration

Author: noahstaveley

vectordb_bench/backend/cases.py

@@ -459,9 +459,21 @@ def __init__(
         )
 
         if isinstance(search_stages, str):
-            search_stages = json.loads(search_stages)
+            # Handle both JSON array format and comma-separated format
+            search_stages = search_stages.strip()
+            if search_stages.startswith('['):
+                search_stages = json.loads(search_stages)
+            else:
+                # Convert comma-separated string to list of floats
+                search_stages = [float(x.strip()) for x in search_stages.split(',')]
         if isinstance(concurrencies, str):
-            concurrencies = json.loads(concurrencies)
+            # Handle both JSON array format and comma-separated format
+            concurrencies = concurrencies.strip()
+            if concurrencies.startswith('['):
+                concurrencies = json.loads(concurrencies)
+            else:
+                # Convert comma-separated string to list of ints
+                concurrencies = [int(x.strip()) for x in concurrencies.split(',')]
 
         super().__init__(
             name=name,

vectordb_bench/cli/cli.py

@@ -193,6 +193,13 @@ def get_custom_case_config(parameters: dict) -> dict:
             "dataset_with_size_type": parameters["dataset_with_size_type"],
             "label_percentage": parameters["label_percentage"],
         }
+    elif parameters["case_type"] == "StreamingPerformanceCase":
+        custom_case_config = {
+            "dataset_with_size_type": parameters["dataset_with_size_type"],
+            "insert_rate": parameters["insert_rate"],
+            "search_stages": parameters["search_stages"],
+            "concurrencies": parameters["concurrencies"],
+        }
     return custom_case_config
 
 
@@ -455,6 +462,33 @@ class CommonTypedDict(TypedDict):
             show_default=True,
         ),
     ]
+    insert_rate: Annotated[
+        int,
+        click.option(
+            "--insert-rate",
+            help="Insert rate (vectors/sec) for StreamingPerformanceCase",
+            default=500,
+            show_default=True,
+        ),
+    ]
+    search_stages: Annotated[
+        str,
+        click.option(
+            "--search-stages",
+            help="Search stages (comma-separated floats, e.g., '0.5,0.6,0.7,0.8,0.9') for StreamingPerformanceCase",
+            default="0.5,0.8",
+            show_default=True,
+        ),
+    ]
+    concurrencies: Annotated[
+        str,
+        click.option(
+            "--concurrencies",
+            help="Concurrency levels (comma-separated ints, e.g., '5,10,15') for StreamingPerformanceCase",
+            default="5,10",
+            show_default=True,
+        ),
+    ]
 
 
 class HNSWBaseTypedDict(TypedDict):