From 7216799ca020b2d39ace19509a1d1f8c3b86c72a Mon Sep 17 00:00:00 2001
From: rafaelmmiller <150964962+rafaelsideguide@users.noreply.github.com>
Date: Wed, 2 Apr 2025 10:45:11 -0300
Subject: [PATCH] revert mog changes

---
 apps/api/src/controllers/auth.ts              | 3 ++-
 apps/api/src/controllers/v1/crawl-status.ts   | 2 +-
 apps/api/src/lib/supabase-jobs.ts             | 8 ++++----
 apps/api/src/services/billing/auto_charge.ts  | 2 +-
 apps/api/src/services/idempotency/validate.ts | 2 +-
 5 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/apps/api/src/controllers/auth.ts b/apps/api/src/controllers/auth.ts
index 70a154c8..7b38181d 100644
--- a/apps/api/src/controllers/auth.ts
+++ b/apps/api/src/controllers/auth.ts
@@ -97,7 +97,8 @@ export async function getACUC(
       mode === RateLimiterMode.Extract ||
       mode === RateLimiterMode.ExtractStatus;
     while (retries < maxRetries) {
-      const client = supabase_service;
+      const client =
+        Math.random() > (2/3) ? supabase_rr_service : supabase_service;
       ({ data, error } = await client.rpc(
         "auth_credit_usage_chunk_27_tally",
         { input_key: api_key, i_is_extract: isExtract, tally_untallied_credits: true },
diff --git a/apps/api/src/controllers/v1/crawl-status.ts b/apps/api/src/controllers/v1/crawl-status.ts
index 7e847ad5..96aa578e 100644
--- a/apps/api/src/controllers/v1/crawl-status.ts
+++ b/apps/api/src/controllers/v1/crawl-status.ts
@@ -246,7 +246,7 @@ export async function crawlStatusController(
   let totalCount = jobIDs.length;
 
   if (totalCount === 0 && process.env.USE_DB_AUTHENTICATION === "true") {
-    const x = await supabase_service
+    const x = await supabase_rr_service
       .from('firecrawl_jobs')
       .select('*', { count: 'exact', head: true })
       .eq("crawl_id", req.params.jobId)
diff --git a/apps/api/src/lib/supabase-jobs.ts b/apps/api/src/lib/supabase-jobs.ts
index bf25c62f..e36f3b97 100644
--- a/apps/api/src/lib/supabase-jobs.ts
+++ b/apps/api/src/lib/supabase-jobs.ts
@@ -8,7 +8,7 @@ import * as Sentry from "@sentry/node";
  * @returns {any | null} Job
  */
 export const supabaseGetJobById = async (jobId: string) => {
-  const { data, error } = await supabase_service
+  const { data, error } = await supabase_rr_service
     .from("firecrawl_jobs")
     .select("*")
     .eq("job_id", jobId)
@@ -31,7 +31,7 @@ export const supabaseGetJobById = async (jobId: string) => {
  * @returns {any[]} Jobs
  */
 export const supabaseGetJobsById = async (jobIds: string[]) => {
-  const { data, error } = await supabase_service
+  const { data, error } = await supabase_rr_service
     .from("firecrawl_jobs")
     .select()
     .in("job_id", jobIds);
@@ -55,7 +55,7 @@ export const supabaseGetJobsById = async (jobIds: string[]) => {
  * @returns {any[]} Jobs
  */
 export const supabaseGetJobsByCrawlId = async (crawlId: string) => {
-  const { data, error } = await supabase_service
+  const { data, error } = await supabase_rr_service
     .from("firecrawl_jobs")
     .select()
     .eq("crawl_id", crawlId);
@@ -74,7 +74,7 @@ export const supabaseGetJobsByCrawlId = async (crawlId: string) => {
 };
 
 export const supabaseGetJobByIdOnlyData = async (jobId: string) => {
-  const { data, error } = await supabase_service
+  const { data, error } = await supabase_rr_service
     .from("firecrawl_jobs")
     .select("docs, team_id")
     .eq("job_id", jobId)
diff --git a/apps/api/src/services/billing/auto_charge.ts b/apps/api/src/services/billing/auto_charge.ts
index 8540b1ea..b30be201 100644
--- a/apps/api/src/services/billing/auto_charge.ts
+++ b/apps/api/src/services/billing/auto_charge.ts
@@ -124,7 +124,7 @@ export async function autoCharge(
           if (chunk.sub_user_id) {
             // Fetch the customer's Stripe information
             const { data: customer, error: customersError } =
-              await supabase_service
+              await supabase_rr_service
                 .from("customers")
                 .select("id, stripe_customer_id")
                 .eq("id", chunk.sub_user_id)
diff --git a/apps/api/src/services/idempotency/validate.ts b/apps/api/src/services/idempotency/validate.ts
index 4a5fbe65..54ec7bd0 100644
--- a/apps/api/src/services/idempotency/validate.ts
+++ b/apps/api/src/services/idempotency/validate.ts
@@ -18,7 +18,7 @@ export async function validateIdempotencyKey(req: Request): Promise<boolean> {
     return false;
   }
 
-  const { data, error } = await supabase_service
+  const { data, error } = await supabase_rr_service
     .from("idempotency_keys")
     .select("key")
     .eq("key", idempotencyKey);