diff --git a/litellm/router_utils/fallback_event_handlers.py b/litellm/router_utils/fallback_event_handlers.py index bd9941d07b..98d9cd92de 100644 --- a/litellm/router_utils/fallback_event_handlers.py +++ b/litellm/router_utils/fallback_event_handlers.py @@ -9,7 +9,9 @@ async def log_success_fallback_event(original_model_group: str, kwargs: dict): for _callback in litellm.callbacks: if isinstance(_callback, CustomLogger): try: - await _callback.log_success_fallback_event() + await _callback.log_success_fallback_event( + original_model_group=original_model_group, kwargs=kwargs + ) except Exception as e: verbose_router_logger.error( f"Error in log_success_fallback_event: {(str(e))}" @@ -21,7 +23,9 @@ async def log_failure_fallback_event(original_model_group: str, kwargs: dict): for _callback in litellm.callbacks: if isinstance(_callback, CustomLogger): try: - await _callback.log_failure_fallback_event() + await _callback.log_failure_fallback_event( + original_model_group=original_model_group, kwargs=kwargs + ) except Exception as e: verbose_router_logger.error( f"Error in log_failure_fallback_event: {(str(e))}"