From e37092aab65e4b99b07395eadd86c31d3cb5f3ea Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 17 Jun 2024 19:45:22 -0700 Subject: [PATCH] use older logfire logging --- litellm/litellm_core_utils/litellm_logging.py | 39 +++++++++---------- 1 file changed, 19 insertions(+), 20 deletions(-) diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index ee3e4bfbb..71cc8e6ab 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -666,7 +666,6 @@ class Logging: end_time=end_time, print_verbose=print_verbose, ) - if callback == "logfire": global logfireLogger verbose_logger.debug("reaches logfire for success logging!") @@ -1473,25 +1472,6 @@ class Logging: end_time=end_time, print_verbose=print_verbose, ) - - if callback == "logfire": - verbose_logger.debug("reaches logfire for failure logging!") - kwargs = {} - for k, v in self.model_call_details.items(): - if ( - k != "original_response" - ): # copy.deepcopy raises errors as this could be a coroutine - kwargs[k] = v - kwargs["exception"] = exception - - logfireLogger.log_event( - kwargs=kwargs, - response_obj=result, - start_time=start_time, - end_time=end_time, - level=LogfireLevel.ERROR.value, - print_verbose=print_verbose, - ) if callback == "sentry": print_verbose("sending exception to sentry") if capture_exception: @@ -1616,6 +1596,25 @@ class Logging: print_verbose=print_verbose, ) + if callback == "logfire": + verbose_logger.debug("reaches logfire for failure logging!") + kwargs = {} + for k, v in self.model_call_details.items(): + if ( + k != "original_response" + ): # copy.deepcopy raises errors as this could be a coroutine + kwargs[k] = v + kwargs["exception"] = exception + + logfireLogger.log_event( + kwargs=kwargs, + response_obj=result, + start_time=start_time, + end_time=end_time, + level=LogfireLevel.ERROR.value, + print_verbose=print_verbose, + ) + except Exception as e: print_verbose( f"LiteLLM.LoggingError: [Non-Blocking] Exception occurred while failure logging with integrations {str(e)}"