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

add async_post_call_failure_hook

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 19:12:38 -08:00
parent 90168de961
commit 2803661ed8
2 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -357,18 +357,16 @@ class DataDogLogger(CustomBatchLogger):
): ):
import json import json
_exception_payload = { _exception_payload = DatadogProxyFailureHookJsonMessage(
"error_str": str(original_exception), exception=str(original_exception),
"error_class": str(original_exception.__class__.__name__), error_class=str(original_exception.__class__.__name__),
"status_code": getattr(original_exception, "status_code", None), status_code=getattr(original_exception, "status_code", None),
"traceback": traceback.format_exc(), traceback=traceback.format_exc(),
"user_api_key_dict": user_api_key_dict.model_dump(), user_api_key_dict=user_api_key_dict.model_dump(),
} )
json_payload = json.dumps(_exception_payload) json_payload = json.dumps(_exception_payload)
verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload) verbose_logger.debug("Datadog: Logger - Logging payload = %s", json_payload)
dd_payload = DatadogPayload( dd_payload = DatadogPayload(
ddsource=os.getenv("DD_SOURCE", "litellm"), ddsource=os.getenv("DD_SOURCE", "litellm"),
ddtags="", ddtags="",

5
litellm/types/integrations/datadog.py
View file

@ -1,5 +1,5 @@
from enum import Enum from enum import Enum
from typing import TypedDict from typing import Optional, TypedDict
class DataDogStatus(str, Enum): class DataDogStatus(str, Enum):
@ -23,6 +23,7 @@ class DD_ERRORS(Enum):
class DatadogProxyFailureHookJsonMessage(TypedDict, total=False): class DatadogProxyFailureHookJsonMessage(TypedDict, total=False):
exception: str exception: str
error_class: str
status_code: Optional[int]
traceback: str traceback: str
request_data: dict
user_api_key_dict: dict user_api_key_dict: dict