fix acuc extract preview

This commit is contained in:
Gergő Móricz 2025-04-18 12:37:41 -07:00
parent 3caeaae074
commit f451b71308

View File

@ -148,7 +148,8 @@ export async function getACUC(
): Promise<AuthCreditUsageChunk | null> { ): Promise<AuthCreditUsageChunk | null> {
let isExtract = let isExtract =
mode === RateLimiterMode.Extract || mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.ExtractStatus; mode === RateLimiterMode.ExtractStatus ||
mode === RateLimiterMode.ExtractAgentPreview;
if (api_key === process.env.PREVIEW_TOKEN) { if (api_key === process.env.PREVIEW_TOKEN) {
const acuc = mockPreviewACUC(api_key, isExtract); const acuc = mockPreviewACUC(api_key, isExtract);
@ -264,7 +265,8 @@ export async function getACUCTeam(
): Promise<AuthCreditUsageChunkFromTeam | null> { ): Promise<AuthCreditUsageChunkFromTeam | null> {
let isExtract = let isExtract =
mode === RateLimiterMode.Extract || mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.ExtractStatus; mode === RateLimiterMode.ExtractStatus ||
mode === RateLimiterMode.ExtractAgentPreview;
if (team_id.startsWith("preview")) { if (team_id.startsWith("preview")) {
const acuc = mockPreviewACUC(team_id, isExtract); const acuc = mockPreviewACUC(team_id, isExtract);