diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py index bcfef0fc2b..586c908192 100644 --- a/litellm/integrations/custom_logger.py +++ b/litellm/integrations/custom_logger.py @@ -28,6 +28,11 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback def log_failure_event(self, kwargs, response_obj, start_time, end_time): pass + async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): + pass + + async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time): + pass #### SINGLE-USE #### - https://docs.litellm.ai/docs/observability/custom_callback#using-your-custom-callback-function @@ -80,24 +85,4 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback except: # traceback.print_exc() print_verbose(f"Custom Logger Error - {traceback.format_exc()}") - pass - - async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time, print_verbose, callback_func): - # Method definition - try: - kwargs["log_event_type"] = "post_api_call" - await callback_func( - kwargs, # kwargs to func - response_obj, - start_time, - end_time, - ) - print_verbose( - f"Custom Logger - final response object: {response_obj}" - ) - except: - # traceback.print_exc() - print_verbose(f"Custom Logger Error - {traceback.format_exc()}") - pass - - + pass \ No newline at end of file