From 708e87e2657648cde93bfd8f192de2ec972de527 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 1 Jul 2024 21:11:13 -0700 Subject: [PATCH] fix(litellm_logging.py): add debug statements to logging --- litellm/litellm_core_utils/litellm_logging.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index add281e43f4..3c32502777e 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -1354,11 +1354,16 @@ class Logging: end_time=end_time, ) if isinstance(callback, CustomLogger): # custom logger class - if self.stream == True: + if self.stream is True: if ( "async_complete_streaming_response" in self.model_call_details ): + print_verbose( + "Enters async log success event for streaming call for callback={}".format( + callback + ) + ) await callback.async_log_success_event( kwargs=self.model_call_details, response_obj=self.model_call_details[ @@ -1367,6 +1372,11 @@ class Logging: start_time=start_time, end_time=end_time, ) + print_verbose( + "Successfully logs async success event for streaming call for callback={}".format( + callback + ) + ) else: await callback.async_log_stream_event( # [TODO]: move this to being an async log stream event function kwargs=self.model_call_details,