fix rate limiting (#5135)

All-Hands-AI · Nov 19, 2024 · 018080a · 018080a
1 parent 302e41d
commit 018080a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/frontend/package-lock.json b/frontend/package-lock.json
diff --git a/openhands/server/listen.py b/openhands/server/listen.py
@@ -90,7 +90,7 @@
 
 app.add_middleware(NoCacheMiddleware)
 app.add_middleware(
-    RateLimitMiddleware, rate_limiter=InMemoryRateLimiter(requests=2, seconds=1)
+    RateLimitMiddleware, rate_limiter=InMemoryRateLimiter(requests=10, seconds=1)
 )
 
 

diff --git a/openhands/server/middleware.py b/openhands/server/middleware.py
@@ -57,6 +57,7 @@ class InMemoryRateLimiter:
     def __init__(self, requests: int = 2, seconds: int = 1, sleep_seconds: int = 1):
         self.requests = requests
         self.seconds = seconds
+        self.sleep_seconds = sleep_seconds
         self.history = defaultdict(list)
 
     def _clean_old_requests(self, key: str) -> None: