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 74

Merge pull request #41971 from BerriAI/litellm_fix_dated_snapshot_test_exact_key

Browse source 
test(cost): point dated snapshot tests at a date the cost map cannot carry
This commit is contained in:
Mateo Wang 2026-09-19 08:39:20 -07:00 • committed by GitHub
commit 638e085dba
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/test_cost_calculator.py
View file

@ -113,7 +113,7 @@ def test_completion_cost_uses_response_model_for_dynamic_routing(_local_model_co
def test_completion_cost_strips_dated_azure_snapshot_model(_local_model_cost_map: None) -> None:
dated_response = ModelResponse(
model="gpt-5.6-luna-2026-07-09",
model="gpt-5.6-luna-2099-01-01",
choices=[Choices(index=0, message=Message(role="assistant", content="hi"), finish_reason="stop")],
usage=Usage(prompt_tokens=100, completion_tokens=50, total_tokens=150),
)

4
tests/test_litellm/test_utils.py
View file

@ -186,8 +186,8 @@ def test_get_model_info_strips_openai_finetune_ids_without_a_custom_suffix(local
@pytest.mark.parametrize(
("model", "custom_llm_provider", "expected_key"),
[
("gpt-5.6-luna-2026-07-09", "openai", "gpt-5.6-luna"),
("gpt-5.6-luna-2026-07-09", "azure", "azure/gpt-5.6-luna"),
("gpt-5.6-luna-2099-01-01", "openai", "gpt-5.6-luna"),
("gpt-5.6-luna-2099-01-01", "azure", "azure/gpt-5.6-luna"),
],
)
def test_get_model_info_falls_back_from_dated_snapshot_to_undated_entry(