From ef098387cccf825fe2b41c2e2f1c90df209bced7 Mon Sep 17 00:00:00 2001
From: Gyubong Lee <jopemachine@naver.com>
Date: Fri, 26 Apr 2024 06:16:24 +0000
Subject: [PATCH] fix: Handle when ip_address not available

---
 src/ai/backend/manager/api/ratelimit.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/ai/backend/manager/api/ratelimit.py b/src/ai/backend/manager/api/ratelimit.py
index f508513ccb0..ca81f6229b6 100644
--- a/src/ai/backend/manager/api/ratelimit.py
+++ b/src/ai/backend/manager/api/ratelimit.py
@@ -80,13 +80,12 @@ async def rlim_middleware(
         response.headers["X-RateLimit-Window"] = str(_rlim_window)
         return response
     else:
-        ip_address = get_client_ip(request)
-        assert ip_address is not None, "Client IP Address is not provided"
-
         root_ctx: RootContext = app["_root.context"]
         rate_limit = root_ctx.local_config["manager"]["anonymous-ratelimit"]
 
-        if rate_limit is None:
+        ip_address = get_client_ip(request)
+
+        if not ip_address or rate_limit is None:
             # No checks for rate limiting.
             response = await handler(request)
             # Arbitrary number for indicating no rate limiting.