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

Nick:

Browse Source
This commit is contained in:
Nicolas 2024-08-26 16:05:11 -03:00
parent d591e0f51c
commit 5606fe5870
2 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -222,7 +222,8 @@ export async function supaAuthenticateUser(
rateLimiter = getRateLimiter( rateLimiter = getRateLimiter(
RateLimiterMode.Scrape, RateLimiterMode.Scrape,
token, token,
subscriptionData.plan subscriptionData.plan,
teamId
); );
break; break;
case RateLimiterMode.Search: case RateLimiterMode.Search:

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

@ -84,16 +84,28 @@ export const testSuiteRateLimiter = new RateLimiterRedis({
duration: 60, // Duration in seconds duration: 60, // Duration in seconds
}); });
export const devBRateLimiter = new RateLimiterRedis({
storeClient: redisRateLimitClient,
keyPrefix: "dev-b",
points: 1200,
duration: 60, // Duration in seconds
});
export function getRateLimiter( export function getRateLimiter(
mode: RateLimiterMode, mode: RateLimiterMode,
token: string, token: string,
plan?: string plan?: string,
teamId?: string
) { ) {
if (token.includes("a01ccae") || token.includes("6254cf9") || token.includes("0f96e673")) { if (token.includes("a01ccae") || token.includes("6254cf9") || token.includes("0f96e673")) {
return testSuiteRateLimiter; return testSuiteRateLimiter;
} }
if(teamId === process.env.DEV_B_TEAM_ID) {
return devBRateLimiter;
}
const rateLimitConfig = RATE_LIMITS[mode]; // {default : 5} const rateLimitConfig = RATE_LIMITS[mode]; // {default : 5}
if (!rateLimitConfig) return serverRateLimiter; if (!rateLimitConfig) return serverRateLimiter;