Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions vectordb_bench/frontend/components/run_test/submitTask.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,11 +55,17 @@ def advancedSettings(st):
"Concurrent Input", value=defaultconcurrentInput, label_visibility="collapsed"
)
container[1].caption("num of concurrencies for search tests to get max-qps")
return index_already_exists, use_aliyun, k, concurrentInput

container = st.columns([1, 2])
concurrency_duration = container[0].number_input(
"Concurrency Duration", value=config.CONCURRENCY_DURATION, label_visibility="collapsed"
)
container[1].caption("concurrency duration for each concurrency search test")
return index_already_exists, use_aliyun, k, concurrentInput, concurrency_duration


def controlPanel(st, tasks: list[TaskConfig], taskLabel, isAllValid):
index_already_exists, use_aliyun, k, concurrentInput = advancedSettings(st)
index_already_exists, use_aliyun, k, concurrentInput, concurrency_duration = advancedSettings(st)

def runHandler():
benchmark_runner.set_drop_old(not index_already_exists)
Expand All @@ -73,7 +79,7 @@ def runHandler():
for task in tasks:
task.case_config.k = k
task.case_config.concurrency_search_config.num_concurrency = concurrentInput_list

task.case_config.concurrency_search_config.concurrency_duration = concurrency_duration
benchmark_runner.set_download_address(use_aliyun)
benchmark_runner.run(tasks, taskLabel)

Expand Down