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" };
} }

7
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) {
case RateLimiterMode.Preview:
return previewRateLimiter; return previewRateLimiter;
}else if(mode === RateLimiterMode.CrawlStatus){ case RateLimiterMode.CrawlStatus:
return crawlStatusRateLimiter; return crawlStatusRateLimiter;
}else{ default:
return serverRateLimiter; return serverRateLimiter;
} }
} }