scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

test(proxy): give pre-call mocks real router_settings and fallbacks

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-10 17:46:44 +00:00
parent c064e576ee
commit 431dcce6a7
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/common_request_processing.py
View file

@ -2121,14 +2121,14 @@ class ProxyBaseLLMRequestProcessing:
fallbacks = None
key_router_settings: Final = getattr(user_api_key_dict, "router_settings", None)
key_router_settings: Final = user_api_key_dict.router_settings
if isinstance(key_router_settings, dict) and "fallbacks" in key_router_settings:
fallbacks = key_router_settings["fallbacks"]
if fallbacks is None:
fallbacks = llm_router.fallbacks
if not isinstance(fallbacks, list) or not fallbacks:
if not fallbacks:
return None
fallback_model_group, generic_fallback_idx = get_fallback_model_group(

8
tests/proxy_unit_tests/test_response_polling_pre_call_checks.py
View file

@ -48,10 +48,10 @@ class TestSkipPreCallLogic:
await processor.base_process_llm_request(
request=MagicMock(spec=Request),
fastapi_response=MagicMock(spec=Response),
user_api_key_dict=MagicMock(spec=UserAPIKeyAuth),
user_api_key_dict=MagicMock(spec=UserAPIKeyAuth, router_settings=None),
route_type="aresponses",
proxy_logging_obj=mock_proxy_logging,
llm_router=MagicMock(),
llm_router=MagicMock(fallbacks=None),
general_settings={},
proxy_config=MagicMock(),
skip_pre_call_logic=True,
@ -87,10 +87,10 @@ class TestSkipPreCallLogic:
await processor.base_process_llm_request(
request=MagicMock(spec=Request),
fastapi_response=MagicMock(spec=Response),
user_api_key_dict=MagicMock(spec=UserAPIKeyAuth),
user_api_key_dict=MagicMock(spec=UserAPIKeyAuth, router_settings=None),
route_type="aresponses",
proxy_logging_obj=mock_proxy_logging,
llm_router=MagicMock(),
llm_router=MagicMock(fallbacks=None),
general_settings={},
proxy_config=MagicMock(),
)