[model-gateway] fix graceful shutdown for TLS/Non-TLS server (#15491)

This commit is contained in:
Simo Lin
2025-12-19 12:35:50 -08:00
committed by GitHub
parent ef908aeb40
commit 50cad014dc
5 changed files with 43 additions and 9 deletions
@@ -59,6 +59,8 @@ class RouterArgs:
request_id_headers: Optional[List[str]] = None
# Request timeout in seconds
request_timeout_secs: int = 1800
# Grace period in seconds to wait for in-flight requests during shutdown
shutdown_grace_period_secs: int = 180
# Max concurrent requests for rate limiting (-1 to disable)
max_concurrent_requests: int = -1
# Queue size for pending requests when max concurrent limit reached
@@ -364,6 +366,12 @@ class RouterArgs:
default=RouterArgs.request_timeout_secs,
help="Request timeout in seconds",
)
parser.add_argument(
f"--{prefix}shutdown-grace-period-secs",
type=int,
default=RouterArgs.shutdown_grace_period_secs,
help="Grace period in seconds to wait for in-flight requests during shutdown",
)
# Retry configuration
parser.add_argument(
f"--{prefix}retry-max-retries",