fix acuc extract preview
This commit is contained in:
@@ -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);
|
||||||
|
|||||||
Reference in New Issue
Block a user