Skip to content

Commit 6e76095

Browse files
committed
feat(oss-opensearch): Add CLI support for disk-based vector storage
Add --on-disk and --compression-level CLI parameters to enable disk-based vector search with compression from command line. Backend support already exists from PR zilliztech#680, this completes the implementation by exposing it via CLI.
1 parent addebc6 commit 6e76095

1 file changed

Lines changed: 22 additions & 0 deletions

File tree

  • vectordb_bench/backend/clients/oss_opensearch

vectordb_bench/backend/clients/oss_opensearch/cli.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,26 @@ class OSSOpenSearchTypedDict(TypedDict):
140140
),
141141
]
142142

143+
on_disk: Annotated[
144+
bool,
145+
click.option(
146+
"--on-disk",
147+
is_flag=True,
148+
help="Enable disk-based vector storage with compression (requires OpenSearch 2.17+)",
149+
default=False,
150+
),
151+
]
152+
153+
compression_level: Annotated[
154+
str,
155+
click.option(
156+
"--compression-level",
157+
type=click.Choice(["4x", "8x", "16x", "32x"], case_sensitive=False),
158+
help="Compression level for disk-based storage (4x, 8x, 16x, 32x)",
159+
default="32x",
160+
),
161+
]
162+
143163

144164
class OSSOpenSearchHNSWTypedDict(CommonTypedDict, OSSOpenSearchTypedDict, HNSWFlavor1): ...
145165

@@ -175,6 +195,8 @@ def OSSOpenSearch(**parameters: Unpack[OSSOpenSearchHNSWTypedDict]):
175195
confidence_interval=parameters["confidence_interval"],
176196
clip=parameters["clip"],
177197
metric_type_name=parameters["metric_type"],
198+
on_disk=parameters["on_disk"],
199+
compression_level=parameters["compression_level"],
178200
),
179201
**parameters,
180202
)

0 commit comments

Comments
 (0)