Revert "Revert "fix: Log errors in Traceloop Integration (reverts previous revert)""

This commit is contained in:
Nir Gazit 2024-05-30 04:02:20 +03:00
parent b7fcec8835
commit b8d97c688c
4 changed files with 177 additions and 140 deletions

View file

@ -2027,6 +2027,7 @@ class Logging:
response_obj=result,
start_time=start_time,
end_time=end_time,
user_id=kwargs.get("user", None),
print_verbose=print_verbose,
)
if callback == "s3":
@ -2598,6 +2599,17 @@ class Logging:
level="ERROR",
kwargs=self.model_call_details,
)
if callback == "traceloop":
traceloopLogger.log_event(
start_time=start_time,
end_time=end_time,
response_obj=None,
user_id=kwargs.get("user", None),
print_verbose=print_verbose,
status_message=str(exception),
level="ERROR",
kwargs=self.model_call_details,
)
if callback == "prometheus":
global prometheusLogger
verbose_logger.debug("reaches prometheus for success logging!")