phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): log failure to sync failure callbacks as well

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-05 14:49:34 -07:00
parent b1b21b0340
commit fd11f63f71
2 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_proxy_reject_logging.py
View file

@ -51,7 +51,8 @@ from litellm.router import Router
class testLogger(CustomLogger):
def __init__(self):
self.reaches_failure_event = False
self.reaches_sync_failure_event = False
self.reaches_async_failure_event = False
async def async_pre_call_hook(
self,
@ -72,10 +73,10 @@ class testLogger(CustomLogger):
)
async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time):
self.reaches_async_failure_event = True
def log_failure_event(self, kwargs, response_obj, start_time, end_time):
self.reaches_failure_event = True
return await super().async_log_failure_event(
kwargs, response_obj, start_time, end_time
)
router = Router(
@ -184,4 +185,5 @@ async def test_chat_completion_request_with_redaction(route, body):
pass
await asyncio.sleep(3)
assert _test_logger.reaches_failure_event is True
assert _test_logger.reaches_async_failure_event is True
assert _test_logger.reaches_sync_failure_event is True