From e026917698edf1bb7565a50c0e7987979b3ceff2 Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Fri, 31 Jan 2025 01:10:59 -0300
Subject: [PATCH] Nick: disable auto rech for extract requests

---
 apps/api/src/controllers/auth.ts                | 5 +++--
 apps/api/src/controllers/v1/types.ts            | 1 +
 apps/api/src/services/billing/credit_billing.ts | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/apps/api/src/controllers/auth.ts b/apps/api/src/controllers/auth.ts
index 8d16acf5..db5dd229 100644
--- a/apps/api/src/controllers/auth.ts
+++ b/apps/api/src/controllers/auth.ts
@@ -94,8 +94,9 @@ export async function getACUC(
     let retries = 0;
     const maxRetries = 5;
 
+    let isExtract = (mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus)
     let rpcName =
-      mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus
+      isExtract
         ? "auth_credit_usage_chunk_extract"
         : "auth_credit_usage_chunk_test_22_credit_pack_n_extract";
     while (retries < maxRetries) {
@@ -132,7 +133,7 @@ export async function getACUC(
       setCachedACUC(api_key, chunk);
     }
 
-    return chunk;
+    return chunk ? { ...chunk, is_extract: isExtract } : null;
   } else {
     return null;
   }
diff --git a/apps/api/src/controllers/v1/types.ts b/apps/api/src/controllers/v1/types.ts
index 465810ba..c3ba3a69 100644
--- a/apps/api/src/controllers/v1/types.ts
+++ b/apps/api/src/controllers/v1/types.ts
@@ -655,6 +655,7 @@ export type AuthCreditUsageChunk = {
   remaining_credits: number;
   sub_user_id: string | null;
   total_credits_sum: number;
+  is_extract?: boolean;
 };
 
 export interface RequestWithMaybeACUC<
diff --git a/apps/api/src/services/billing/credit_billing.ts b/apps/api/src/services/billing/credit_billing.ts
index 082a32d2..d3827f30 100644
--- a/apps/api/src/services/billing/credit_billing.ts
+++ b/apps/api/src/services/billing/credit_billing.ts
@@ -150,7 +150,8 @@ export async function supaCheckTeamCredits(
 
   if (
     isAutoRechargeEnabled &&
-    chunk.remaining_credits < autoRechargeThreshold
+    chunk.remaining_credits < autoRechargeThreshold &&
+    !chunk.is_extract
   ) {
     const autoChargeResult = await autoCharge(chunk, autoRechargeThreshold);
     if (autoChargeResult.success) {