From 3d0e183beafd830854395995b8db87bea349da11 Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Mon, 7 Oct 2024 14:36:12 -0700
Subject: [PATCH] Add more log lines when encountering rate limiting

---
 src/khoj/database/admin.py   |  2 ++
 src/khoj/routers/api_chat.py |  2 +-
 src/khoj/routers/helpers.py  | 13 +++++++++++++
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/khoj/database/admin.py b/src/khoj/database/admin.py
index f6222006..ea15760e 100644
--- a/src/khoj/database/admin.py
+++ b/src/khoj/database/admin.py
@@ -27,6 +27,7 @@ from khoj.database.models import (
     Subscription,
     TextToImageModelConfig,
     UserConversationConfig,
+    UserRequests,
     UserSearchModelConfig,
     UserVoiceModelConfig,
     VoiceModelOption,
@@ -103,6 +104,7 @@ admin.site.register(NotionConfig)
 admin.site.register(UserVoiceModelConfig)
 admin.site.register(VoiceModelOption)
 admin.site.register(UserConversationConfig)
+admin.site.register(UserRequests)
 
 
 @admin.register(Agent)
diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py
index ce5f8fd8..392d9943 100644
--- a/src/khoj/routers/api_chat.py
+++ b/src/khoj/routers/api_chat.py
@@ -66,7 +66,7 @@ from khoj.utils.rawconfig import FileFilterRequest, FilesFilterRequest, Location
 # Initialize Router
 logger = logging.getLogger(__name__)
 conversation_command_rate_limiter = ConversationCommandRateLimiter(
-    trial_rate_limit=100, subscribed_rate_limit=100, slug="command"
+    trial_rate_limit=100, subscribed_rate_limit=6000, slug="command"
 )
 
 
diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py
index 26591d7f..8011d656 100644
--- a/src/khoj/routers/helpers.py
+++ b/src/khoj/routers/helpers.py
@@ -1029,13 +1029,23 @@ class ApiUserRateLimiter:
 
         # Check if the user has exceeded the rate limit
         if subscribed and count_requests >= self.subscribed_requests:
+            logger.info(
+                f"Rate limit: {count_requests} requests in {self.window} seconds for user: {user}. Limit is {self.subscribed_requests} requests."
+            )
             raise HTTPException(status_code=429, detail="Slow down! Too Many Requests")
         if not subscribed and count_requests >= self.requests:
             if self.requests >= self.subscribed_requests:
+                logger.info(
+                    f"Rate limit: {count_requests} requests in {self.window} seconds for user: {user}. Limit is {self.subscribed_requests} requests."
+                )
                 raise HTTPException(
                     status_code=429,
                     detail="Slow down! Too Many Requests",
                 )
+
+            logger.info(
+                f"Rate limit: {count_requests} requests in {self.window} seconds for user: {user}. Limit is {self.subscribed_requests} requests."
+            )
             raise HTTPException(
                 status_code=429,
                 detail="We're glad you're enjoying Khoj! You've exceeded your usage limit for today. Come back tomorrow or subscribe to increase your usage limit via [your settings](https://app.khoj.dev/settings).",
@@ -1073,6 +1083,9 @@ class ConversationCommandRateLimiter:
         ).acount()
 
         if subscribed and count_requests >= self.subscribed_rate_limit:
+            logger.info(
+                f"Rate limit: {count_requests} requests in 24 hours for user: {user}. Limit is {self.subscribed_rate_limit} requests."
+            )
             raise HTTPException(status_code=429, detail="Slow down! Too Many Requests")
         if not subscribed and count_requests >= self.trial_rate_limit:
             raise HTTPException(