Skip to content

Commit b7dc7d6

Browse files
committed
feat: add VectorChord support and VCHORDRQ index type 🎉
- Introduced VectorChord as a new database type in the DB enum. - Added VCHORDRQ as a new index type in the IndexType enum.
1 parent cf536a9 commit b7dc7d6

2 files changed

Lines changed: 17 additions & 0 deletions

File tree

‎vectordb_bench/backend/clients/__init__.py‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ class DB(Enum):
5959
Zvec = "Zvec"
6060
Endee = "Endee"
6161
Lindorm = "Lindorm"
62+
VectorChord = "VectorChord"
6263

6364
@property
6465
def init_cls(self) -> type[VectorDB]: # noqa: PLR0911, PLR0912, C901, PLR0915
@@ -246,6 +247,11 @@ def init_cls(self) -> type[VectorDB]: # noqa: PLR0911, PLR0912, C901, PLR0915
246247

247248
return LindormVector
248249

250+
if self == DB.VectorChord:
251+
from .vectorchord.vectorchord import VectorChord
252+
253+
return VectorChord
254+
249255
msg = f"Unknown DB: {self.name}"
250256
raise ValueError(msg)
251257

@@ -435,6 +441,11 @@ def config_cls(self) -> type[DBConfig]: # noqa: PLR0911, PLR0912, C901, PLR0915
435441

436442
return LindormConfig
437443

444+
if self == DB.VectorChord:
445+
from .vectorchord.config import VectorChordConfig
446+
447+
return VectorChordConfig
448+
438449
msg = f"Unknown DB: {self.name}"
439450
raise ValueError(msg)
440451

@@ -601,6 +612,11 @@ def case_config_cls( # noqa: C901, PLR0911, PLR0912, PLR0915
601612

602613
return _lindorm_vector_case_config.get(index_type)
603614

615+
if self == DB.VectorChord:
616+
from .vectorchord.config import _vectorchord_case_config
617+
618+
return _vectorchord_case_config.get(index_type)
619+
604620
# DB.Pinecone, DB.Redis
605621
return EmptyDBCaseConfig
606622

‎vectordb_bench/backend/clients/api.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ class IndexType(StrEnum):
4242
GPU_IVF_PQ = "GPU_IVF_PQ"
4343
GPU_CAGRA = "GPU_CAGRA"
4444
SCANN = "scann"
45+
VCHORDRQ = "vchordrq"
4546
SCANN_MILVUS = "SCANN_MILVUS"
4647
Hologres_HGraph = "HGraph"
4748
Hologres_Graph = "Graph"

0 commit comments

Comments
 (0)