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:
Krrish Dholakia 2024-06-10 17:33:43 -07:00
parent 8f3c25d0c5
commit b2c9b5bbf4

View file

@ -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()
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)