Set local logging level via env variable (#2774)

This commit is contained in:
Lukas 2024-02-05 23:26:50 +01:00 committed by GitHub
parent 56f738ae9b
commit b92adec8e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -3,6 +3,7 @@
"""Logging configuration for vLLM."""
import logging
import sys
import os
_FORMAT = "%(levelname)s %(asctime)s %(filename)s:%(lineno)d] %(message)s"
_DATE_FORMAT = "%m-%d %H:%M:%S"
@ -50,7 +51,7 @@ _setup_logger()
def init_logger(name: str):
# Use the same settings as above for root logger
logger = logging.getLogger(name)
logger.setLevel(logging.DEBUG)
logger.setLevel(os.getenv("LOG_LEVEL", "DEBUG"))
logger.addHandler(_default_handler)
logger.propagate = False
return logger