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 108

test fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-20 15:32:59 -07:00
parent 10de012e12
commit 8ec73db36b
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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(