From 14f696805c397b9949ce53787be0c5d6b1ff103c Mon Sep 17 00:00:00 2001 From: Nicolas Date: Wed, 8 Jan 2025 17:04:57 -0300 Subject: [PATCH] Update auth.ts --- apps/api/src/controllers/auth.ts | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/apps/api/src/controllers/auth.ts b/apps/api/src/controllers/auth.ts index d344625d..3089ae23 100644 --- a/apps/api/src/controllers/auth.ts +++ b/apps/api/src/controllers/auth.ts @@ -204,7 +204,7 @@ export async function supaAuthenticateUser( } chunk = await getACUC(normalizedApi); - + if (chunk === null) { return { success: false, @@ -251,6 +251,13 @@ export async function supaAuthenticateUser( subscriptionData.plan, ); break; + case RateLimiterMode.Extract: + rateLimiter = getRateLimiter( + RateLimiterMode.Extract, + token, + subscriptionData.plan, + ); + break; case RateLimiterMode.CrawlStatus: rateLimiter = getRateLimiter(RateLimiterMode.CrawlStatus, token); break; @@ -304,6 +311,7 @@ export async function supaAuthenticateUser( mode === RateLimiterMode.Map || mode === RateLimiterMode.Crawl || mode === RateLimiterMode.CrawlStatus || + mode === RateLimiterMode.Extract || mode === RateLimiterMode.Search) ) { return { success: true, team_id: "preview", chunk: null };