scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 156

test(rate_limiter): raise on ProxyRateLimitError instead of bare Exception

Browse source
This commit is contained in:
Priyansh Nandwana 2026-09-05 13:47:49 +05:30
parent 494880ddde
commit eda5f5d405
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/proxy/hooks/test_rate_limit_reset_time.py
View file

@ -11,6 +11,7 @@ import pytest
sys.path.insert(0, os.path.abspath("../../../.."))
from litellm.proxy.common_utils.proxy_rate_limit_error import ProxyRateLimitError
from litellm.proxy.hooks.batch_rate_limiter import BatchFileUsage, _PROXY_BatchRateLimiter
from litellm.proxy.hooks.parallel_request_limiter_v3 import (
_PROXY_MaxParallelRequestsHandler_v3,
@ -78,7 +79,7 @@ class TestParallelLimiterReportsUTC:
def test_a_non_utc_proxy_still_reports_utc(self, tz):
with a_proxy_running_in(tz):
limiter = _parallel_limiter()
with pytest.raises(Exception) as exc_info:
with pytest.raises(ProxyRateLimitError) as exc_info:
limiter._handle_rate_limit_error(
response=_over_limit_response(),
descriptors=_descriptors(),
@ -89,7 +90,7 @@ class TestParallelLimiterReportsUTC:
def test_the_reset_time_is_one_window_ahead(self):
limiter = _parallel_limiter()
with pytest.raises(Exception) as exc_info:
with pytest.raises(ProxyRateLimitError) as exc_info:
limiter._handle_rate_limit_error(
response=_over_limit_response(),
descriptors=_descriptors(),
@ -113,7 +114,7 @@ class TestBatchLimiterReportsUTC:
with a_proxy_running_in(tz):
before = datetime.now(timezone.utc)
with pytest.raises(Exception) as exc_info:
with pytest.raises(ProxyRateLimitError) as exc_info:
limiter._raise_rate_limit_error(
status=_over_limit_response()["statuses"][0],
descriptors=_descriptors(),