From 321fff16954d581cd4a986c54ce04d282ba0d61b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gerg=C5=91=20M=C3=B3ricz?= <mo.geryy@gmail.com>
Date: Fri, 23 May 2025 11:41:34 +0200
Subject: [PATCH] ok what

---
 apps/api/src/routes/v1.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/apps/api/src/routes/v1.ts b/apps/api/src/routes/v1.ts
index d7cc07a1..c05bf585 100644
--- a/apps/api/src/routes/v1.ts
+++ b/apps/api/src/routes/v1.ts
@@ -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),
 );