fix(proxy/_logging.py): fix default logging level

This commit is contained in:
Krrish Dholakia 2024-06-05 17:42:49 -07:00
parent 6c9303f811
commit 2d2e411c11
4 changed files with 13 additions and 6 deletions

View file

@ -41,12 +41,19 @@ verbose_logger.addHandler(handler)
def _turn_on_json(): def _turn_on_json():
handler = logging.StreamHandler() handler = logging.StreamHandler()
handler.setLevel(logging.DEBUG)
handler.setFormatter(JsonFormatter()) handler.setFormatter(JsonFormatter())
verbose_router_logger.addHandler(handler) # Define a list of the loggers to update
verbose_proxy_logger.addHandler(handler) loggers = [verbose_router_logger, verbose_proxy_logger, verbose_logger]
verbose_logger.addHandler(handler)
# Iterate through each logger and update its handlers
for logger in loggers:
# Remove all existing handlers
for h in logger.handlers[:]:
logger.removeHandler(h)
# Add the new handler
logger.addHandler(handler)
def _turn_on_debug(): def _turn_on_debug():

View file

@ -17,4 +17,4 @@ logger = logging.root
handler = logging.StreamHandler() handler = logging.StreamHandler()
handler.setFormatter(JsonFormatter()) handler.setFormatter(JsonFormatter())
logger.handlers = [handler] logger.handlers = [handler]
logger.setLevel(logging.DEBUG) logger.setLevel(logging.INFO)

View file

@ -56,6 +56,7 @@ router_settings:
litellm_settings: litellm_settings:
success_callback: ["langfuse"] success_callback: ["langfuse"]
json_logs: true
general_settings: general_settings:
alerting: ["email"] alerting: ["email"]

View file

@ -534,7 +534,6 @@ def run_server(
ssl_certfile=ssl_certfile_path, ssl_certfile=ssl_certfile_path,
) # run uvicorn ) # run uvicorn
else: else:
print(f"litellm.json_logs: {litellm.json_logs}")
if litellm.json_logs: if litellm.json_logs:
from litellm.proxy._logging import logger from litellm.proxy._logging import logger