向量数据库基准测试

支持的数据库/扩展

• pgvecto.rs
• pgvector
• qdrant

支持的数据集

• 随机生成
• GIST 960

安装

pip install vector_bench

运行

服务器

在server文件夹下运行docker compose文件。

cd server/pgvecto.rs && docker compose up -d

客户端

# help
vector_bench --help
# only insert the data
vector_bench --insert --url postgresql://postgres:password@127.0.0.1:5432/postgres -s gist_960_l2
# only query the data (make sure the data is already inserted)
vector_bench --query --url postgresql://postgres:password@localhost:5432/postgres -s gist_960_l2
# insert and query the data
vector_bench --insert --query --url postgresql://postgres:password@localhost:5432/postgres -s gist_960_l2

如何贡献

# install all the necessary dependencies:
make dev
# format code
make format
# lint
make lint

添加更多数据集

• 将新的DatasetConfig添加到vector_bench/dataset/source.py

添加更多客户端

• 在vector_bench/client/base.py中继承并实现BaseClient类