From 199a0f82fe9e8dd9e88708039700cd3c308fa32b Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 8 Jul 2026 17:33:40 +0000 Subject: [PATCH] refactor(register_model): copy base_model cost entry to match resolver contract (cherry picked from commit 136a4b39fdb11fa93ad841765225d68e77ef1d56) --- litellm/utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 2830a9e0abf..97d621b4e6e 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2701,9 +2701,10 @@ def register_model(model_cost: Union[str, dict]): if builtin_entry is None: base_model = value.get("base_model") if isinstance(base_model, str) and base_model: - builtin_entry = _resolve_builtin_model_cost_entry( - key=base_model, provider=provider - ) or litellm.model_cost.get(base_model) + builtin_entry = _resolve_builtin_model_cost_entry(key=base_model, provider=provider) + if builtin_entry is None: + base_model_cost = litellm.model_cost.get(base_model) + builtin_entry = dict(base_model_cost) if base_model_cost is not None else None if builtin_entry is not None: for field in _CACHE_PRICING_FIELDS: if value.get(field) is None and builtin_entry.get(field) is not None: