mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
20 lines
463 B
Python
20 lines
463 B
Python
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.INFO)
|