github/v-firecrawl
github/v-firecrawl
0
Fork 0
Code Actions

Nick:

Browse Source
This commit is contained in:
Nicolas 2024-04-20 14:10:29 -07:00
parent 43c2e877e7
commit 5b3c75b06e
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/api/src/index.ts
View File

@ -70,7 +70,7 @@ async function authenticateUser(req, res, mode?: RateLimiterMode): Promise<{ suc
return { success: false, error: "Rate limit exceeded. Too many requests, try again in 1 minute.", status: 429 }; return { success: false, error: "Rate limit exceeded. Too many requests, try again in 1 minute.", status: 429 };
} }
if (token === "this_is_just_a_preview_token" && mode === "scrape") { if (token === "this_is_just_a_preview_token" && (mode === RateLimiterMode.Scrape || mode === RateLimiterMode.Preview)) {
return { success: true, team_id: "preview" }; return { success: true, team_id: "preview" };
} }

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

@ -70,11 +70,12 @@ export function crawlRateLimit(plan: string){
export function getRateLimiter(mode: RateLimiterMode){ export function getRateLimiter(mode: RateLimiterMode){
if(mode === RateLimiterMode.Preview){ switch(mode) {
return previewRateLimiter; case RateLimiterMode.Preview:
}else if(mode === RateLimiterMode.CrawlStatus){ return previewRateLimiter;
return crawlStatusRateLimiter; case RateLimiterMode.CrawlStatus:
}else{ return crawlStatusRateLimiter;
return serverRateLimiter; default:
return serverRateLimiter;
} }
} }