From 31a4cb65bf7e56a913de8c9d8fc199402df5316d Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 24 Jan 2026 12:59:51 -0800 Subject: [PATCH] test_get_default_unvicorn_init_args --- tests/test_litellm/proxy/test_proxy_cli.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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()