scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

refactor(types): import litellm lazily in prometheus types helpers (#45776)

Browse source 
Part of LIT-9423
This commit is contained in:
nate-berri 2026-10-10 12:07:32 -07:00 • committed by GitHub
parent 9d559d8579
commit 4606bf532e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/types/integrations/prometheus.py
View file

@ -5,8 +5,6 @@ from enum import Enum
from types import MappingProxyType
from typing import Any, ClassVar, Final, Literal, cast
import litellm
def sanitize_prometheus_label_name(label: str) -> str:
"""
@ -98,6 +96,8 @@ class LabelValidationError:
@property
def message(self) -> str:
import litellm
base_message: Final = f"Invalid labels for metric '{self.metric_name}': {self.invalid_labels}"
if self.metric_name in PROMETHEUS_DEPLOYMENT_AND_LATENCY_CALLER_IDENTITY_METRICS and any(
label in ("api_key_alias", "user_email") for label in self.invalid_labels
@ -349,6 +349,8 @@ PROMETHEUS_DEPLOYMENT_AND_LATENCY_CALLER_IDENTITY_VALUES: Final[tuple[str, ...]]
def validate_prometheus_deployment_and_latency_caller_identity() -> str:
"""Return the configured caller-identity mode, raising on an invalid value."""
import litellm
caller_identity: Final[object] = getattr(
litellm,
"prometheus_deployment_and_latency_caller_identity",
@ -367,6 +369,8 @@ def validate_caller_identity_settings(litellm_settings: Mapping[str, object]) ->
"""Store the caller-identity mode from litellm_settings and validate it together
with prometheus_metrics_config, raising on an invalid value or on include_labels
that request a label the selected mode removes."""
import litellm
if "prometheus_deployment_and_latency_caller_identity" not in litellm_settings:
return
litellm.prometheus_deployment_and_latency_caller_identity = (
@ -978,6 +982,8 @@ class PrometheusMetricLabels:
@staticmethod
def get_labels(label_name: DEFINED_PROMETHEUS_METRICS) -> list[str]:
import litellm
default_labels: Final = _resolve_deployment_and_latency_caller_identity_labels(
metric_name=label_name,
labels=getattr(PrometheusMetricLabels, label_name),