Update auth.ts

This commit is contained in:
Nicolas 2025-01-19 13:23:51 -03:00
parent 382476cb36
commit e4b45e9e7c

View File

@ -94,7 +94,10 @@ export async function getACUC(
let retries = 0; let retries = 0;
const maxRetries = 5; const maxRetries = 5;
let rpcName = mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus ? "auth_credit_usage_chunk_extract" : "auth_credit_usage_chunk_test_21_credit_pack_n_extract"; let rpcName =
mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus
? "auth_credit_usage_chunk_extract"
: "auth_credit_usage_chunk_test_21_credit_pack_n_extract";
while (retries < maxRetries) { while (retries < maxRetries) {
({ data, error } = await supabase_service.rpc( ({ data, error } = await supabase_service.rpc(
rpcName, rpcName,
@ -203,7 +206,7 @@ export async function supaAuthenticateUser(
}; };
} }
chunk = await getACUC(normalizedApi,false, true, mode); chunk = await getACUC(normalizedApi, false, true, mode);
if (chunk === null) { if (chunk === null) {
return { return {
@ -258,6 +261,9 @@ export async function supaAuthenticateUser(
subscriptionData.plan, subscriptionData.plan,
); );
break; break;
case RateLimiterMode.ExtractStatus:
rateLimiter = getRateLimiter(RateLimiterMode.ExtractStatus, token);
break;
case RateLimiterMode.CrawlStatus: case RateLimiterMode.CrawlStatus:
rateLimiter = getRateLimiter(RateLimiterMode.CrawlStatus, token); rateLimiter = getRateLimiter(RateLimiterMode.CrawlStatus, token);
break; break;