Rate limit per user instead of per chat

teknologi-umum · Feb 4, 2024 · ab28e62 · ab28e62
1 parent 17b78c9
commit ab28e62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/BotNet.CommandHandlers/AI/Gemini/GeminiTextPromptHandler.cs b/BotNet.CommandHandlers/AI/Gemini/GeminiTextPromptHandler.cs
@@ -23,8 +23,8 @@ public sealed class GeminiTextPromptHandler(
 		ICommandQueue commandQueue,
 		ILogger<GeminiTextPromptHandler> logger
 	) : ICommandHandler<GeminiTextPrompt> {
-		internal static readonly RateLimiter CHAT_RATE_LIMITER = RateLimiter.PerChat(5, TimeSpan.FromMinutes(5));
-		internal static readonly RateLimiter VIP_CHAT_RATE_LIMITER = RateLimiter.PerChat(5, TimeSpan.FromMinutes(2));
+		internal static readonly RateLimiter CHAT_RATE_LIMITER = RateLimiter.PerUserPerChat(5, TimeSpan.FromMinutes(5));
+		internal static readonly RateLimiter VIP_CHAT_RATE_LIMITER = RateLimiter.PerUserPerChat(5, TimeSpan.FromMinutes(2));
 
 		private readonly ITelegramBotClient _telegramBotClient = telegramBotClient;
 		private readonly GeminiClient _geminiClient = geminiClient;