scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

refactor: make the code easier to read

Browse source
This commit is contained in:
mateo-berri 2026-07-14 13:58:16 -07:00
parent ab87ebe26d
commit a21669aaef
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/e2e/quota_management/ratelimit/test_rate_limit_e2e.py
View file

@ -102,7 +102,9 @@ class _CrossedLimit:
spent: int
def _spend_until_blocked_or_crossed(client: QuotaClient, key: str, first: _FirstOk) -> _BlockedByReservation | _CrossedLimit:
def _spend_until_blocked_or_crossed(
client: QuotaClient, key: str, first: _FirstOk
) -> _BlockedByReservation | _CrossedLimit:
"""Drive chat traffic, summing each body's actual usage.total_tokens, until
the limiter blocks (which the reservation may do before the actual spend
crosses the limit) or the actual spend reaches the limit."""
@ -238,8 +240,8 @@ class TestKeyRateLimits:
f"success response must report the key's request limit, headers: "
f"{ {k: v for k, v in first.headers.items() if 'ratelimit' in k} }"
)
assert first.headers.get("x-ratelimit-api_key-remaining-requests") == "4", (
f"first call against rpm_limit=5 must leave 4 remaining, got "
assert first.headers.get("x-ratelimit-api_key-remaining-requests") == str(5 - 1), (
f"first call against rpm_limit=5 must leave {5 - 1} remaining, got "
f"{first.headers.get('x-ratelimit-api_key-remaining-requests')!r}"
)
assert first.headers.get("x-ratelimit-api_key-limit-tokens") == "100000", (