From 82de82f1b6e495ae19d3b55d650116f9aeacfc4b Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Wed, 11 Mar 2026 17:12:15 -0700 Subject: [PATCH] Fix test_completion_cost_prompt_caching gemini parametrization 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 --- tests/local_testing/test_completion_cost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index 96335908cd0..dd060a56d20 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -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):