scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

feat(logging): enhance ECSFormatter to use service_name from env var and improve log formatting consistency

Browse source
This commit is contained in:
Supul 2026-06-04 23:34:25 +05:30
parent c8b0d8b1ae
commit e0a08ccf1e
1 changed files with 16 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
litellm/_logging.py
View file

@ -216,10 +216,12 @@ class ECSFormatter(Formatter):
def __init__(
self,
service_name: str = os.getenv("LITELLM_SERVICE_NAME", "litellm"),
service_name: Optional[str] = None,
):
super().__init__()
self._service_name = service_name
self._service_name = service_name or os.getenv(
"LITELLM_SERVICE_NAME", "litellm"
)
def formatTime(
self, record: logging.LogRecord, datefmt: Optional[str] = None
@ -397,12 +399,17 @@ def _initialize_loggers_with_handler(handler: logging.Handler):
def _get_uvicorn_json_log_config():
"""
Generate a uvicorn log_config dictionary that applies JSON formatting to all loggers.
Generate a uvicorn log_config dictionary that applies structured formatting to all loggers.
This ensures that uvicorn's access logs, error logs, and all application logs
are formatted as JSON when json_logs is enabled.
Uses ECSFormatter when LITELLM_ECS_LOGS=true so uvicorn access/error logs
are consistent with application logs for downstream ECS consumers.
Falls back to JsonFormatter when only JSON_LOGS=true.
"""
json_formatter_class = "litellm._logging.JsonFormatter"
formatter_class = (
"litellm._logging.ECSFormatter"
if ecs_logs
else "litellm._logging.JsonFormatter"
)
# Use the module-level log_level variable for consistency
uvicorn_log_level = log_level.upper()
@ -412,13 +419,13 @@ def _get_uvicorn_json_log_config():
"disable_existing_loggers": False,
"formatters": {
"json": {
"()": json_formatter_class,
"()": formatter_class,
},
"default": {
"()": json_formatter_class,
"()": formatter_class,
},
"access": {
"()": json_formatter_class,
"()": formatter_class,
},
},
"handlers": {