AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-12 02:18:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(v1): avoid overwriting rateLimiterMode with FIRE-1 rate limiter (#1593)

Browse Source
This commit is contained in:
Gergő Móricz 2025-05-23 16:50:59 +02:00 committed by GitHub
parent 3df687e4db
commit 3ec17e2d1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
apps/api/src/routes/v1.ts
View File

@ -90,39 +90,20 @@ function checkCreditsMiddleware(
}
export function authMiddleware(
rateLimiterMode: RateLimiterMode,
reference?: {
extract: RateLimiterMode,
extractAgentPreview: RateLimiterMode,
}
rateLimiterMode: RateLimiterMode
): (req: RequestWithMaybeAuth, res: Response, next: NextFunction) => void {
const middlewareSpawn = new Error().stack;
return (req, res, next) => {
(async () => {
if (rateLimiterMode === RateLimiterMode.Extract && isAgentExtractModelValid((req.body as any)?.agent?.model)) {
logger.debug("Picking extract agent preview mode", {
body: req.body,
model: (req.body as any)?.agent?.model,
});
rateLimiterMode = RateLimiterMode.ExtractAgentPreview;
} else if (rateLimiterMode === RateLimiterMode.ExtractAgentPreview) {
logger.warn("EAP passed into authMiddleware directly?", {
body: req.body,
callStack: new Error().stack,
modes: {
extract: RateLimiterMode.Extract,
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
},
middlewareSpawn,
reference,
});
let currentRateLimiterMode = rateLimiterMode;
if (currentRateLimiterMode === RateLimiterMode.Extract && isAgentExtractModelValid((req.body as any)?.agent?.model)) {
currentRateLimiterMode = RateLimiterMode.ExtractAgentPreview;
}
// if (rateLimiterMode === RateLimiterMode.Scrape && isAgentExtractModelValid((req.body as any)?.agent?.model)) {
// rateLimiterMode = RateLimiterMode.ScrapeAgentPreview;
// if (currentRateLimiterMode === RateLimiterMode.Scrape && isAgentExtractModelValid((req.body as any)?.agent?.model)) {
// currentRateLimiterMode = RateLimiterMode.ScrapeAgentPreview;
// }
const auth = await authenticateUser(req, res, rateLimiterMode);
const auth = await authenticateUser(req, res, currentRateLimiterMode);
if (!auth.success) {
if (!res.headersSent) {
@ -273,10 +254,7 @@ v1Router.ws("/crawl/:jobId", crawlStatusWSController);
v1Router.post(
"/extract",
authMiddleware(RateLimiterMode.Extract, {
extract: RateLimiterMode.Extract,
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
}),
authMiddleware(RateLimiterMode.Extract),
checkCreditsMiddleware(1),
wrap(extractController),
);