From b21484d229341ad46869a539028fc12e0228f235 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 15 Jan 2026 16:48:06 -0800 Subject: [PATCH] test fix --- litellm/cost_calculator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/cost_calculator.py b/litellm/cost_calculator.py index 870b97530fa..f18e8d62aa9 100644 --- a/litellm/cost_calculator.py +++ b/litellm/cost_calculator.py @@ -952,7 +952,8 @@ def completion_cost( # noqa: PLR0915 ) potential_model_names = [selected_model, _get_response_model(completion_response)] - + if model is not None: + potential_model_names.append(model) for idx, model in enumerate(potential_model_names): try: