feat(quota): add Wafer.ai quota provider (#1312)

* feat(quota): add Wafer.ai quota provider - New provider: wafer.js fetches from https://pass.wafer.ai/v1/inference/quota - Auth: reads wafer/wafer-ai/wafer_ai keys from auth file - Timeout: AbortSignal.timeout(15_000) with timeoutSignal.aborted detection - Response: parses remaining/limit/overage/usedPercent/window_end/plan_tier - valueLabel: planTier + remaining/limit + overage suffix - Window: 5h (18000s) via resolveWindowLabel - Cross-runtime: added to web registry, UI types, VS Code dispatcher * fix(quota): wafer provider fixes — auth alias, decompression, logo - Add 'wafer.ai' auth alias to match actual auth key format - Use 'Accept-Encoding: identity' header to fix Bun fetch decompression issue with Cloudflare-backed responses - Match copilot valueLabel format: 'planTier · X / Y left' - Add wafer logo alias so Providers and Usage pages resolve to the same wafer.ai logo from models.dev * style(quota): fix indentation of timeoutSignal declaration * fix(quota): derive window duration from API instead of hardcoding Compute windowSeconds from window_end - window_start timestamps, with WAFER_WINDOW_SECONDS (5h) as fallback if timestamps are missing. --------- Co-authored-by: Bohdan Triapitsyn <artmore@protonmail.com>
2026-05-19 17:29:36 +03:00
parent 6970cb45c0
commit 4cfe7c80a3
8 changed files with 274 additions and 3 deletions
@@ -97,10 +97,19 @@ type ZhipuaiMcpTimeLimit = {
 type ZhipuaiPayload = {
  data?: {
    limits?: Array<ZhipuaiTokensLimit | ZhipuaiMcpTimeLimit>;
-    level?: string;
  };
 };

+type WaferPayload = {
+  remaining_included_requests?: number | string;
+  included_request_limit?: number | string;
+  overage_request_count?: number | string;
+  current_period_used_percent?: number | string;
+  window_start?: number | string;
+  window_end?: number | string;
+  plan_tier?: string;
+};
+
 export type ProviderResult = {
  providerId: string;
  providerName: string;
@@ -439,6 +448,11 @@ export const listConfiguredQuotaProviders = () => {
    configured.add('ollama-cloud');
  }

+  const waferAuth = normalizeAuthEntry(getAuthEntry(auth, ['wafer', 'wafer-ai', 'wafer_ai', 'wafer.ai']));
+  if (waferAuth && ((waferAuth as Record<string, unknown>).key || (waferAuth as Record<string, unknown>).token)) {
+    configured.add('wafer');
+  }
+
  return Array.from(configured);
 };

@@ -1738,6 +1752,116 @@ export const fetchNanoGptQuota = async (): Promise<ProviderResult> => {
  }
 };

+const WAFER_QUOTA_URL = 'https://pass.wafer.ai/v1/inference/quota';
+const WAFER_WINDOW_SECONDS = 5 * 3600;
+
+export const fetchWaferQuota = async (): Promise<ProviderResult> => {
+  const auth = readAuthFile();
+  const entry = normalizeAuthEntry(getAuthEntry(auth, ['wafer', 'wafer-ai', 'wafer_ai', 'wafer.ai'])) as Record<string, unknown> | null;
+  const apiKey = (entry?.key as string | undefined) ?? (entry?.token as string | undefined);
+
+  if (!apiKey) {
+    return buildResult({
+      providerId: 'wafer',
+      providerName: 'Wafer.ai',
+      ok: false,
+      configured: false,
+      error: 'Not configured',
+    });
+  }
+
+  const timeoutSignal = AbortSignal.timeout(15_000);
+
+  try {
+    const response = await fetch(WAFER_QUOTA_URL, {
+      method: 'GET',
+      headers: {
+        Authorization: `Bearer ${apiKey}`,
+        'Accept-Encoding': 'identity',
+      },
+      signal: timeoutSignal,
+    });
+
+    if (!response.ok) {
+      return buildResult({
+        providerId: 'wafer',
+        providerName: 'Wafer.ai',
+        ok: false,
+        configured: true,
+        error: `API error: ${response.status}`,
+      });
+    }
+
+    const payload = await response.json() as WaferPayload;
+    const remaining = toNumber(payload?.remaining_included_requests);
+    const limit = toNumber(payload?.included_request_limit);
+    const overage = toNumber(payload?.overage_request_count);
+    const usedPercentRaw = toNumber(payload?.current_period_used_percent);
+    const windowStart = toTimestamp(payload?.window_start);
+    const windowEnd = toTimestamp(payload?.window_end);
+    const planTier = asNonEmptyString(payload?.plan_tier);
+
+    if (remaining === null && limit === null && overage === null && usedPercentRaw === null) {
+      return buildResult({
+        providerId: 'wafer',
+        providerName: 'Wafer.ai',
+        ok: false,
+        configured: true,
+        error: 'No quota data in response',
+      });
+    }
+
+    const hasOverage = overage !== null && overage > 0;
+    const usedPercent = hasOverage
+      ? Math.max(0, usedPercentRaw ?? 0)
+      : Math.max(0, Math.min(100, usedPercentRaw ?? 0));
+
+    const windowSeconds = windowStart !== null && windowEnd !== null
+      ? Math.round((windowEnd - windowStart) / 1000)
+      : WAFER_WINDOW_SECONDS;
+    const windowLabel = resolveWindowLabel(windowSeconds);
+
+    let valueLabel: string | null = null;
+    if (remaining !== null && limit !== null) {
+      const parts: string[] = [];
+      if (planTier) parts.push(planTier);
+      parts.push(`${remaining} / ${limit} left`);
+      if (hasOverage) parts.push(`+${overage} overage`);
+      valueLabel = parts.join(' · ');
+    }
+
+    const windows: Record<string, UsageWindow> = {};
+    windows[windowLabel] = toUsageWindow({
+      usedPercent,
+      windowSeconds,
+      resetAt: windowEnd,
+      valueLabel,
+    });
+
+    return buildResult({
+      providerId: 'wafer',
+      providerName: 'Wafer.ai',
+      ok: true,
+      configured: true,
+      usage: { windows },
+    });
+  } catch (error) {
+    const isTimeout = error instanceof DOMException && error.name === 'AbortError' && timeoutSignal.aborted;
+    const isParseError = error instanceof SyntaxError;
+    return buildResult({
+      providerId: 'wafer',
+      providerName: 'Wafer.ai',
+      ok: false,
+      configured: true,
+      error: isTimeout
+        ? 'Request timed out'
+        : isParseError
+          ? 'Invalid response from provider'
+          : (error instanceof Error ? error.message : 'Request failed'),
+    });
+  }
+};
+
 export const fetchQuotaForProvider = async (providerId: string): Promise<ProviderResult> => {
  switch (providerId) {
    case 'claude':
@@ -1766,6 +1890,8 @@ export const fetchQuotaForProvider = async (providerId: string): Promise<Provide
      return fetchZaiQuota();
    case 'zhipuai-coding-plan':
      return fetchZhipuaiCodingPlanQuota();
+    case 'wafer':
+      return fetchWaferQuota();
    default:
      return buildResult({
        providerId,