Update

src/utils/rate-limiter.ts

@@ -15,7 +15,7 @@ export class RateLimiter {
     this.queue = new PQueue({ concurrency: 1 });
   }
 
-  async throttle<T>(fn: () => Promise<T>): Promise<T> {
+  async throttle<T>(fn: () => Promise<T>, shouldCount: (result: T) => boolean = () => true): Promise<T> {
     return this.queue.add(async () => {
       const now = Date.now();
 
@@ -36,6 +36,7 @@ export class RateLimiter {
       }
 
       // Enforce minimum delay with jitter
+      // We check against the previously completed message time
       const timeSinceLastMessage = now - this.lastMessageTime;
       const minDelay = this.config.rateLimit.minDelayBetweenMessagesMs;
       const jitter = Math.random() * this.config.rateLimit.jitterMs;
@@ -46,10 +47,16 @@ export class RateLimiter {
         await this.sleep(waitTime);
       }
 
-      this.lastMessageTime = Date.now();
-      this.messagesThisHour++;
+      // Execute function
+      const result = await fn();
 
-      return fn();
+      // Update stats only if shouldCount returns true
+      if (shouldCount(result)) {
+        this.lastMessageTime = Date.now();
+        this.messagesThisHour++;
+      }
+
+      return result;
     }) as Promise<T>;
   }