From 8ec73db36b601d9b3d3a73dbc9612e3279a1b27a Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 20 Sep 2025 15:32:59 -0700 Subject: [PATCH] test fix --- .../test_prometheus_logging_callbacks.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 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 d18305da13b..bc048f7407d 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 @@ -563,15 +563,15 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger): prometheus_logger.litellm_spend_metric = MagicMock() prometheus_logger._increment_top_level_request_and_spend_metrics( - end_user_id=None, - user_api_key="test_hash", - user_api_key_alias="test_alias", - model="gpt-3.5-turbo", - user_api_team="test_team", - user_api_team_alias="test_team_alias", - user_id=None, - response_cost=0.1, - enum_values=enum_values, + None, # end_user_id + "test_hash", # user_api_key + "test_alias", # user_api_key_alias + "gpt-3.5-turbo", # model + "test_team", # user_api_team + "test_team_alias", # user_api_team_alias + None, # user_id + 0.1, # response_cost + enum_values, # enum_values ) prometheus_logger.litellm_requests_metric.labels.assert_called_once_with(