scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 134

test(mock_completion): cover the provider inference fallback for direct calls

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-17 18:07:58 +00:00
parent c2dd7bd98a
commit 3cf42f6565
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_litellm/test_main.py
View file

@ -2472,6 +2472,21 @@ async def test_mock_completion_prices_azure_ai_router_deployment_with_custom_pri
assert response._hidden_params["custom_llm_provider"] == "azure_ai"
@pytest.mark.parametrize(
("model", "expected_provider"),
(("anthropic/claude-sonnet-5", "anthropic"), ("no-such-provider-model", None)),
)
def test_mock_completion_infers_provider_when_called_directly_without_one(model: str, expected_provider: str | None):
response: Final = litellm.mock_completion(
model=model,
messages=[{"role": "user", "content": "hello"}],
mock_response="ok",
)
assert response.choices[0].message.content == "ok"
assert response._hidden_params.get("custom_llm_provider") == expected_provider
_ADMISSION_INPUT_TOKENS: Final = 51234