scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

linting fix prometheus services

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-02 21:19:05 -07:00
parent 5222cce510
commit bcf42fd82d
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/prometheus_services.py
View file

@ -3,7 +3,7 @@
# On success + failure, log events to Prometheus for litellm / adjacent services (litellm, redis, postgres, llm api providers)
from typing import List, Optional, Union
from typing import Dict, List, Optional, Union
from litellm._logging import print_verbose, verbose_logger
from litellm.types.integrations.prometheus import LATENCY_BUCKETS
@ -43,7 +43,9 @@ class PrometheusServicesLogger:
verbose_logger.debug("in init prometheus services metrics")
self.services = [item for item in ServiceTypes]
self.payload_to_prometheus_map = {}
self.payload_to_prometheus_map: Dict[
str, List[Union[Histogram, Counter, Gauge, Collector]]
] = {}
for service in self.services:
service_metrics: List[Union[Histogram, Counter, Gauge, Collector]] = []
@ -78,7 +80,7 @@ class PrometheusServicesLogger:
service_metrics.append(gauge)
if service_metrics:
self.payload_to_prometheus_map[service] = service_metrics
self.payload_to_prometheus_map[service.value] = service_metrics
self.prometheus_to_amount_map: dict = {}
### MOCK TESTING ###