diff --git a/litellm/integrations/prometheus_services.py b/litellm/integrations/prometheus_services.py index 151529b23a..548d0a2a3a 100644 --- a/litellm/integrations/prometheus_services.py +++ b/litellm/integrations/prometheus_services.py @@ -24,7 +24,12 @@ class PrometheusServicesLogger: **kwargs, ): try: - from prometheus_client import Counter, Histogram, REGISTRY + try: + from prometheus_client import Counter, Histogram, REGISTRY + except ImportError: + raise Exception( + "Missing prometheus_client. Run `pip install prometheus-client`" + ) self.Histogram = Histogram self.Counter = Counter