Skip to content

Conversation

@Denise2004
Copy link
Contributor

Add Batch CLI Support and Refactor Task Building Logic

Summary

This PR adds batch CLI support, allowing users to build and execute multiple test tasks from YAML configuration files. It also refactors the task building logic by separating task construction from execution, improving code reusability.

Key Changes

1. New Batch CLI Feature (vectordb_bench/cli/batch_cli.py)

  • Supports reading multiple task configurations from batch config files
  • Automatically collects all tasks and uses a unified task_label

2. Refactored Task Building Logic (vectordb_bench/cli/cli.py)

  • Split run function into two functions:
    • build_task: Only builds task configuration without executing
    • run: Calls build_task to build configuration and executes tasks

@sre-ci-robot
Copy link

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by: Denise2004
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.

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

@Denise2004 Denise2004 force-pushed the batch-cli branch 3 times, most recently from 6cdbb82 to a5095bf Compare December 2, 2025 08:21
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.

2 participants