Skip to content

feat(lakebase): add Lakebase Vector backend - #857

Open
junyulabs wants to merge 2 commits into
zilliztech:mainfrom
junyulabs:add-lakebase-vector
Open

feat(lakebase): add Lakebase Vector backend#857
junyulabs wants to merge 2 commits into
zilliztech:mainfrom
junyulabs:add-lakebase-vector

Conversation

@junyulabs

Copy link
Copy Markdown

Summary

This PR adds Lakebase Vector backend to VectorDBBench.

Usage

To run the Performance768D100M workload on Neon:

  1. Create a Neon instance with 40 CUs and wait until it is ready.
  2. Enable the Lakebase Vector extension by https://neon.com/docs/extensions/lakebase-vector
  3. Install and run VectorDBBench
export POSTGRES_HOST="<POSTGRES_HOST>"
export POSTGRES_PASSWORD="<POSTGRES_PASSWORD>"
export NUM_PER_BATCH=10000

vectordbbench lakebaseann \
  --host $POSTGRES_HOST \
  --user-name neondb_owner \
  --db-name neondb \
  --table-name vdbbench_laion100m \
  --db-label lakebase-search-laion100m \
  --task-label lakebase-search-laion100m \
  --case-type Performance768D100M \
  --k 100 \
  --load-concurrency 1 \
  --num-concurrency 1,5,10,20,30,40,60,80,100,120 \
  --concurrency-duration 30 \
  --concurrency-timeout 3600 \
  --max-parallel-workers 95 \
  --drop-old \
  --load \
  --search-concurrent \
  --search-serial \
  --probes 54,380 \
  --epsilon 3.2

@sre-ci-robot

Copy link
Copy Markdown

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by: junyulabs
To complete the pull request process, please assign xuanyang-cn after the PR has been reviewed.
You can assign the PR to them by writing /assign @xuanyang-cn in a comment when ready.

The full list of commands accepted by this bot can be found here.

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@junyulabs

Copy link
Copy Markdown
Author

/assign @XuanYang-cn

Comment thread vectordb_bench/backend/clients/lakebase_vector/lakebase_vector.py
Comment thread vectordb_bench/backend/clients/lakebase_vector/config.py
Comment thread vectordb_bench/backend/clients/lakebase_vector/lakebase_vector.py Outdated
Comment thread vectordb_bench/backend/clients/lakebase_vector/cli.py Outdated
@junyulabs
junyulabs force-pushed the add-lakebase-vector branch from d4c031e to ce73f04 Compare August 24, 2026 12:25
@junyulabs
junyulabs requested a review from yhmo August 25, 2026 02:04
Signed-off-by: Junyu Chen <junyu.chen@databricks.com>
@junyulabs
junyulabs force-pushed the add-lakebase-vector branch from ce73f04 to 8d225cd Compare August 26, 2026 11:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants