feat(proxy_cli.py): support json logs on proxy

allow user to enable 'json logs' for proxy server
This commit is contained in:
Krrish Dholakia 2024-05-20 09:18:12 -07:00
parent 0016477d3b
commit 00d47f3d3b
4 changed files with 34 additions and 4 deletions

20
litellm/proxy/_logging.py Normal file
View file

@ -0,0 +1,20 @@
import json
import logging
from logging import Formatter
class JsonFormatter(Formatter):
def __init__(self):
super(JsonFormatter, self).__init__()
def format(self, record):
json_record = {}
json_record["message"] = record.getMessage()
return json.dumps(json_record)
logger = logging.root
handler = logging.StreamHandler()
handler.setFormatter(JsonFormatter())
logger.handlers = [handler]
logger.setLevel(logging.DEBUG)