[Benchmark] Remove re-exports from bench_serving.py (#19130)

test/manual/hicache/test_disaggregation_hicache.py

@@ -7,7 +7,7 @@ from typing import Dict
 
 import requests
 
-from sglang.bench_serving import get_tokenizer
+from sglang.benchmark.utils import get_tokenizer
 from sglang.test.server_fixtures.disaggregation_fixture import (
     PDDisaggregationServerBase,
 )

test/registered/bench_fn/test_bench_serving_functionality.py

@@ -6,7 +6,8 @@ import unittest
 from http.server import BaseHTTPRequestHandler, HTTPServer
 from pathlib import Path
 
-from sglang.bench_serving import parse_custom_headers, run_benchmark
+from sglang.bench_serving import run_benchmark
+from sglang.benchmark.utils import parse_custom_headers
 from sglang.srt.utils import kill_process_tree
 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.test_utils import (

test/registered/hicache/test_hicache_storage_file_backend.py

@@ -16,7 +16,7 @@ from urllib.parse import urlparse
 
 import requests
 
-from sglang.bench_serving import get_tokenizer
+from sglang.benchmark.utils import get_tokenizer
 from sglang.srt.utils import kill_process_tree
 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k

test/registered/hicache/test_hicache_variants.py

@@ -12,7 +12,7 @@ from types import SimpleNamespace
 
 import requests
 
-from sglang.bench_serving import get_tokenizer
+from sglang.benchmark.utils import get_tokenizer
 from sglang.srt.utils import is_hip, kill_process_tree
 from sglang.test.run_eval import run_eval
 from sglang.test.test_utils import (