From df4593d58bf5f3047061a3ce7ece2fb89900f3fa Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 4 Apr 2025 21:30:05 -0700 Subject: [PATCH] test prom unit tests --- tests/logging_callback_tests/test_prometheus_unit_tests.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/logging_callback_tests/test_prometheus_unit_tests.py b/tests/logging_callback_tests/test_prometheus_unit_tests.py index 6bc5b42c45..c24bb27691 100644 --- a/tests/logging_callback_tests/test_prometheus_unit_tests.py +++ b/tests/logging_callback_tests/test_prometheus_unit_tests.py @@ -713,7 +713,7 @@ async def test_async_post_call_failure_hook(prometheus_logger): team_alias="test_team_alias", user="test_user", exception_status="429", - exception_class="RateLimitError", + exception_class="Openai.RateLimitError", ) prometheus_logger.litellm_proxy_failed_requests_metric.labels().inc.assert_called_once() @@ -948,7 +948,7 @@ async def test_log_success_fallback_event(prometheus_logger): team="test_team", team_alias="test_team_alias", exception_status="429", - exception_class="RateLimitError", + exception_class="Openai.RateLimitError", ) prometheus_logger.litellm_deployment_successful_fallbacks.labels().inc.assert_called_once() @@ -985,7 +985,7 @@ async def test_log_failure_fallback_event(prometheus_logger): team="test_team", team_alias="test_team_alias", exception_status="429", - exception_class="RateLimitError", + exception_class="Openai.RateLimitError", ) prometheus_logger.litellm_deployment_failed_fallbacks.labels().inc.assert_called_once()