scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

refactor(model_info): return Mapping from fill_missing matchers

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-14 17:35:24 +00:00
parent 5781678477
commit e366d3bd30
1 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/litellm_core_utils/fallback_generalizations.py
View file

@ -53,7 +53,9 @@ O(number of rules); callers must only invoke them on a cache miss.
import logging
import re
from collections.abc import Mapping
from dataclasses import dataclass
from types import MappingProxyType
from typing import Final
verbose_logger: Final = logging.getLogger("LiteLLM")
@ -209,9 +211,7 @@ class _FallbackGeneralizations:
return None
return {key: value for model_info in matched for key, value in model_info.items()}
def match_fill_missing(
self, model: str, provider: str
) -> dict[str, object] | None: # mutable-ok: preserve the existing dict return contract
def match_fill_missing(self, model: str, provider: str) -> Mapping[str, object] | None:
if not model or not provider:
return None
matched = tuple(
@ -221,9 +221,9 @@ class _FallbackGeneralizations:
)
if not matched:
return None
fill_missing: Final[dict[str, object]] = { # mutable-ok: preserve the existing dict merge input
key: value for model_info in matched for key, value in model_info.items() if key != PROVIDER_KEY
}
fill_missing: Final[Mapping[str, object]] = MappingProxyType(
{key: value for model_info in matched for key, value in model_info.items() if key != PROVIDER_KEY}
)
return fill_missing or None
@ -263,9 +263,7 @@ def match_capability_generalizations(model: str) -> dict | None:
return _registry.match_capabilities(model)
def match_fill_missing_generalizations(
model: str, provider: str
) -> dict[str, object] | None: # mutable-ok: preserve the existing dict return contract
def match_fill_missing_generalizations(model: str, provider: str) -> Mapping[str, object] | None:
"""Return flagged capability rules matching ``model`` for ``provider``.
Later rules override earlier ones on key conflicts. Only rules listing