From 218383f739e2654b9055d08b3638fcf90bbb499e Mon Sep 17 00:00:00 2001 From: Harshit Jain Date: Thu, 26 Feb 2026 11:45:28 +0530 Subject: [PATCH] fix: req changes --- litellm/integrations/datadog/datadog_metrics.py | 1 + litellm/proxy/health_endpoints/_health_endpoints.py | 11 ++++++++++- litellm/types/integrations/datadog_metrics.py | 5 +++-- 3 files changed, 14 insertions(+), 3 deletions(-) diff --git a/litellm/integrations/datadog/datadog_metrics.py b/litellm/integrations/datadog/datadog_metrics.py index a15918bc6df..a22efd66000 100644 --- a/litellm/integrations/datadog/datadog_metrics.py +++ b/litellm/integrations/datadog/datadog_metrics.py @@ -150,6 +150,7 @@ class DatadogMetricsLogger(CustomBatchLogger): "type": 1, # count "points": [{"timestamp": timestamp, "value": 1.0}], "tags": tags, + "interval": self.flush_interval, } self.log_queue.append(series_count) diff --git a/litellm/proxy/health_endpoints/_health_endpoints.py b/litellm/proxy/health_endpoints/_health_endpoints.py index 83891c90e14..f37eb344d35 100644 --- a/litellm/proxy/health_endpoints/_health_endpoints.py +++ b/litellm/proxy/health_endpoints/_health_endpoints.py @@ -285,8 +285,17 @@ async def health_services_endpoint( # noqa: PLR0915 from litellm.integrations.datadog.datadog_metrics import ( DatadogMetricsLogger, ) + from litellm.litellm_core_utils.litellm_logging import ( + get_custom_logger_compatible_class, + ) - datadog_metrics_logger = DatadogMetricsLogger(start_periodic_flush=False) + datadog_metrics_logger = get_custom_logger_compatible_class( + "datadog_metrics" + ) + if datadog_metrics_logger is None: + datadog_metrics_logger = DatadogMetricsLogger( + start_periodic_flush=False + ) response = await datadog_metrics_logger.async_health_check() return { "status": response["status"], diff --git a/litellm/types/integrations/datadog_metrics.py b/litellm/types/integrations/datadog_metrics.py index 7269a77cc25..4c980cdee6d 100644 --- a/litellm/types/integrations/datadog_metrics.py +++ b/litellm/types/integrations/datadog_metrics.py @@ -1,4 +1,4 @@ -from typing import List +from typing import List, Optional from typing_extensions import TypedDict @@ -8,11 +8,12 @@ class DatadogMetricPoint(TypedDict): value: float # The metric value -class DatadogMetricSeries(TypedDict): +class DatadogMetricSeries(TypedDict, total=False): metric: str type: int # 0=unspecified, 1=count, 2=rate, 3=gauge points: List[DatadogMetricPoint] tags: List[str] + interval: Optional[int] # Required for count (type=1) and rate (type=2) metrics class DatadogMetricsPayload(TypedDict):