phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

use standard logging payload for DD

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 17:58:46 -08:00
parent 6d0424881a
commit d2b6054f0c
1 changed files with 2 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/integrations/datadog/datadog.py
View file

@ -258,30 +258,10 @@ class DataDogLogger(CustomBatchLogger):
) )
if standard_logging_object is None: if standard_logging_object is None:
raise ValueError("standard_logging_object not found in kwargs") 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 # Build the initial payload
payload = { make_json_serializable(standard_logging_object)
"id": standard_logging_object.get("id"), json_payload = json.dumps(standard_logging_object)
"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)
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)