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 { 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 { 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);