From 0d3e52373c9312e4f0e2be034dd4a088c73d02b3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 5 Jun 2024 17:42:49 -0700 Subject: [PATCH] fix(proxy/_logging.py): fix default logging level --- litellm/_logging.py | 15 +++++++++++---- litellm/proxy/_logging.py | 2 +- litellm/proxy/_super_secret_config.yaml | 1 + litellm/proxy/proxy_cli.py | 1 - 4 files changed, 13 insertions(+), 6 deletions(-) diff --git a/litellm/_logging.py b/litellm/_logging.py index a8121d9a88..1ff6e45ddb 100644 --- a/litellm/_logging.py +++ b/litellm/_logging.py @@ -41,12 +41,19 @@ verbose_logger.addHandler(handler) def _turn_on_json(): handler = logging.StreamHandler() - handler.setLevel(logging.DEBUG) handler.setFormatter(JsonFormatter()) - verbose_router_logger.addHandler(handler) - verbose_proxy_logger.addHandler(handler) - verbose_logger.addHandler(handler) + # Define a list of the loggers to update + loggers = [verbose_router_logger, verbose_proxy_logger, verbose_logger] + + # 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(): diff --git a/litellm/proxy/_logging.py b/litellm/proxy/_logging.py index fcabad7cd6..22cbd88cb7 100644 --- a/litellm/proxy/_logging.py +++ b/litellm/proxy/_logging.py @@ -17,4 +17,4 @@ logger = logging.root handler = logging.StreamHandler() handler.setFormatter(JsonFormatter()) logger.handlers = [handler] -logger.setLevel(logging.DEBUG) +logger.setLevel(logging.INFO) diff --git a/litellm/proxy/_super_secret_config.yaml b/litellm/proxy/_super_secret_config.yaml index 48d4294cb8..72479bd5d5 100644 --- a/litellm/proxy/_super_secret_config.yaml +++ b/litellm/proxy/_super_secret_config.yaml @@ -56,6 +56,7 @@ router_settings: litellm_settings: success_callback: ["langfuse"] + json_logs: true general_settings: alerting: ["email"] diff --git a/litellm/proxy/proxy_cli.py b/litellm/proxy/proxy_cli.py index 0f62a98d3b..2960d9b1c1 100644 --- a/litellm/proxy/proxy_cli.py +++ b/litellm/proxy/proxy_cli.py @@ -534,7 +534,6 @@ def run_server( ssl_certfile=ssl_certfile_path, ) # run uvicorn else: - print(f"litellm.json_logs: {litellm.json_logs}") if litellm.json_logs: from litellm.proxy._logging import logger