mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-16 23:41:43 +00:00
3cb64978d4
LITELLM_LOG=ERROR still printed INFO lines from uvicorn (startup and access log) and from the litellm_proxy_extras migration logger, because neither read the variable. Forward the resolved level to uvicorn when LITELLM_LOG is set and no explicit log_config or JSON logging is in use, and let the extras logger take its level from LITELLM_LOG Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
from datetime import datetime
|
|
|
|
|
|
class JsonFormatter(logging.Formatter):
|
|
def formatTime(self, record, datefmt=None):
|
|
dt = datetime.fromtimestamp(record.created)
|
|
return dt.isoformat()
|
|
|
|
def format(self, record):
|
|
json_record = {
|
|
"message": record.getMessage(),
|
|
"level": record.levelname,
|
|
"timestamp": self.formatTime(record),
|
|
}
|
|
if record.exc_info:
|
|
json_record["stacktrace"] = self.formatException(record.exc_info)
|
|
return json.dumps(json_record)
|
|
|
|
|
|
def _is_json_enabled():
|
|
try:
|
|
import litellm
|
|
|
|
return getattr(litellm, "json_logs", False)
|
|
except (ImportError, AttributeError):
|
|
return os.getenv("JSON_LOGS", "false").lower() == "true"
|
|
|
|
|
|
logger = logging.getLogger("litellm_proxy_extras")
|
|
|
|
if not logger.handlers:
|
|
handler = logging.StreamHandler()
|
|
if _is_json_enabled():
|
|
handler.setFormatter(JsonFormatter())
|
|
else:
|
|
handler.setFormatter(
|
|
logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
|
)
|
|
logger.addHandler(handler)
|
|
logger.setLevel(os.getenv("LITELLM_LOG", "INFO").upper())
|