From fd60471a84ecb8b946d0f3f778ede5c7801da38d Mon Sep 17 00:00:00 2001 From: Eduardo Pessin Date: Thu, 1 Oct 2026 19:27:43 +0100 Subject: [PATCH] Pass completion_response_model as str or None to _priced_provider_response_model basedpyright counted the bare variable as partially unknown (reportUnknownArgumentType +1 over the budget). --- litellm/cost_calculator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/cost_calculator.py b/litellm/cost_calculator.py index f85ca0462da..b5c6c3a914f 100644 --- a/litellm/cost_calculator.py +++ b/litellm/cost_calculator.py @@ -888,7 +888,9 @@ def _select_model_name_for_cost_calc( base_model if base_model is not None else _priced_provider_response_model( - provider_response_model, completion_response_model, custom_llm_provider + provider_response_model, + completion_response_model if isinstance(completion_response_model, str) else None, + custom_llm_provider, ) )