From 67d0d5e35667cf5265626b81c67947e8bfe74bc6 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 13 Apr 2024 19:03:32 -0700
Subject: [PATCH] fix(prometheus_services.py): add better import error
 statement

---
 litellm/integrations/prometheus_services.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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