scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 128

fix(test): update CustomPrometheusLogger to use new enum_values API

Browse source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Sameer Kankute 2026-06-24 18:49:30 +05:30
parent 0b354c1d95
commit 774c8e2e7d
No known key found for this signature in database
1 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/enterprise/litellm_enterprise/integrations/test_prometheus_unit_tests.py
View file

@ -29,6 +29,7 @@ from litellm import Router
from litellm.caching.caching import DualCache
from litellm.router_strategy.budget_limiter import RouterBudgetLimiting
from litellm.router_utils.cooldown_callbacks import router_cooldown_event_callback
from litellm.types.integrations.prometheus import UserAPIKeyLabelValues
from litellm.types.router import ModelInfo
from litellm.types.utils import BudgetConfig, GenericBudgetConfigType
@ -226,21 +227,17 @@ class CustomPrometheusLogger(PrometheusLogger):
def increment_deployment_cooled_down(
self,
litellm_model_name: str,
model_id: str,
api_base: str,
api_provider: str,
enum_values: "UserAPIKeyLabelValues",
exception_status: str,
model_group: Optional[str] = None,
):
self.deployment_cooled_downs.append(
[
litellm_model_name,
model_id,
api_base,
api_provider,
enum_values.litellm_model_name,
enum_values.model_id,
enum_values.api_base,
enum_values.api_provider,
exception_status,
model_group,
enum_values.model_group,
]
)