Tiny support http headers in bench serving (#16606)

2026-01-07 10:15:17 +08:00
parent 9a21d89c5b
commit d874c8bba4
3 changed files with 109 additions and 5 deletions
--- a/python/sglang/test/test_utils.py
+++ b/python/sglang/test/test_utils.py
@@ -778,6 +778,7 @@ def get_benchmark_args(
    gsp_question_len=32,
    gsp_output_len=32,
    gsp_num_turns=1,
+    header=None,
 ):
    return SimpleNamespace(
        backend=backend,
@@ -818,6 +819,7 @@ def get_benchmark_args(
        gsp_question_len=gsp_question_len,
        gsp_output_len=gsp_output_len,
        gsp_num_turns=gsp_num_turns,
+        header=header,
    )