From 1c229dad91357cefc3527839662ff2722cdffdc9 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Mon, 8 Apr 2024 21:16:48 +0530 Subject: [PATCH] Set daily limit for unsubsribed users to 5 in websocket API --- src/khoj/routers/api_chat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py index 5258a7e8..5cac7058 100644 --- a/src/khoj/routers/api_chat.py +++ b/src/khoj/routers/api_chat.py @@ -313,9 +313,9 @@ async def websocket_endpoint( user, client_application=websocket.user.client_app, conversation_id=conversation_id ) - hourly_limiter = ApiUserRateLimiter(requests=1, subscribed_requests=60, window=60, slug="chat_minute") + hourly_limiter = ApiUserRateLimiter(requests=5, subscribed_requests=60, window=60, slug="chat_minute") - daily_limiter = ApiUserRateLimiter(requests=1, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day") + daily_limiter = ApiUserRateLimiter(requests=5, subscribed_requests=600, window=60 * 60 * 24, slug="chat_day") await is_ready_to_chat(user)