This commit is contained in:
Gergő Móricz 2025-05-23 11:41:34 +02:00
parent 00cc733972
commit 321fff1695

View File

@ -91,6 +91,10 @@ function checkCreditsMiddleware(
export function authMiddleware(
rateLimiterMode: RateLimiterMode,
reference?: {
extract: RateLimiterMode,
extractAgentPreview: RateLimiterMode,
}
): (req: RequestWithMaybeAuth, res: Response, next: NextFunction) => void {
const middlewareSpawn = new Error().stack;
return (req, res, next) => {
@ -110,6 +114,7 @@ export function authMiddleware(
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
},
middlewareSpawn,
reference,
});
}
@ -268,7 +273,10 @@ v1Router.ws("/crawl/:jobId", crawlStatusWSController);
v1Router.post(
"/extract",
authMiddleware(RateLimiterMode.Extract),
authMiddleware(RateLimiterMode.Extract, {
extract: RateLimiterMode.Extract,
extractAgentPreview: RateLimiterMode.ExtractAgentPreview,
}),
checkCreditsMiddleware(1),
wrap(extractController),
);