[model-gateway] fix version output (#14276)

This commit is contained in:
Simo Lin
2025-12-01 22:51:57 -08:00
committed by GitHub
parent 236a7c2370
commit 383689e3ad
6 changed files with 120 additions and 169 deletions
@@ -15,7 +15,10 @@ import os
import sys
from typing import List, Optional
from sglang_router.sglang_router_rs import get_short_version_string, get_version_string
from sglang_router.sglang_router_rs import (
get_verbose_version_string,
get_version_string,
)
def create_parser() -> argparse.ArgumentParser:
@@ -54,11 +57,11 @@ def main(argv: Optional[List[str]] = None) -> None:
argv = sys.argv[1:]
# Handle version flags before parsing
if argv and argv[0] in ["--version", "-V"]:
if argv[0] == "--version":
print(get_version_string())
if argv and argv[0] in ["--version", "-V", "--version-verbose"]:
if argv[0] == "--version-verbose":
print(get_verbose_version_string())
else:
print(get_short_version_string())
print(get_version_string())
sys.exit(0)
# Handle empty command - show help