From 170581e7101ba821a6bd6a2cb039718f899fc85d Mon Sep 17 00:00:00 2001 From: jesus Date: Sat, 12 Sep 2026 01:04:40 +0000 Subject: [PATCH] fix(router): annotate cast for unpriced model info Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/router.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/router.py b/litellm/router.py index 8a543373692..9ff23c597f1 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -10641,7 +10641,9 @@ class Router: litellm_model_name_model_info.get("key") if litellm_model_name_model_info is not None else None ) if cost_map_omits_token_price(model_id, builtin_key, base_model_key): - return cast(ModelInfo, {**model_info, "input_cost_per_token": None, "output_cost_per_token": None}) + return cast( # cast-ok: TypedDict spread with overridden keys loses its type + ModelInfo, {**model_info, "input_cost_per_token": None, "output_cost_per_token": None} + ) return model_info def _set_model_group_info(self, model_group: str, user_facing_model_group_name: str) -> ModelGroupInfo | None: