diff --git a/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py b/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py index 7d026c1ce25..bc28fc0b6ca 100644 --- a/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py +++ b/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py @@ -265,6 +265,11 @@ class TestPerSecondCostUsesActualDuration: model=model, optional_params={}, litellm_params={}, + # Bypass the explicit per-provider branches in cost_per_token (e.g. + # the OpenAI one, which unprefixed model names resolve to by + # default) so the lookup reaches the generic input_cost_per_second + # fallback this test is exercising. + custom_llm_provider="custom", ) return logging_obj