chore: code re-factoring

epam · Mar 14, 2024 · 8f25ada · 8f25ada
1 parent 95c336a
commit 8f25ada
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/main/java/com/epam/aidial/core/limiter/RateLimiter.java b/src/main/java/com/epam/aidial/core/limiter/RateLimiter.java
@@ -193,11 +193,12 @@ private Limit getLimitByUser(ProxyContext context) {
             if (candidate != null) {
                 if (limit == null) {
                     limit = new Limit();
-                    limit.setMinute(0);
-                    limit.setDay(0);
+                    limit.setMinute(candidate.getMinute());
+                    limit.setDay(candidate.getDay());
+                } else {
+                    limit.setMinute(Math.max(candidate.getMinute(), limit.getMinute()));
+                    limit.setDay(Math.max(candidate.getDay(), limit.getDay()));
                 }
-                limit.setMinute(Math.max(candidate.getMinute(), limit.getMinute()));
-                limit.setDay(Math.max(candidate.getDay(), limit.getDay()));
             }
         }
         return limit == null ? defaultUserLimit : limit;