From 550b47233d3e2c8e22a84d73698056a7c603c0c4 Mon Sep 17 00:00:00 2001 From: ademicho123 Date: Sun, 26 Jul 2026 17:55:24 +0100 Subject: [PATCH] style: ruff format --- litellm/integrations/prometheus.py | 22 +++++----------------- litellm/types/integrations/prometheus.py | 3 +++ 2 files changed, 8 insertions(+), 17 deletions(-) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index da2547a7556..8ce2971ea0d 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -681,9 +681,7 @@ class PrometheusLogger(CustomLogger): if validation_results.has_errors: self._pretty_print_validation_errors(validation_results) - error_message = "Configuration validation failed:\n" + "\n".join( - validation_results.all_error_messages - ) + error_message = "Configuration validation failed:\n" + "\n".join(validation_results.all_error_messages) raise ValueError(error_message) # Build label filters from valid configurations @@ -717,11 +715,7 @@ class PrometheusLogger(CustomLogger): for metric_name in get_args(DEFINED_PROMETHEUS_METRICS) for label in PrometheusMetricLabels.get_labels(metric_name) } - unknown_labels = [ - label - for label in config.include_labels - if label not in all_valid_labels - ] + unknown_labels = [label for label in config.include_labels if label not in all_valid_labels] if unknown_labels: label_errors.append( LabelValidationError( @@ -739,9 +733,7 @@ class PrometheusLogger(CustomLogger): continue if config.include_labels: - label_error = self._validate_single_metric_labels( - metric_name, config.include_labels - ) + label_error = self._validate_single_metric_labels(metric_name, config.include_labels) if label_error: label_errors.append(label_error) @@ -782,12 +774,8 @@ class PrometheusLogger(CustomLogger): label_filters: Dict[str, List[str]] = {} - wildcard_configs = [ - c for c in parsed_configs if c.metrics == [PROMETHEUS_METRICS_WILDCARD] - ] - named_configs = [ - c for c in parsed_configs if c.metrics != [PROMETHEUS_METRICS_WILDCARD] - ] + wildcard_configs = [c for c in parsed_configs if c.metrics == [PROMETHEUS_METRICS_WILDCARD]] + named_configs = [c for c in parsed_configs if c.metrics != [PROMETHEUS_METRICS_WILDCARD]] for config in wildcard_configs: if not config.include_labels: diff --git a/litellm/types/integrations/prometheus.py b/litellm/types/integrations/prometheus.py index 9ac31e8ac4b..6e37d8a0da5 100644 --- a/litellm/types/integrations/prometheus.py +++ b/litellm/types/integrations/prometheus.py @@ -68,8 +68,10 @@ def _sanitize_prometheus_label_value(value: Optional[Any]) -> Optional[str]: append(ch) return "".join(parts) + PROMETHEUS_METRICS_WILDCARD = "*" + @dataclass class MetricValidationError: """Error for invalid metric name""" @@ -843,6 +845,7 @@ _USER_API_KEY_LABEL_VALUE_INIT_ALIASES: Dict[str, str] = { "api_key_hash": "hashed_api_key", } + @dataclass(frozen=True, init=False) class UserAPIKeyLabelValues: """