scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-23 05:34:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 24

test(router): use real Logging in fallback cooldown test

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-10-17 18:40:54 +00:00
parent ae550e90b6
commit aa5918c0cb
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/router_utils/test_fallback_event_handlers.py
View file

@ -1224,9 +1224,9 @@ class TestRunAsyncFallbackTriggersCooldown:
async def async_function_with_fallbacks(self, *args, **kwargs):
raise RuntimeError("fallback model also failed")
def _logging_obj(self, has_logged_async_failure: bool) -> MagicMock:
logging_obj = MagicMock()
logging_obj.model_call_details = {"has_logged_async_failure": has_logged_async_failure}
def _logging_obj(self, has_logged_async_failure: bool) -> Logging:
logging_obj: Final = _request_logging_obj()
logging_obj.model_call_details["has_logged_async_failure"] = has_logged_async_failure
return logging_obj
@pytest.mark.asyncio