scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 93

test: fix invalid 'rpm' rate_limit_type in v3 limiter test mocks

Browse source 
The v3 rate limiter only emits 'requests', 'tokens', or
'max_parallel_requests'. Using 'rpm' caused map_v3_rate_limit_type to
return None, leaving the expected RateLimitType.REQUESTS untested.

Co-authored-by: Yassin Kortam <yassin@berri.ai>
This commit is contained in:
Cursor Agent 2026-05-14 21:45:02 +00:00
parent 2621f368cd
commit f3907ea0ce
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/hooks/test_proxy_rate_limit_provider_field.py
View file

@ -486,7 +486,7 @@ async def test_parallel_request_limiter_v1_missing_model_falls_back():
# ---------------------------------------------------------------------------
def _v3_over_limit_response(rate_limit_type: str = "rpm") -> dict:
def _v3_over_limit_response(rate_limit_type: str = "requests") -> dict:
return {
"overall_code": "OVER_LIMIT",
"statuses": [
@ -666,7 +666,7 @@ async def test_dynamic_rate_limiter_v3_model_capacity_path_populates_provider():
"descriptor_key": "model_saturation_check",
"current_limit": 100,
"limit_remaining": 0,
"rate_limit_type": "rpm",
"rate_limit_type": "requests",
}
],
}
@ -716,7 +716,7 @@ async def test_dynamic_rate_limiter_v3_unknown_descriptor_path_populates_provide
"descriptor_key": "something_we_dont_handle",
"current_limit": 1,
"limit_remaining": 0,
"rate_limit_type": "rpm",
"rate_limit_type": "requests",
}
],
}