Skip to content

Commit 8b3e3c1

Browse files
committed
tes fix
1 parent c656982 commit 8b3e3c1

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -586,9 +586,16 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger):
586586
)
587587
prometheus_logger.litellm_requests_metric.labels().inc.assert_called_once()
588588

589-
# The spend metric uses keyword arguments
589+
# The spend metric uses keyword arguments (same as requests metric)
590590
prometheus_logger.litellm_spend_metric.labels.assert_called_once_with(
591-
"user1", "key1", "alias1", "gpt-3.5-turbo", "team1", "team_alias1", "user1"
591+
end_user=None,
592+
user=None,
593+
hashed_api_key="test_hash",
594+
api_key_alias="test_alias",
595+
team="test_team",
596+
team_alias="test_team_alias",
597+
model="gpt-3.5-turbo",
598+
user_email=None,
592599
)
593600
prometheus_logger.litellm_spend_metric.labels().inc.assert_called_once_with(0.1)
594601

0 commit comments

Comments
 (0)