From 2da50087b0786c3c0b9b01893df717e8a63cd57b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Dec 2023 15:05:47 -0800 Subject: [PATCH] (fix) logging - better prints for async logger --- litellm/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index a56c06304..f2c363fb4 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -993,11 +993,12 @@ class Logging: Implementing async callbacks, to handle asyncio event loop issues when custom integrations need to use async functions. """ start_time, end_time, result, complete_streaming_response = self._success_handler_helper_fn(start_time=start_time, end_time=end_time, result=result) - print_verbose(f"success callbacks: {litellm.success_callback}") + print_verbose(f"Async success callbacks: {litellm._async_success_callback}") for callback in litellm._async_success_callback: try: if isinstance(callback, CustomLogger): # custom logger class + print_verbose(f"Async success callbacks: CustomLogger") await callback.async_log_success_event( kwargs=self.model_call_details, response_obj=result, @@ -1005,6 +1006,7 @@ class Logging: end_time=end_time, ) if callable(callback): # custom logger functions + print_verbose(f"Async success callbacks: async_log_event") await customLogger.async_log_event( kwargs=self.model_call_details, response_obj=result,