From b19e7dcc5a5a71157a8da1e046a383f2a33fcac3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 9 Nov 2023 17:09:49 -0800 Subject: [PATCH] fix(utils.py): fix logging integrations --- litellm/utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 6714cc7ff12..f1ef8e6c93f 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -479,11 +479,12 @@ class Logging: ) elif isinstance(callback, CustomLogger): # custom logger class callback.log_post_api_call( - model=self.model, - messages=self.messages, kwargs=self.model_call_details, + response_obj=None, + start_time=self.start_time, + end_time=None ) - except: + except Exception as e: print_verbose( f"LiteLLM.LoggingError: [Non-Blocking] Exception occurred while post-call logging with integrations {traceback.format_exc()}" )