mirror of
https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl
synced 2025-08-12 04:59:02 +08:00
ok what
This commit is contained in:
parent
00cc733972
commit
321fff1695
@ -91,6 +91,10 @@ function checkCreditsMiddleware(
|
|||||||
|
|
||||||
export function authMiddleware(
|
export function authMiddleware(
|
||||||
rateLimiterMode: RateLimiterMode,
|
rateLimiterMode: RateLimiterMode,
|
||||||
|
reference?: {
|
||||||
|
extract: RateLimiterMode,
|
||||||
|
extractAgentPreview: RateLimiterMode,
|
||||||
|
}
|
||||||
): (req: RequestWithMaybeAuth, res: Response, next: NextFunction) => void {
|
): (req: RequestWithMaybeAuth, res: Response, next: NextFunction) => void {
|
||||||
const middlewareSpawn = new Error().stack;
|
const middlewareSpawn = new Error().stack;
|
||||||
return (req, res, next) => {
|
return (req, res, next) => {
|
||||||
@ -110,6 +114,7 @@ export function authMiddleware(
|
|||||||
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
|
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
|
||||||
},
|
},
|
||||||
middlewareSpawn,
|
middlewareSpawn,
|
||||||
|
reference,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -268,7 +273,10 @@ v1Router.ws("/crawl/:jobId", crawlStatusWSController);
|
|||||||
|
|
||||||
v1Router.post(
|
v1Router.post(
|
||||||
"/extract",
|
"/extract",
|
||||||
authMiddleware(RateLimiterMode.Extract),
|
authMiddleware(RateLimiterMode.Extract, {
|
||||||
|
extract: RateLimiterMode.Extract,
|
||||||
|
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
|
||||||
|
}),
|
||||||
checkCreditsMiddleware(1),
|
checkCreditsMiddleware(1),
|
||||||
wrap(extractController),
|
wrap(extractController),
|
||||||
);
|
);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user