(fix) GCS bucket logger - apply truncate_standard_logging_payload_content to standard_logging_payload and ensure GCS flushes queue on fails (#7519)

* fix async_send_batch for gcs

* fix truncate GCS logger

* test_truncate_standard_logging_payload
This commit is contained in:
Ishaan Jaff 2025-01-03 08:09:03 -08:00 committed by GitHub
parent 6ffdd5c250
commit ec3bcf189f
5 changed files with 87 additions and 84 deletions

View file

@ -256,10 +256,6 @@ class DataDogLogger(CustomBatchLogger):
"""
import json
from litellm.litellm_core_utils.litellm_logging import (
truncate_standard_logging_payload_content,
)
standard_logging_object: Optional[StandardLoggingPayload] = kwargs.get(
"standard_logging_object", None
)
@ -271,7 +267,7 @@ class DataDogLogger(CustomBatchLogger):
status = DataDogStatus.ERROR
# Build the initial payload
truncate_standard_logging_payload_content(standard_logging_object)
self.truncate_standard_logging_payload_content(standard_logging_object)
json_payload = json.dumps(standard_logging_object, default=str)
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)