From d3397ba2e6c4af2876b1f06dd5ffd2a58c84753b Mon Sep 17 00:00:00 2001 From: "min.tian" Date: Thu, 23 Oct 2025 15:11:02 +0800 Subject: [PATCH] feat: Add concurrency duration configuration Signed-off-by: min.tian --- .../frontend/components/run_test/submitTask.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/vectordb_bench/frontend/components/run_test/submitTask.py b/vectordb_bench/frontend/components/run_test/submitTask.py index 5cb5ded43..01d0c5876 100644 --- a/vectordb_bench/frontend/components/run_test/submitTask.py +++ b/vectordb_bench/frontend/components/run_test/submitTask.py @@ -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) @@ -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)