AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-06-21 00:33:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(auth): preview acuc team more

Browse Source
This commit is contained in:
Gergő Móricz 2025-04-13 23:21:34 -07:00
parent 713d58675e
commit 07cdde7409
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/api/src/controllers/auth.ts
View File

@ -72,7 +72,7 @@ export async function setCachedACUC(
} }
} }
const mockPreviewACUC: (team_id: string) => AuthCreditUsageChunkFromTeam = (team_id) => ({ const mockPreviewACUC: (team_id: string, is_extract: boolean) => AuthCreditUsageChunk = (team_id, is_extract) => ({
api_key: "preview", api_key: "preview",
team_id, team_id,
sub_id: "bypass", sub_id: "bypass",
@ -100,8 +100,8 @@ const mockPreviewACUC: (team_id: string) => AuthCreditUsageChunkFromTeam = (team
bucketLimit: 25, bucketLimit: 25,
planModifier: 0.1, planModifier: 0.1,
}, },
concurrency: 2, concurrency: is_extract ? 200 : 2,
is_extract: false, is_extract,
}); });
const mockACUC: () => AuthCreditUsageChunk = () => ({ const mockACUC: () => AuthCreditUsageChunk = () => ({
@ -146,6 +146,12 @@ export async function getACUC(
mode === RateLimiterMode.Extract || mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.ExtractStatus; mode === RateLimiterMode.ExtractStatus;
if (api_key === process.env.PREVIEW_TOKEN) {
const acuc = mockPreviewACUC(api_key, isExtract);
acuc.is_extract = isExtract;
return acuc;
}
if (process.env.USE_DB_AUTHENTICATION !== "true") { if (process.env.USE_DB_AUTHENTICATION !== "true") {
const acuc = mockACUC(); const acuc = mockACUC();
acuc.is_extract = isExtract; acuc.is_extract = isExtract;
@ -257,8 +263,7 @@ export async function getACUCTeam(
mode === RateLimiterMode.ExtractStatus; mode === RateLimiterMode.ExtractStatus;
if (team_id.startsWith("preview")) { if (team_id.startsWith("preview")) {
const acuc = mockPreviewACUC(team_id); const acuc = mockPreviewACUC(team_id, isExtract);
acuc.is_extract = isExtract;
return acuc; return acuc;
} }