diff --git a/tests/unit/proxy/test_bug_report_config.py b/tests/unit/proxy/test_bug_report_config.py index 6cffa55781e..6136a65186f 100644 --- a/tests/unit/proxy/test_bug_report_config.py +++ b/tests/unit/proxy/test_bug_report_config.py @@ -5,6 +5,7 @@ from collections.abc import Iterator, Mapping import pytest from litellm.proxy import proxy_server +from litellm.proxy._types import ProxyRuntimeConfig from litellm.proxy.bug_report_config import build_proxy_bug_report, build_proxy_environment_report, safe_config_lines CUSTOMER_STRINGS = ( @@ -183,7 +184,7 @@ def test_malformed_sections_produce_no_lines(): @pytest.fixture def loaded_proxy_config(monkeypatch: pytest.MonkeyPatch) -> Iterator[None]: previous_config = proxy_server.proxy_config.get_config_state() - proxy_server.proxy_config.update_config_state(config=CUSTOMER_CONFIG) + proxy_server.proxy_config.update_config_state(config=ProxyRuntimeConfig.from_resolved(CUSTOMER_CONFIG)) monkeypatch.setattr(proxy_server, "general_settings", dict(CUSTOMER_GENERAL_SETTINGS)) yield proxy_server.proxy_config.update_config_state(config=previous_config) diff --git a/tests/unit/proxy/test_proxy_cli.py b/tests/unit/proxy/test_proxy_cli.py index 1c3360f0cca..71d9ada7351 100644 --- a/tests/unit/proxy/test_proxy_cli.py +++ b/tests/unit/proxy/test_proxy_cli.py @@ -1259,7 +1259,7 @@ class TestProxyInitializationHelpers: save_worker_config=MagicMock(), ) mock_proxy_module.ProxyConfig.return_value.get_config = AsyncMock( - return_value={"general_settings": general_settings} + return_value=ProxyRuntimeConfig.from_resolved({"general_settings": general_settings}) ) clean_env = { @@ -1334,12 +1334,12 @@ class TestProxyInitializationHelpers: save_worker_config=MagicMock(), ) mock_proxy_module.ProxyConfig.return_value.get_config = AsyncMock( - return_value={ + return_value=ProxyRuntimeConfig.from_resolved({ "general_settings": { "database_url": "postgresql://test:test@localhost:5432/test", "database_disable_prepared_statements": True, } - } + }) ) clean_env = {