From 6f46550c183dc40b12a8913260339852c8b825f4 Mon Sep 17 00:00:00 2001 From: DennyHo0917 Date: Mon, 7 Sep 2026 18:00:51 +0800 Subject: [PATCH] test: avoid private API route config access --- tests/test_litellm/llms/api_route/test_api_route.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/llms/api_route/test_api_route.py b/tests/test_litellm/llms/api_route/test_api_route.py index e476fbfd54b..218f0590711 100644 --- a/tests/test_litellm/llms/api_route/test_api_route.py +++ b/tests/test_litellm/llms/api_route/test_api_route.py @@ -40,9 +40,11 @@ def test_api_route_credentials_from_environment(monkeypatch): def test_api_route_uses_default_base_url(monkeypatch): monkeypatch.delenv("API_ROUTE_BASE_URL", raising=False) - api_base, api_key = APIRouteChatConfig()._get_openai_compatible_provider_info( - api_base=None, + _, _, api_key, api_base = get_llm_provider( + model="api_route/model-name", + custom_llm_provider=None, api_key="test-key", + api_base=None, ) assert api_base == "https://global.api-route.com/v1" @@ -52,9 +54,11 @@ def test_api_route_uses_default_base_url(monkeypatch): def test_api_route_explicit_api_base_overrides_default(monkeypatch): monkeypatch.delenv("API_ROUTE_BASE_URL", raising=False) - api_base, _ = APIRouteChatConfig()._get_openai_compatible_provider_info( - api_base="https://custom.example.com/v1", + _, _, _, api_base = get_llm_provider( + model="api_route/model-name", + custom_llm_provider=None, api_key="test-key", + api_base="https://custom.example.com/v1", ) assert api_base == "https://custom.example.com/v1"