scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

fix(router): annotate cast for unpriced model info

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
jesus 2026-09-12 01:04:40 +00:00 • committed by ryan-crabbe-berri
parent 32e6a86319
commit 170581e710
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/router.py
View file

@ -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: