skullbite/cartridges
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 Fixed wrong variable in rate limiter init

Browse Source
This commit is contained in:
GeoffreyCoulaud
2023-06-03 16:35:25 +02:00
parent 58054f1c26
commit b1476a744d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/utils/rate_limiter.py
View File

@@ -45,7 +45,7 @@ class TokenBucketRateLimiter(AbstractContextManager):
# Initialize the bucket # Initialize the bucket
self.bucket = BoundedSemaphore(self.MAX_TOKENS) self.bucket = BoundedSemaphore(self.MAX_TOKENS)
missing = 0 if initial_tokens is None else self.MAX_TOKENS - initial_tokens missing = 0 if initial_tokens is None else self.MAX_TOKENS - initial_tokens
missing = max(0, min(missing, max_tokens)) missing = max(0, min(missing, self.MAX_TOKENS))
for _ in range(missing): for _ in range(missing):
self.bucket.acquire() self.bucket.acquire()