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

Merge pull request #41975 from BerriAI/litellm_fix_azure_snapshot_fallback_test

Browse source 
test(utils): isolate dated model fallback from pricing additions
This commit is contained in:
joshua-berri 2026-09-19 16:20:51 +00:00 • committed by GitHub
commit f08b787685
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/test_litellm/test_utils.py
View file

@ -191,15 +191,32 @@ def test_get_model_info_strips_openai_finetune_ids_without_a_custom_suffix(local
],
)
def test_get_model_info_falls_back_from_dated_snapshot_to_undated_entry(
local_model_cost_map, model, custom_llm_provider, expected_key
):
info = litellm.get_model_info(model=model, custom_llm_provider=custom_llm_provider)
local_model_cost_map: None,
monkeypatch: pytest.MonkeyPatch,
model: str,
custom_llm_provider: str,
expected_key: str,
) -> None:
monkeypatch.delitem(litellm.model_cost, model, raising=False)
monkeypatch.delitem(litellm.model_cost, f"{custom_llm_provider}/{model}", raising=False)
assert expected_key in litellm.model_cost
info: Final = litellm.get_model_info(model=model, custom_llm_provider=custom_llm_provider)
assert info["key"] == expected_key
def test_get_model_info_prefers_exact_dated_key_over_stripped(local_model_cost_map):
info = litellm.get_model_info(model="gpt-4o-2024-08-06", custom_llm_provider="openai")
assert info["key"] == "gpt-4o-2024-08-06"
@pytest.mark.parametrize(
("model", "custom_llm_provider", "expected_key"),
[
("gpt-4o-2024-08-06", "openai", "gpt-4o-2024-08-06"),
("gpt-5.6-luna-2026-07-09", "azure", "azure/gpt-5.6-luna-2026-07-09"),
],
)
def test_get_model_info_prefers_exact_dated_key_over_stripped(
local_model_cost_map: None, model: str, custom_llm_provider: str, expected_key: str
) -> None:
assert expected_key in litellm.model_cost
info: Final = litellm.get_model_info(model=model, custom_llm_provider=custom_llm_provider)
assert info["key"] == expected_key
def test_check_provider_match_azure_ai_allows_openai_and_azure():