scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

Fix test_completion_cost_prompt_caching gemini parametrization

Browse source 
gemini/gemini-2.5-flash lacks cache_creation_input_token_cost in the
model cost map, causing a TypeError when the test multiplies
cache_creation_input_tokens by None. Use claude-haiku-4-5 instead,
which has the required prompt caching cost fields.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-03-11 17:12:15 -07:00
parent c9f7075690
commit 82de82f1b6
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_completion_cost.py
View file

@ -1013,7 +1013,7 @@ def test_completion_cost_azure_common_deployment_name():
"model, custom_llm_provider",
[
("claude-sonnet-4-6", "anthropic"),
("gemini/gemini-2.5-flash", "gemini"),
("claude-haiku-4-5", "anthropic"),
],
)
def test_completion_cost_prompt_caching(model, custom_llm_provider):