scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

fix(utils): custom_provider_map lookups take the last entry for a duplicated provider

Browse source
This commit is contained in:
devin-ai-integration[bot] 2026-11-03 18:02:52 +00:00
parent 03c26c6e07
commit cab3a594e7
2 changed files with 29 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -11226,7 +11226,7 @@ class ProviderConfigManager:
return next(
(
item["custom_handler"]
for item in litellm.custom_provider_map
for item in reversed(litellm.custom_provider_map)
if item["provider"] == provider and isinstance(item["custom_handler"], CustomLLM)
),
None,
@ -11240,7 +11240,7 @@ class ProviderConfigManager:
return next(
(
item["custom_handler"]
for item in litellm.custom_provider_map
for item in reversed(litellm.custom_provider_map)
if item["provider"] == provider and isinstance(item["custom_handler"], BaseVideoConfig)
),
None,

27
tests/unit/test_utils.py
View file

@ -1727,6 +1727,33 @@ def test_get_provider_video_config_resolves_a_custom_provider_map_base_video_con
assert ProviderConfigManager.get_provider_video_config(model="seedance-1", provider="not-registered") is None
def test_custom_provider_map_lookups_take_the_last_entry_for_a_duplicated_provider(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""A provider listed twice in `custom_provider_map` resolves to its last entry, as the per-call loops did
before the lookup was shared."""
from litellm.llms.custom_llm import CustomLLM
from litellm.llms.openai.videos.transformation import OpenAIVideoConfig
first_llm: Final = CustomLLM()
last_llm: Final = CustomLLM()
first_video: Final = OpenAIVideoConfig()
last_video: Final = OpenAIVideoConfig()
monkeypatch.setattr(
litellm,
"custom_provider_map",
[
{"provider": "dup", "custom_handler": first_llm},
{"provider": "dup", "custom_handler": first_video},
{"provider": "dup", "custom_handler": last_llm},
{"provider": "dup", "custom_handler": last_video},
],
)
assert ProviderConfigManager.get_custom_llm_handler("dup") is last_llm
assert ProviderConfigManager.get_custom_provider_video_config("dup") is last_video
def test_get_provider_text_to_speech_config_vertex_gemini_skips_cloud_tts():
"""Regression for LIT-6501: mapping vertex Gemini TTS params through Google Cloud TTS
dropped response_format before the speech_to_completion bridge could honor it."""