Merge pull request #3780 from nirga/traceloop-failures

Log errors in Traceloop Integration
This commit is contained in:
Krish Dholakia 2024-05-24 14:23:26 -07:00 committed by GitHub
commit bcd7534153
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 168 additions and 135 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!")