mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-15 12:21:11 +00:00
feat(proxy/_logging.py): support controlling log level when json logs=True
allow user to control log level when json_logs=True
This commit is contained in:
@@ -1,7 +1,12 @@
|
||||
import json
|
||||
import logging
|
||||
from logging import Formatter
|
||||
import sys
|
||||
import os
|
||||
from litellm import json_logs
|
||||
|
||||
# Set default log level to INFO
|
||||
log_level = os.getenv("LITELLM_LOG", "INFO")
|
||||
numeric_level: str = getattr(logging, log_level.upper())
|
||||
|
||||
|
||||
class JsonFormatter(Formatter):
|
||||
@@ -16,6 +21,14 @@ class JsonFormatter(Formatter):
|
||||
|
||||
logger = logging.root
|
||||
handler = logging.StreamHandler()
|
||||
handler.setFormatter(JsonFormatter())
|
||||
if json_logs:
|
||||
handler.setFormatter(JsonFormatter())
|
||||
else:
|
||||
formatter = logging.Formatter(
|
||||
"\033[92m%(asctime)s - %(name)s:%(levelname)s\033[0m: %(filename)s:%(lineno)s - %(message)s",
|
||||
datefmt="%H:%M:%S",
|
||||
)
|
||||
|
||||
handler.setFormatter(formatter)
|
||||
logger.handlers = [handler]
|
||||
logger.setLevel(logging.INFO)
|
||||
logger.setLevel(numeric_level)
|
||||
|
||||
Reference in New Issue
Block a user