scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(router): cover xai-oauth alias helpers for router coverage gate

Browse source
This commit is contained in:
Devin AI 2026-07-10 23:23:12 +00:00
parent 30d98b9d58
commit d3a2cb2b75
1 changed files with 27 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
tests/test_litellm/test_router.py
View file

@ -1111,6 +1111,33 @@ def test_router_leaves_plain_xai_deployment_untouched():
assert not deployment.get("use_xai_oauth")
def test_normalize_xai_oauth_alias_model_helper():
assert litellm.Router._normalize_xai_oauth_alias_model("xai-oauth/grok-4.5", None) == "xai/grok-4.5"
assert litellm.Router._normalize_xai_oauth_alias_model("grok-4.5", "xai_oauth") == "xai/grok-4.5"
assert litellm.Router._normalize_xai_oauth_alias_model("xai/grok-4.5", "xai-oauth") == "xai/grok-4.5"
assert litellm.Router._normalize_xai_oauth_alias_model("xai/grok-4.5", None) is None
assert litellm.Router._normalize_xai_oauth_alias_model("openai/gpt-4o", None) is None
def test_apply_xai_oauth_alias_helper():
router = litellm.Router(
model_list=[
{"model_name": "grok-4.5", "litellm_params": {"model": "xai/grok-4.5"}}
]
)
normalized = router._apply_xai_oauth_alias(
{"model": "grok-4.5", "custom_llm_provider": "xai-oauth", "api_base": "https://x"}
)
assert normalized["model"] == "xai/grok-4.5"
assert normalized["use_xai_oauth"] is True
assert "custom_llm_provider" not in normalized
assert normalized["api_base"] == "https://x"
untouched = {"model": "openai/gpt-4o"}
assert router._apply_xai_oauth_alias(untouched) is untouched
@pytest.mark.asyncio
async def test_router_ageneric_api_call_with_fallbacks_helper():
"""