From b3ccbb96ba88d9a106f9f24f202ceb8e0c4812c4 Mon Sep 17 00:00:00 2001 From: harish-berri Date: Fri, 17 Apr 2026 19:06:23 +0000 Subject: [PATCH] resolve lint errors --- litellm/integrations/prometheus.py | 8 -------- litellm/types/integrations/prometheus.py | 3 +-- 2 files changed, 1 insertion(+), 10 deletions(-) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index 58850d1aff4..522458f3487 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -1014,8 +1014,6 @@ class PrometheusLogger(CustomLogger): async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): # Define prometheus client - from litellm.types.utils import StandardLoggingPayload - verbose_logger.debug( "prometheus Logging - Enters success logging function (kwargs keys: %s)", list(kwargs.keys()) if isinstance(kwargs, dict) else type(kwargs).__name__, @@ -1538,8 +1536,6 @@ class PrometheusLogger(CustomLogger): ) async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time): - from litellm.types.utils import StandardLoggingPayload - verbose_logger.debug( "prometheus Logging - Enters failure logging function (kwargs keys: %s)", list(kwargs.keys()) if isinstance(kwargs, dict) else type(kwargs).__name__, @@ -2752,8 +2748,6 @@ class PrometheusLogger(CustomLogger): """ Initialize API key budget metrics by reusing the generic pagination logic. """ - from typing import Union - from litellm.constants import UI_SESSION_TOKEN_TEAM_ID from litellm.proxy.management_endpoints.key_management_endpoints import ( _list_key_helper, @@ -2800,7 +2794,6 @@ class PrometheusLogger(CustomLogger): """ Initialize user budget metrics by reusing the generic pagination logic. """ - from litellm.proxy._types import LiteLLM_UserTable from litellm.proxy.proxy_server import prisma_client if prisma_client is None: @@ -3441,7 +3434,6 @@ class PrometheusLogger(CustomLogger): It emits the current remaining budget metrics for all Keys and Teams. """ from litellm.constants import PROMETHEUS_BUDGET_METRICS_REFRESH_INTERVAL_MINUTES - from litellm.integrations.custom_logger import CustomLogger prometheus_loggers: List[ CustomLogger diff --git a/litellm/types/integrations/prometheus.py b/litellm/types/integrations/prometheus.py index 6fe859370cd..fd84fc0f290 100644 --- a/litellm/types/integrations/prometheus.py +++ b/litellm/types/integrations/prometheus.py @@ -4,8 +4,6 @@ from enum import Enum from types import MappingProxyType from typing import Any, ClassVar, Dict, List, Literal, Mapping, Optional, Tuple, Union -from typing_extensions import Annotated - import litellm @@ -786,6 +784,7 @@ class UserAPIKeyLabelValues: org_id: Optional[str] = None org_alias: Optional[str] = None + #Added for test compatibility. def __init__(self, **kwargs: Any) -> None: """ Match former Pydantic behavior: unknown keys are ignored; ``api_key_hash`` maps to