feat: add custom request header logging (#17786)

This commit is contained in:
Joe Redmond
2026-01-28 19:33:08 -08:00
committed by GitHub
parent f1384f5293
commit 0ff0d181ca
4 changed files with 59 additions and 2 deletions

View File

@@ -163,6 +163,7 @@ class Envs:
SGLANG_LOG_MS = EnvBool(False)
SGLANG_DISABLE_REQUEST_LOGGING = EnvBool(False)
SGLANG_LOG_REQUEST_EXCEEDED_MS = EnvInt(-1)
SGLANG_LOG_REQUEST_HEADERS = EnvTuple(tuple())
SGLANG_LOG_SCHEDULER_STATUS_TARGET = EnvStr("")
SGLANG_LOG_SCHEDULER_STATUS_INTERVAL = EnvFloat(60.0)

View File

@@ -29,7 +29,10 @@ if TYPE_CHECKING:
logger = logging.getLogger(__name__)
WHITELISTED_HEADERS = ["x-smg-routing-key"]
_DEFAULT_WHITELISTED_HEADERS = ["x-smg-routing-key"]
WHITELISTED_HEADERS = _DEFAULT_WHITELISTED_HEADERS + [
h.lower() for h in envs.SGLANG_LOG_REQUEST_HEADERS.get()
]
def _extract_whitelisted_headers(