fix(prometheus_services.py): add better import error statement

2024-04-13 19:03:32 -07:00 · 2024-04-13 19:03:32 -07:00 · 67d0d5e356
commit 67d0d5e356
parent 71580c1245
1 changed files with 6 additions and 1 deletions
--- 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