scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(tests): stop patching an SDK internal in the discovery regression test

Browse source 
TQ008 flagged patching litellm.proxy.auth.model_checks.get_valid_models.
Assert the gate's own return value instead of mocking past it.
This commit is contained in:
mayuriphad 2026-08-28 10:05:55 +05:30
parent 0ac1c45fdf
commit 3d7d8f69f4
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
tests/test_litellm/proxy/auth/test_model_checks.py
View file

@ -809,6 +809,11 @@ def test_get_provider_models_admits_providers_without_a_static_catalog():
"""litellm_proxy and hosted_vllm have no entry in litellm.models_by_provider """litellm_proxy and hosted_vllm have no entry in litellm.models_by_provider
(their model list only exists behind the provider's own endpoint), so the (their model list only exists behind the provider's own endpoint), so the
static-dict gate must not reject them before endpoint discovery runs. static-dict gate must not reject them before endpoint discovery runs.
Regression check only, not a discovery test: with check_provider_endpoint
left at its default (off), get_valid_models never reaches the network and
falls back to models_by_provider.get(provider, []) -- an empty list, not
None. Before the fix, the gate itself returned None for these providers.
""" """
import litellm import litellm
from litellm.proxy.auth.model_checks import get_provider_models from litellm.proxy.auth.model_checks import get_provider_models
@ -817,21 +822,16 @@ def test_get_provider_models_admits_providers_without_a_static_catalog():
assert "litellm_proxy" not in litellm.models_by_provider assert "litellm_proxy" not in litellm.models_by_provider
assert "hosted_vllm" not in litellm.models_by_provider assert "hosted_vllm" not in litellm.models_by_provider
with patch( result = get_provider_models(
"litellm.proxy.auth.model_checks.get_valid_models", "litellm_proxy",
return_value=["litellm_proxy/gpt-4o"], litellm_params=LiteLLM_Params(
) as mock_get_valid_models: model="litellm_proxy/*",
result = get_provider_models( api_base="http://upstream:4000",
"litellm_proxy", api_key="sk-upstream",
litellm_params=LiteLLM_Params( ),
model="litellm_proxy/*", )
api_base="http://upstream:4000",
api_key="sk-upstream",
),
)
assert result == ["litellm_proxy/gpt-4o"] assert result == []
mock_get_valid_models.assert_called_once()
def test_get_provider_models_returns_none_for_an_unknown_provider(): def test_get_provider_models_returns_none_for_an_unknown_provider():