From a600cabd2d1561d16e051a3ed13b4e8a3f5ddc90 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 19 Nov 2024 16:29:20 -0800 Subject: [PATCH] fix import error --- litellm/integrations/prometheus.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index 22216438e..0b6cd8b3b 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -15,7 +15,6 @@ import requests # type: ignore import litellm from litellm._logging import print_verbose, verbose_logger from litellm.integrations.custom_logger import CustomLogger -from litellm.litellm_core_utils.litellm_logging import StandardLoggingPayloadAccessors from litellm.proxy._types import UserAPIKeyAuth from litellm.types.integrations.prometheus import * from litellm.types.utils import StandardLoggingPayload @@ -338,6 +337,9 @@ class PrometheusLogger(CustomLogger): async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): # Define prometheus client + from litellm.litellm_core_utils.litellm_logging import ( + StandardLoggingPayloadAccessors, + ) from litellm.types.utils import StandardLoggingPayload verbose_logger.debug(