From 8adbf3562389f3fa73dd2acd40878c0d81051424 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 6 Dec 2023 17:15:27 -0800 Subject: [PATCH] (feat) add async loggers under custom logger --- litellm/utils.py | 22 ++++++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 08c1066686e..0b2f5243d0a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -995,6 +995,13 @@ class Logging: for callback in litellm._async_success_callback: try: + if isinstance(callback, CustomLogger): # custom logger class + await callback.async_log_success_event( + kwargs=self.model_call_details, + response_obj=result, + start_time=start_time, + end_time=end_time, + ) if callable(callback): # custom logger functions await customLogger.async_log_event( kwargs=self.model_call_details, @@ -1129,15 +1136,22 @@ class Logging: for callback in litellm._async_failure_callback: try: + if isinstance(callback, CustomLogger): # custom logger class + await callback.async_log_failure_event( + kwargs=self.model_call_details, + response_obj=result, + start_time=start_time, + end_time=end_time, + ) if callable(callback): # custom logger functions - await customLogger.async_log_failure_event( + await customLogger.async_log_event( kwargs=self.model_call_details, response_obj=result, start_time=start_time, end_time=end_time, print_verbose=print_verbose, callback_func=callback - ) + ) except: print_verbose( f"LiteLLM.LoggingError: [Non-Blocking] Exception occurred while success logging {traceback.format_exc()}" @@ -1239,6 +1253,10 @@ def client(original_function): litellm.success_callback.append(callback) if callback not in litellm.failure_callback: litellm.failure_callback.append(callback) + if callback not in litellm._async_success_callback: + litellm._async_success_callback.append(callback) + if callback not in litellm._async_failure_callback: + litellm._async_failure_callback.append(callback) if ( len(litellm.input_callback) > 0 or len(litellm.success_callback) > 0