Function getACUC

apps/api/src/controllers/auth.ts:245–324

(
  api_key: string,
  cacheOnly = false,
  useCache = true,
  mode?: RateLimiterMode,
)

/** @public used by auto_charge.ts (disabled, Autumn handles auto-recharge) */
export async function getACUC(
  api_key: string,
  cacheOnly = false,
  useCache = true,
  mode?: RateLimiterMode,
): Promise<AuthCreditUsageChunk | null> {
  let isExtract =
    mode === RateLimiterMode.Extract ||
    mode === RateLimiterMode.ExtractStatus ||
    mode === RateLimiterMode.ExtractAgentPreview;

  if (api_key === config.PREVIEW_TOKEN) {
    const acuc = mockPreviewACUC(api_key, isExtract);
    acuc.is_extract = isExtract;
    return acuc;
  }

  if (config.USE_DB_AUTHENTICATION !== true) {
    const acuc = mockACUC();
    acuc.is_extract = isExtract;
    return acuc;
  }

  const cacheKeyACUC = `acuc_${api_key}_${isExtract ? "extract" : "scrape"}`;

  if (useCache) {
    const cachedACUC = await getValue(cacheKeyACUC);
    if (cachedACUC !== null) {
      return JSON.parse(cachedACUC);
    }
  }

  if (!cacheOnly) {
    let data: AuthCreditUsageChunkRow[] = [];
    let retries = 0;
    const maxRetries = 5;
    while (retries < maxRetries) {
      const database = Math.random() > 2 / 3 ? dbRr : db;
      try {
        data = await authCreditUsageChunk(database, api_key, isExtract);
        break;
      } catch (error) {
        logger.warn(
          `Failed to retrieve authentication and credit usage data after ${retries}, trying again...`,
          { error },
        );
        retries++;
        if (retries === maxRetries) {
          throw new Error(
            "Failed to retrieve authentication and credit usage data after 3 attempts: " +
              JSON.stringify(error),
          );
        }

        // Wait for a short time before retrying
        await new Promise(resolve => setTimeout(resolve, 200));
      }
    }