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

tes fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-20 17:00:30 -07:00
parent 8fb6d238d4
commit 57ae20611b
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -586,9 +586,16 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger):
)
prometheus_logger.litellm_requests_metric.labels().inc.assert_called_once()
# The spend metric uses keyword arguments
# The spend metric uses keyword arguments (same as requests metric)
prometheus_logger.litellm_spend_metric.labels.assert_called_once_with(
"user1", "key1", "alias1", "gpt-3.5-turbo", "team1", "team_alias1", "user1"
end_user=None,
user=None,
hashed_api_key="test_hash",
api_key_alias="test_alias",
team="test_team",
team_alias="test_team_alias",
model="gpt-3.5-turbo",
user_email=None,
)
prometheus_logger.litellm_spend_metric.labels().inc.assert_called_once_with(0.1)