From d5e2a37414275f116f54a4df6b57146d92c461b5 Mon Sep 17 00:00:00 2001 From: Muqi Li Date: Sat, 13 Sep 2025 07:29:04 +0800 Subject: [PATCH] Benchmark: Support API_KEY without 'bearer' (#10380) --- python/sglang/bench_serving.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/python/sglang/bench_serving.py b/python/sglang/bench_serving.py index 27ff8a6da..401f702a4 100644 --- a/python/sglang/bench_serving.py +++ b/python/sglang/bench_serving.py @@ -105,10 +105,13 @@ def remove_suffix(text: str, suffix: str) -> str: def get_auth_headers() -> Dict[str, str]: - api_key = os.environ.get("OPENAI_API_KEY") - if api_key: - return {"Authorization": f"Bearer {api_key}"} + openai_api_key = os.environ.get("OPENAI_API_KEY") + if openai_api_key: + return {"Authorization": f"Bearer {openai_api_key}"} else: + api_key = os.environ.get("API_KEY") + if api_key: + return {"Authorization": f"{api_key}"} return {}