Skip to content

Commit be540cc

Browse files
alwayslove2013XuanYang-cn
authored andcommitted
feat: Add concurrency duration configuration
Signed-off-by: min.tian <min.tian.cn@gmail.com>
1 parent 5fe69e0 commit be540cc

1 file changed

Lines changed: 9 additions & 3 deletions

File tree

vectordb_bench/frontend/components/run_test/submitTask.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,17 @@ def advancedSettings(st):
5555
"Concurrent Input", value=defaultconcurrentInput, label_visibility="collapsed"
5656
)
5757
container[1].caption("num of concurrencies for search tests to get max-qps")
58-
return index_already_exists, use_aliyun, k, concurrentInput
58+
59+
container = st.columns([1, 2])
60+
concurrency_duration = container[0].number_input(
61+
"Concurrency Duration", value=config.CONCURRENCY_DURATION, label_visibility="collapsed"
62+
)
63+
container[1].caption("concurrency duration for each concurrency search test")
64+
return index_already_exists, use_aliyun, k, concurrentInput, concurrency_duration
5965

6066

6167
def controlPanel(st, tasks: list[TaskConfig], taskLabel, isAllValid):
62-
index_already_exists, use_aliyun, k, concurrentInput = advancedSettings(st)
68+
index_already_exists, use_aliyun, k, concurrentInput, concurrency_duration = advancedSettings(st)
6369

6470
def runHandler():
6571
benchmark_runner.set_drop_old(not index_already_exists)
@@ -73,7 +79,7 @@ def runHandler():
7379
for task in tasks:
7480
task.case_config.k = k
7581
task.case_config.concurrency_search_config.num_concurrency = concurrentInput_list
76-
82+
task.case_config.concurrency_search_config.concurrency_duration = concurrency_duration
7783
benchmark_runner.set_download_address(use_aliyun)
7884
benchmark_runner.run(tasks, taskLabel)
7985

0 commit comments

Comments
 (0)