AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-06-20 14:45:59 +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",
team_id,
sub_id: "bypass",
@ -100,8 +100,8 @@ const mockPreviewACUC: (team_id: string) => AuthCreditUsageChunkFromTeam = (team
bucketLimit: 25,
planModifier: 0.1,
},
concurrency: 2,
is_extract: false,
concurrency: is_extract ? 200 : 2,
is_extract,
});
const mockACUC: () => AuthCreditUsageChunk = () => ({
@ -145,6 +145,12 @@ export async function getACUC(
let isExtract =
mode === RateLimiterMode.Extract ||
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") {
const acuc = mockACUC();
@ -257,8 +263,7 @@ export async function getACUCTeam(
mode === RateLimiterMode.ExtractStatus;
if (team_id.startsWith("preview")) {
const acuc = mockPreviewACUC(team_id);
acuc.is_extract = isExtract;
const acuc = mockPreviewACUC(team_id, isExtract);
return acuc;
}