fix(_logging.py): add loglevel and timestamp to json logs

Fixes https://github.com/BerriAI/litellm/issues/4248
This commit is contained in:
Krrish Dholakia 2024-06-17 18:57:20 -07:00
parent 4f32bca8d9
commit 4cddd1eb31
3 changed files with 18 additions and 7 deletions

View file

@ -1,6 +1,8 @@
import logging, os, json import json
from logging import Formatter import logging
import os
import traceback import traceback
from logging import Formatter
set_verbose = False set_verbose = False
@ -21,8 +23,12 @@ class JsonFormatter(Formatter):
super(JsonFormatter, self).__init__() super(JsonFormatter, self).__init__()
def format(self, record): def format(self, record):
json_record = {} json_record = {
json_record["message"] = record.getMessage() "message": record.getMessage(),
"level": record.levelname,
"timestamp": self.formatTime(record, self.datefmt),
}
return json.dumps(json_record) return json.dumps(json_record)

View file

@ -1,7 +1,8 @@
import json import json
import logging import logging
from logging import Formatter
import os import os
from logging import Formatter
from litellm import json_logs from litellm import json_logs
# Set default log level to INFO # Set default log level to INFO
@ -14,8 +15,11 @@ class JsonFormatter(Formatter):
super(JsonFormatter, self).__init__() super(JsonFormatter, self).__init__()
def format(self, record): def format(self, record):
json_record = {} json_record = {
json_record["message"] = record.getMessage() "message": record.getMessage(),
"level": record.levelname,
"timestamp": self.formatTime(record, self.datefmt),
}
return json.dumps(json_record) return json.dumps(json_record)

View file

@ -79,6 +79,7 @@ litellm_settings:
success_callback: ["langfuse"] success_callback: ["langfuse"]
failure_callback: ["langfuse"] failure_callback: ["langfuse"]
cache: true cache: true
json_logs: true
general_settings: general_settings:
alerting: ["slack"] alerting: ["slack"]