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

test fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-20 14:09:24 -07:00
parent 59d2f0f465
commit 18d5177aec
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py
View file

@ -223,6 +223,7 @@ def test_increment_token_metrics(prometheus_logger):
prometheus_logger.litellm_tokens_metric.labels.assert_called_once_with(
end_user=None,
user=None,
user_email=None,
hashed_api_key="test_hash",
api_key_alias="test_alias",
team="test_team",
@ -715,12 +716,13 @@ async def test_async_post_call_failure_hook(prometheus_logger):
# Assert failed requests metric was incremented with correct labels
prometheus_logger.litellm_proxy_failed_requests_metric.labels.assert_called_once_with(
end_user=None,
user="test_user",
user_email=None,
hashed_api_key="test_key",
api_key_alias="test_alias",
requested_model="gpt-3.5-turbo",
team="test_team",
team_alias="test_team_alias",
user="test_user",
requested_model="gpt-3.5-turbo",
exception_status="429",
exception_class="Openai.RateLimitError",
route=user_api_key_dict.request_route,