From 57ae20611b48546b84c6642272ed45a0df2555c1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 20 Sep 2025 17:00:30 -0700 Subject: [PATCH] tes fix --- .../test_prometheus_logging_callbacks.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py b/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py index eb2b2242e73..a18af7a54d2 100644 --- a/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py +++ b/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py @@ -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)