From 5ca037d7691c769089a6b0c47f4290bf4ce399e5 Mon Sep 17 00:00:00 2001
From: "kiloconnect[bot]" <240665456+kiloconnect[bot]@users.noreply.github.com>
Date: Fri, 20 Feb 2026 10:15:57 +0000
Subject: [PATCH 1/2] perf: optimize free model usage IP limit counting in SQL

Move IP limit counting from JavaScript iteration to SQL HAVING clause.
Previously fetched all per-IP rows and counted in JS, now uses subquery
with HAVING COUNT(*) >= threshold to avoid transferring large result sets.
---
 .../admin/api/free-model-usage/stats/route.ts | 31 ++++++++-----------
 1 file changed, 13 insertions(+), 18 deletions(-)

diff --git a/src/app/admin/api/free-model-usage/stats/route.ts b/src/app/admin/api/free-model-usage/stats/route.ts
index 9eacb5ee9..4d6c5f07a 100644
--- a/src/app/admin/api/free-model-usage/stats/route.ts
+++ b/src/app/admin/api/free-model-usage/stats/route.ts
@@ -50,17 +50,20 @@ export async function GET(
       sql`${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(FREE_MODEL_RATE_LIMIT_WINDOW_HOURS))} hours'`
     );
 
-  // Get per-IP stats to find IPs at limits
-  const perIpStats = await db
+  // Count IPs at or above the rate limit threshold using a SQL subquery
+  const ipsAtLimitResult = await db
     .select({
-      ip_address: free_model_usage.ip_address,
-      request_count: sql<number>`COUNT(*)`,
+      count: sql<number>`COUNT(*)`,
     })
-    .from(free_model_usage)
-    .where(
-      sql`${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(FREE_MODEL_RATE_LIMIT_WINDOW_HOURS))} hours'`
-    )
-    .groupBy(free_model_usage.ip_address);
+    .from(
+      sql`(
+        SELECT ${free_model_usage.ip_address}
+        FROM ${free_model_usage}
+        WHERE ${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(FREE_MODEL_RATE_LIMIT_WINDOW_HOURS))} hours'
+        GROUP BY ${free_model_usage.ip_address}
+        HAVING COUNT(*) >= ${FREE_MODEL_MAX_REQUESTS_PER_WINDOW}
+      ) sub`
+    );
 
   // Get stats for the last 24 hours
   const dailyResult = await db
@@ -85,15 +88,7 @@ export async function GET(
 
   const windowUniqueIps = bigIntToNumber(windowStats.unique_ips);
   const windowTotalRequests = bigIntToNumber(windowStats.total_requests);
-
-  // Count IPs at or near limits
-  let ipsAtRequestLimit = 0;
-  for (const ip of perIpStats) {
-    const requestCount = bigIntToNumber(ip.request_count);
-    if (requestCount >= FREE_MODEL_MAX_REQUESTS_PER_WINDOW) {
-      ipsAtRequestLimit++;
-    }
-  }
+  const ipsAtRequestLimit = bigIntToNumber(ipsAtLimitResult[0]?.count ?? 0);
 
   return NextResponse.json({
     // Current window stats

From 985f0ab630004a4b13f8f5de7ac6bf4d19f1fb9c Mon Sep 17 00:00:00 2001
From: "kiloconnect[bot]" <240665456+kiloconnect[bot]@users.noreply.github.com>
Date: Fri, 20 Feb 2026 10:25:35 +0000
Subject: [PATCH 2/2] perf: optimize promoted model usage IP limit counting in
 SQL

---
 .../api/promoted-model-usage/stats/route.ts   | 31 ++++++++-----------
 1 file changed, 13 insertions(+), 18 deletions(-)

diff --git a/src/app/admin/api/promoted-model-usage/stats/route.ts b/src/app/admin/api/promoted-model-usage/stats/route.ts
index 8fc508de9..ba48ddd3e 100644
--- a/src/app/admin/api/promoted-model-usage/stats/route.ts
+++ b/src/app/admin/api/promoted-model-usage/stats/route.ts
@@ -39,17 +39,20 @@ export async function GET(
       sql`${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(PROMOTION_WINDOW_HOURS))} hours' AND ${ANONYMOUS_FILTER}`
     );
 
-  // Get per-IP stats to find IPs at limits (anonymous only)
-  const perIpStats = await db
+  // Count IPs at or above the promotion limit threshold using a SQL subquery (anonymous only)
+  const ipsAtLimitResult = await db
     .select({
-      ip_address: free_model_usage.ip_address,
-      request_count: sql<number>`COUNT(*)`,
+      count: sql<number>`COUNT(*)`,
     })
-    .from(free_model_usage)
-    .where(
-      sql`${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(PROMOTION_WINDOW_HOURS))} hours' AND ${ANONYMOUS_FILTER}`
-    )
-    .groupBy(free_model_usage.ip_address);
+    .from(
+      sql`(
+        SELECT ${free_model_usage.ip_address}
+        FROM ${free_model_usage}
+        WHERE ${free_model_usage.created_at} >= NOW() - INTERVAL '${sql.raw(String(PROMOTION_WINDOW_HOURS))} hours' AND ${ANONYMOUS_FILTER}
+        GROUP BY ${free_model_usage.ip_address}
+        HAVING COUNT(*) >= ${PROMOTION_MAX_REQUESTS}
+      ) sub`
+    );
 
   const bigIntToNumber = (value: unknown): number => {
     if (value === null || value === undefined) return 0;
@@ -62,15 +65,7 @@ export async function GET(
 
   const windowUniqueIps = bigIntToNumber(windowStats.unique_ips);
   const windowTotalRequests = bigIntToNumber(windowStats.total_requests);
-
-  // Count IPs at or near the promotion limit
-  let ipsAtRequestLimit = 0;
-  for (const ip of perIpStats) {
-    const requestCount = bigIntToNumber(ip.request_count);
-    if (requestCount >= PROMOTION_MAX_REQUESTS) {
-      ipsAtRequestLimit++;
-    }
-  }
+  const ipsAtRequestLimit = bigIntToNumber(ipsAtLimitResult[0]?.count ?? 0);
 
   return NextResponse.json({
     // Current window stats