From d2b6054f0c694a16c1864f2c2838bb3b4706ad58 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 26 Nov 2024 17:58:46 -0800 Subject: [PATCH] use standard logging payload for DD --- litellm/integrations/datadog/datadog.py | 24 ++---------------------- 1 file changed, 2 insertions(+), 22 deletions(-) diff --git a/litellm/integrations/datadog/datadog.py b/litellm/integrations/datadog/datadog.py index 83dc34341..01283ac2d 100644 --- a/litellm/integrations/datadog/datadog.py +++ b/litellm/integrations/datadog/datadog.py @@ -258,30 +258,10 @@ class DataDogLogger(CustomBatchLogger): ) if standard_logging_object is None: raise ValueError("standard_logging_object not found in kwargs") - _start_time: float = standard_logging_object.get("startTime", 0) - _end_time: float = standard_logging_object.get("endTime", 0) - response_time_seconds: float = _end_time - _start_time # Build the initial payload - payload = { - "id": standard_logging_object.get("id"), - "call_type": standard_logging_object.get("call_type"), - "cache_hit": standard_logging_object.get("cache_hit"), - "start_time": _start_time, - "end_time": _end_time, - "response_time": response_time_seconds, - "model": standard_logging_object.get("model"), - "user": standard_logging_object.get("end_user"), - "model_parameters": standard_logging_object.get("model_parameters"), - "spend": standard_logging_object.get("response_cost"), - "messages": standard_logging_object.get("messages"), - "response": standard_logging_object.get("response"), - "usage": standard_logging_object.get("usage"), - "metadata": standard_logging_object.get("metadata"), - } - - make_json_serializable(payload) - json_payload = json.dumps(payload) + make_json_serializable(standard_logging_object) + json_payload = json.dumps(standard_logging_object) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)