support rust sglang server (#29799)

This commit is contained in:
Rain Jiang
2026-07-31 11:56:31 -07:00
committed by GitHub
parent 77c77a3da8
commit 4af8ddb576
23 changed files with 1314 additions and 84 deletions
+10 -2
View File
@@ -16,6 +16,7 @@ from sglang.test.simple_eval_common import (
ChatCompletionSampler,
CompletionSampler,
Eval,
GenerateSampler,
make_report,
set_ulimit,
)
@@ -81,6 +82,13 @@ def run_eval_once(args, base_url: str, eval_obj: Eval) -> dict:
**common_kwargs,
stop=stop,
)
elif api_mode == "generate":
# SGLang-native `/generate` (raw text + sampling_params), same stop defaults.
stop = getattr(args, "stop", ["Question", "Assistant:", "<|separator|>"])
sampler = GenerateSampler(
**common_kwargs,
stop=stop,
)
else:
sampler = ChatCompletionSampler(
**common_kwargs,
@@ -454,8 +462,8 @@ if __name__ == "__main__":
"--api",
type=str,
default="chat",
choices=["chat", "completion"],
help="API mode: 'chat' for /v1/chat/completions, 'completion' for /v1/completions",
choices=["chat", "completion", "generate"],
help="API mode: 'chat' for /v1/chat/completions, 'completion' for /v1/completions, 'generate' for SGLang-native /generate",
)
parser.add_argument("--num-examples", type=int)
parser.add_argument("--num-threads", type=int, default=512)