diff --git a/tests/test_litellm/proxy/test_proxy_cli.py b/tests/test_litellm/proxy/test_proxy_cli.py index f52f14b860..12065ad5b4 100644 --- a/tests/test_litellm/proxy/test_proxy_cli.py +++ b/tests/test_litellm/proxy/test_proxy_cli.py @@ -103,7 +103,11 @@ class TestProxyInitializationHelpers: args = ProxyInitializationHelpers._get_default_unvicorn_init_args( "localhost", 8000 ) - assert args["log_config"] is None + # When json_logs is True, log_config should be set to the JSON log config dict + assert args["log_config"] is not None + assert isinstance(args["log_config"], dict) + assert "version" in args["log_config"] + assert "formatters" in args["log_config"] # Test with keepalive_timeout args = ProxyInitializationHelpers._get_default_unvicorn_init_args( @@ -218,6 +222,7 @@ class TestProxyInitializationHelpers: @patch("atexit.register") # 🔥 critical def test_skip_server_startup(self, mock_atexit_register, mock_uvicorn_run): from click.testing import CliRunner + from litellm.proxy.proxy_cli import run_server runner = CliRunner()