From c8223725ad4d4311d222f7b0d8e97cf74fe9ffc1 Mon Sep 17 00:00:00 2001 From: Conatum Date: Sun, 1 Oct 2023 13:33:30 +0300 Subject: [PATCH] fix(utils): Fix delay issue in Bucket. --- src/utils/ratelimits.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/utils/ratelimits.py b/src/utils/ratelimits.py index 4c050e94..7322336c 100644 --- a/src/utils/ratelimits.py +++ b/src/utils/ratelimits.py @@ -56,7 +56,10 @@ class Bucket: def delay(self): self._leak() if self._level + 1 > self.max_level: - return (self._level + 1 - self.max_level) * self.leak_rate + delay = (self._level + 1 - self.max_level) * self.leak_rate + else: + delay = 0 + return delay def _leak(self): if self._level: