From f451b71308e368ac57c80a10587a68a28a557436 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gerg=C5=91=20M=C3=B3ricz?= <mo.geryy@gmail.com>
Date: Fri, 18 Apr 2025 12:37:41 -0700
Subject: [PATCH] fix acuc extract preview

---
 apps/api/src/controllers/auth.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/api/src/controllers/auth.ts b/apps/api/src/controllers/auth.ts
index e6a16cd7..c9c0e4b2 100644
--- a/apps/api/src/controllers/auth.ts
+++ b/apps/api/src/controllers/auth.ts
@@ -148,7 +148,8 @@ export async function getACUC(
 ): Promise<AuthCreditUsageChunk | null> {
   let isExtract =
       mode === RateLimiterMode.Extract ||
-      mode === RateLimiterMode.ExtractStatus;
+      mode === RateLimiterMode.ExtractStatus ||
+      mode === RateLimiterMode.ExtractAgentPreview;
 
   if (api_key === process.env.PREVIEW_TOKEN) {
     const acuc = mockPreviewACUC(api_key, isExtract);
@@ -264,7 +265,8 @@ export async function getACUCTeam(
 ): Promise<AuthCreditUsageChunkFromTeam | null> {
   let isExtract =
       mode === RateLimiterMode.Extract ||
-      mode === RateLimiterMode.ExtractStatus;
+      mode === RateLimiterMode.ExtractStatus ||
+      mode === RateLimiterMode.ExtractAgentPreview;
 
   if (team_id.startsWith("preview")) {
     const acuc = mockPreviewACUC(team_id, isExtract);