Austin/firecrawl
1
0
Fork 0
mirror of https://github.com/mendableai/firecrawl.git synced 2024-11-16 11:42:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update rate-limiter.ts

Browse Source
This commit is contained in:
Nicolas 2024-06-06 11:56:00 -07:00
parent 1129d33321
commit 27a8fd0c3c
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/api/src/services/rate-limiter.ts
View File

@ -69,19 +69,19 @@ const createRateLimiter = (keyPrefix, points) =>
export const previewRateLimiter = createRateLimiter(
"preview",
RATE_LIMITS.preview
RATE_LIMITS.preview.default
);
export const serverRateLimiter = createRateLimiter(
"server",
RATE_LIMITS.account
RATE_LIMITS.account.default
);
export const crawlStatusRateLimiter = createRateLimiter(
"crawl-status",
RATE_LIMITS.crawlStatus
RATE_LIMITS.crawlStatus.default
);
export const testSuiteRateLimiter = createRateLimiter(
"test-suite",
RATE_LIMITS.testSuite
RATE_LIMITS.testSuite.default
);
export function getRateLimiter(
@ -89,6 +89,7 @@ export function getRateLimiter(
token: string,
plan?: string
) {
console.log(`${mode}-${plan}`);
if (token.includes("a01ccae") || token.includes("6254cf9")) {
return testSuiteRateLimiter;
}
@ -100,5 +101,6 @@ export function getRateLimiter(
const points =
rateLimitConfig[planKey] || rateLimitConfig.default || rateLimitConfig; // 5
console.log(`${mode}-${planKey}`);
return createRateLimiter(`${mode}-${planKey}`, points);
}