phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test litellm_spend_metric

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-19 16:33:13 -08:00
parent a600cabd2d
commit 1a95af41fd
1 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/logging_callback_tests/test_prometheus_unit_tests.py
View file

@ -333,15 +333,30 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger):
user_api_team_alias="team_alias1",
user_id="user1",
response_cost=0.1,
custom_llm_provider="openai",
)
prometheus_logger.litellm_requests_metric.labels.assert_called_once_with(
"user1", "key1", "alias1", "gpt-3.5-turbo", "team1", "team_alias1", "user1"
"user1",
"key1",
"alias1",
"gpt-3.5-turbo",
"team1",
"team_alias1",
"user1",
"openai",
)
prometheus_logger.litellm_requests_metric.labels().inc.assert_called_once()
prometheus_logger.litellm_spend_metric.labels.assert_called_once_with(
"user1", "key1", "alias1", "gpt-3.5-turbo", "team1", "team_alias1", "user1"
"user1",
"key1",
"alias1",
"gpt-3.5-turbo",
"team1",
"team_alias1",
"user1",
"openai",
)
prometheus_logger.litellm_spend_metric.labels().inc.assert_called_once_with(0.1)