fix(utils.py): fix logging integrations

This commit is contained in:
Krrish Dholakia 2023-11-09 17:09:49 -08:00
parent 1affa89ad7
commit cfd2ccf429

View file

@ -479,11 +479,12 @@ class Logging:
) )
elif isinstance(callback, CustomLogger): # custom logger class elif isinstance(callback, CustomLogger): # custom logger class
callback.log_post_api_call( callback.log_post_api_call(
model=self.model,
messages=self.messages,
kwargs=self.model_call_details, kwargs=self.model_call_details,
response_obj=None,
start_time=self.start_time,
end_time=None
) )
except: except Exception as e:
print_verbose( print_verbose(
f"LiteLLM.LoggingError: [Non-Blocking] Exception occurred while post-call logging with integrations {traceback.format_exc()}" f"LiteLLM.LoggingError: [Non-Blocking] Exception occurred while post-call logging with integrations {traceback.format_exc()}"
) )