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( 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),
); );