ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tests): update tests to use models still present in pricing JSON

Browse Source 
Replace removed deprecated models (claude-3-5-sonnet-20241022,
claude-3-5-haiku-20241022, claude-3-5-haiku-latest) with current
models in web_search and cost calculation tests.
This commit is contained in:
Chesars 2026-03-11 14:50:47 -03:00
parent 45ca8b5b2b
commit d81d751af0
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
View File

@ -354,7 +354,7 @@ def test_generic_cost_per_token_anthropic_prompt_caching():
def test_generic_cost_per_token_anthropic_prompt_caching_with_cache_creation():
model = "claude-3-5-haiku-20241022"
model = "claude-haiku-4-5-20251001"
usage = Usage(
completion_tokens=90,
prompt_tokens=28436,
@ -379,7 +379,7 @@ def test_generic_cost_per_token_anthropic_prompt_caching_with_cache_creation():
)
print(f"prompt_cost: {prompt_cost}")
assert round(prompt_cost, 3) == 0.023
assert round(prompt_cost, 3) == 0.029
def test_string_cost_values():

4
tests/test_litellm/test_utils.py
View File

@ -444,9 +444,7 @@ def test_anthropic_web_search_in_model_info():
supported_models = [
"anthropic/claude-4-sonnet-20250514",
"anthropic/claude-sonnet-4-5-20250929",
"anthropic/claude-3-5-sonnet-20241022",
"anthropic/claude-3-5-haiku-20241022",
"anthropic/claude-3-5-haiku-latest",
"anthropic/claude-3-7-sonnet-20250219",
]
for model in supported_models:
from litellm.utils import get_model_info