phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

feat(proxy/_logging.py): support controlling log level when json logs=True

Browse source 
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
1 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/_logging.py
View file

@ -1,7 +1,12 @@
import json import json
import logging import logging
from logging import Formatter 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): class JsonFormatter(Formatter):
@ -16,6 +21,14 @@ class JsonFormatter(Formatter):
logger = logging.root logger = logging.root
handler = logging.StreamHandler() handler = logging.StreamHandler()
if json_logs:
handler.setFormatter(JsonFormatter()) 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.handlers = [handler]
logger.setLevel(logging.INFO) logger.setLevel(numeric_level)