From 90168de961a03dba52e646568baccc4e0dd487af Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 26 Nov 2024 19:09:16 -0800 Subject: [PATCH] use correct loc for types --- litellm/integrations/datadog/datadog.py | 2 +- .../datadog/types.py => types/integrations/datadog.py} | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) rename litellm/{integrations/datadog/types.py => types/integrations/datadog.py} (76%) diff --git a/litellm/integrations/datadog/datadog.py b/litellm/integrations/datadog/datadog.py index 1df037ba4..9f95780aa 100644 --- a/litellm/integrations/datadog/datadog.py +++ b/litellm/integrations/datadog/datadog.py @@ -33,10 +33,10 @@ from litellm.llms.custom_httpx.http_handler import ( httpxSpecialProvider, ) from litellm.proxy._types import UserAPIKeyAuth +from litellm.types.integrations.datadog import * from litellm.types.services import ServiceLoggerPayload from litellm.types.utils import StandardLoggingPayload -from .types import DD_ERRORS, DatadogPayload, DataDogStatus from .utils import make_json_serializable DD_MAX_BATCH_SIZE = 1000 # max number of logs DD API can accept diff --git a/litellm/integrations/datadog/types.py b/litellm/types/integrations/datadog.py similarity index 76% rename from litellm/integrations/datadog/types.py rename to litellm/types/integrations/datadog.py index 87aa3ce17..4d11daa75 100644 --- a/litellm/integrations/datadog/types.py +++ b/litellm/types/integrations/datadog.py @@ -19,3 +19,10 @@ class DatadogPayload(TypedDict, total=False): class DD_ERRORS(Enum): DATADOG_413_ERROR = "Datadog API Error - Payload too large (batch is above 5MB uncompressed). If you want this logged either disable request/response logging or set `DD_BATCH_SIZE=50`" + + +class DatadogProxyFailureHookJsonMessage(TypedDict, total=False): + exception: str + traceback: str + request_data: dict + user_api_key_dict: dict