scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 188

fix: correct test_no_double_strip_on_second_call assertions

Browse source 
The second get_llm_provider call on an already-resolved model like
openrouter/aurora-alpha correctly strips the openrouter/ prefix to yield
the bare model ID (aurora-alpha) — not the prefixed form. Update the
parametrize signature to use separate expected_first/expected_second values
and fix the assertions accordingly, with an explanatory docstring.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
tombii 2026-02-28 21:38:08 +01:00
parent d292da2c14
commit 73de1acdc7
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/test_litellm/llms/openrouter/test_openrouter_provider_routing.py
View file

@ -44,20 +44,31 @@ class TestOpenRouterNativeModelRouting:
assert result_model == expected_model
@pytest.mark.parametrize(
"input_model,expected_model",
"input_model,expected_first,expected_second",
[
("openrouter/openrouter/aurora-alpha", "openrouter/aurora-alpha"),
("openrouter/openrouter/auto", "openrouter/auto"),
# After the first call strips outer prefix: openrouter/openrouter/aurora-alpha
# → openrouter/aurora-alpha. A second call on that result splits at the
# first "/" giving provider=openrouter, model=aurora-alpha — which is the
# correct model ID to send to the OpenRouter API.
("openrouter/openrouter/aurora-alpha", "openrouter/aurora-alpha", "aurora-alpha"),
("openrouter/openrouter/auto", "openrouter/auto", "auto"),
],
)
def test_no_double_strip_on_second_call(self, input_model, expected_model):
"""Simulates two consecutive get_llm_provider calls (bridge → completion)."""
def test_no_double_strip_on_second_call(self, input_model, expected_first, expected_second):
"""Simulates two consecutive get_llm_provider calls (bridge → completion).
The first call (bridge) converts openrouter/openrouter/<model> to
openrouter/<model>. The second call (completion) further strips the
remaining openrouter/ provider prefix and returns <model> — the bare
model ID that should be sent to the OpenRouter API.
"""
model_first, provider, _, _ = litellm.get_llm_provider(model=input_model)
assert model_first == expected_model
assert provider == "openrouter"
assert model_first == expected_first
model_second, provider2, _, _ = litellm.get_llm_provider(model=model_first)
assert provider2 == "openrouter"
assert model_second == expected_model
assert model_second == expected_second
@pytest.mark.parametrize(
"input_model,expected_model",