fix: linting fixes

This commit is contained in:
Krrish Dholakia 2024-07-03 21:55:00 -07:00
parent 17869fc5e9
commit 19c982d0f9
5 changed files with 62 additions and 16 deletions

View file

@ -426,13 +426,22 @@ class Logging:
self.model_call_details["additional_args"] = additional_args
self.model_call_details["log_event_type"] = "post_api_call"
verbose_logger.debug(
"RAW RESPONSE:\n{}\n\n".format(
self.model_call_details.get(
"original_response", self.model_call_details
if json_logs:
verbose_logger.debug(
"RAW RESPONSE:\n{}\n\n".format(
self.model_call_details.get(
"original_response", self.model_call_details
)
),
)
else:
print_verbose(
"RAW RESPONSE:\n{}\n\n".format(
self.model_call_details.get(
"original_response", self.model_call_details
)
)
),
)
)
if self.logger_fn and callable(self.logger_fn):
try:
self.logger_fn(