scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 280

test(router): cover resolve_deployment_provider accept and reject paths

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Moe Khalil 2026-10-26 21:31:46 +00:00
parent c280401542
commit 08220e7c84
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/unit/test_router/test_router.py
View file

@ -30149,3 +30149,24 @@ def test_request_carries_image_sees_chat_parts_and_responses_input_items():
assert litellm.Router._request_carries_image(None, {"input": _IMAGE_RESPONSES_INPUT}) is True
assert litellm.Router._request_carries_image([{"role": "user", "content": "Say hi"}], {}) is False
assert litellm.Router._request_carries_image(None, {"input": "Say hi"}) is False
@pytest.mark.parametrize(
("model", "expected_provider"),
[("cloudflare/@cf/cloudflare/clef-flash", "cloudflare"), ("dotprompt/clef", None)],
)
def test_resolve_deployment_provider_accepts_what_the_router_loads(model: str, expected_provider: str | None):
resolved: Final = Router.resolve_deployment_provider(LiteLLM_Params(model=model))
assert resolved[1] == expected_provider, resolved
@pytest.mark.parametrize(
"params",
[
LiteLLM_Params(model="@cf/cloudflare/clef-flash"),
LiteLLM_Params(model="my-model", custom_llm_provider="not-a-provider"),
],
)
def test_resolve_deployment_provider_rejects_what_the_router_drops(params: LiteLLM_Params):
with pytest.raises(Exception, match=r"LLM Provider NOT provided|Unsupported provider"):
Router.resolve_deployment_provider(params)