Compare commits

..

2 Commits

Author SHA1 Message Date
adevwithpurpose
7d7c6ac2ac fix(usage): render AgentRouter wallet balance as USD in the Quota UI (#10078)
The prior fix wired AgentRouter's balance into getUsageForProvider() and
USAGE_SUPPORTED_PROVIDERS, but the actual dollar figure never reached the
Dashboard Quota UI: quotas.balance.remaining carried a synthetic two-state
percent (100/0) instead of the real dollarBalance, and the Provider Limits
renderer only formats a row as "$X.XX" when isCredits/currency/creditCount
are set, which the generic quota-parsing path never sets. A configured
balance rendered as a bare "100% left" percentage, not USD.

Shape quotas.balance.remaining as the real USD amount (clamped to 0) and add
an agentrouter branch to quotaParsing.ts that builds a credits-style row
(same buildCreditsQuota() pattern as DeepSeek/Claude extra-usage), so a
configured balance shows a currency-formatted dollar amount and an
exhausted balance always renders as exactly $0.00.
2026-08-17 22:40:07 -03:00
adevwithpurpose
3338fb8495 fix(usage): wire agentrouter balance quota into dashboard Quota UI (#10078) 2026-08-15 11:15:06 -03:00
12 changed files with 330 additions and 217 deletions

View File

@@ -0,0 +1,2 @@
- Fix: wire AgentRouter's existing console balance fetcher into the Dashboard Quota UI (visibility gate + provider-limits data path + background sync) so its wallet balance renders instead of falling back to "Usage API not implemented" (#10078)
- Fix: AgentRouter's dollar balance now renders as a currency-formatted "$X.XX" credits row in the Dashboard Quota UI instead of a bare percentage, and an exhausted wallet always shows exactly $0.00 (#10078)

View File

@@ -1 +0,0 @@
- fix(resilience): keep combo quality and auth failure reasons separate and redact connection labels in terminal errors (#10314)

View File

@@ -93,13 +93,6 @@ import {
expandPromptCacheAffinityTargetsFromConnections,
resolvePromptCacheAffinityKey,
} from "./combo/promptCacheAffinity.ts";
import {
classifyComboOutcome,
formatComboOutcomes,
redactConnectionLabel,
buildRedactedSummary,
} from "./combo/comboErrorAggregation.ts";
import type { ComboErrorEntry } from "./combo/comboErrorAggregation.ts";
import type { CompressionMode } from "./compression/types.ts";
import { getCachedProviderConnections } from "../../src/lib/db/readCache";
import { isProviderInCooldown, recordProviderCooldown } from "./providerCooldownTracker.ts";
@@ -860,7 +853,7 @@ export async function handleComboChat({
let comboExpired = false;
// Accumulator for per-model error details across targets in the current set try.
// Reset at the start of each set retry (same lifecycle as lastError/recordedAttempts).
let comboErrors: Array<ComboErrorEntry> = [];
let comboErrors: Array<{ model: string; status: number; error: string }> = [];
// Quota trust spans set retries and recursive cooldown re-dispatches. Once any
// failure is non-quota, a nested caller must never treat this dispatch as quota-only.
let observedFailure = false;
@@ -1350,15 +1343,6 @@ export async function handleComboChat({
// misleading ALL_ACCOUNTS_INACTIVE when the real issue is quality.
lastError = `Upstream response failed quality validation: ${quality.reason}`;
lastStatus = 502;
// #10314: record quality failures as a FIRST-CLASS per-target outcome
// so a quality reason is never silently dropped from the aggregated
// terminal message when a later sibling overwrites lastError.
comboErrors.push({
model: modelStr,
status: 502,
error: quality.reason || "upstream response failed quality validation",
kind: "quality",
});
if (i > 0) fallbackCount++;
if (provider && rawModel) {
const mlSettings = resolveModelLockoutSettings(settings);
@@ -1866,7 +1850,6 @@ export async function handleComboChat({
model: modelStr,
status: result.status,
error: errorText || String(result.status),
kind: classifyComboOutcome(result.status, errorText),
});
lastStatus = result.status;
if (i > 0) fallbackCount++;
@@ -2060,7 +2043,6 @@ export async function handleComboChat({
model: modelStr,
status: result.status,
error: errorText || String(result.status),
kind: classifyComboOutcome(result.status, errorText),
});
lastStatus = result.status;
if (i > 0) fallbackCount++;
@@ -2215,10 +2197,15 @@ export async function handleComboChat({
// Global combo timeout: return aggregated error immediately, skipping set retries.
if (comboExpired) {
const summary = buildRedactedSummary(comboErrors);
const summary = comboErrors
.slice(0, 5)
.map((e) => `${e.model} (${e.status})`)
.join(", ");
const msg =
`Combo global timeout (${comboTimeoutMs}ms) after ${recordedAttempts}/${orderedTargets.length} targets` +
(comboErrors.length > 0 ? ` | tried: ${summary}` : "");
(comboErrors.length > 0
? ` | tried: ${summary}${comboErrors.length > 5 ? `... (+${comboErrors.length - 5})` : ""}`
: "");
const latencyMs = Date.now() - startTime;
if (recordedAttempts === 0) {
recordComboRequest(combo.name, null, {
@@ -2289,12 +2276,18 @@ export async function handleComboChat({
}
const status = lastStatus;
// #10314: build the terminal message from the structured per-target
// outcomes (each distinct class+reason listed separately) instead of
// mashing a single lastError with raw `[model (status)]` markers. Connection
// identifiers are redacted. Falls back to lastError when no target recorded
// a structured outcome.
const msg = formatComboOutcomes(comboErrors) || lastError || "All combo models unavailable";
// Build aggregated error message with per-model failure details for diagnostics.
const comboErrorSummary =
comboErrors.length > 0
? " [" +
comboErrors
.slice(0, 5)
.map((e) => `${e.model} (${e.status})`)
.join(", ") +
(comboErrors.length > 5 ? `... (+${comboErrors.length - 5})` : "") +
"]"
: "";
const msg = (lastError || "All combo models unavailable") + comboErrorSummary;
// Cooldown-aware retry: instead of crystallizing a transient failure, wait
// out a SHORT cooldown and re-run the whole set loop. Guarded by the helper
@@ -2722,10 +2715,6 @@ async function handleRoundRobinCombo({
let globalAttempts = 0;
let fallbackCount = 0;
let recordedAttempts = 0;
// #10314: per-target outcome accumulator for the round-robin twin so the
// terminal message lists each distinct reason separately (see the quality path
// and the "Done with this model" path below), mirroring handleComboChat.
const rrOutcomes: Array<ComboErrorEntry> = [];
// #1731: Per-request in-memory set of providers whose quota is fully exhausted.
// When a target returns a quota-exhausted 429, remaining targets from the same
@@ -2922,12 +2911,6 @@ async function handleRoundRobinCombo({
// misleading ALL_ACCOUNTS_INACTIVE when the real issue is quality.
lastError = `Upstream response failed quality validation: ${quality.reason}`;
lastStatus = 502;
rrOutcomes.push({
model: modelStr,
status: 502,
error: quality.reason || "upstream response failed quality validation",
kind: "quality",
});
if (offset > 0) fallbackCount++;
break; // move to next model
}
@@ -3234,12 +3217,6 @@ async function handleRoundRobinCombo({
recordedAttempts++;
lastError = errorText || String(result.status);
lastStatus = result.status;
rrOutcomes.push({
model: modelStr,
status: result.status,
error: errorText || String(result.status),
kind: classifyComboOutcome(result.status, errorText),
});
if (offset > 0) fallbackCount++;
log.warn("COMBO-RR", `${modelStr} failed, trying next model`, { status: result.status });
@@ -3360,10 +3337,7 @@ async function handleRoundRobinCombo({
}
const status = lastStatus;
// #10314: same structured per-target aggregation as handleComboChat — list each
// distinct reason separately (redacted), fall back to lastError when no outcome.
const msg =
formatComboOutcomes(rrOutcomes) || lastError || "All round-robin combo models unavailable";
const msg = lastError || "All round-robin combo models unavailable";
if (earliestRetryAfter && isRetryAfterEligibleStatus(status)) {
const retryHuman = formatRetryAfter(toRetryAfterDisplayValue(earliestRetryAfter));

View File

@@ -1,115 +0,0 @@
/**
* Shared combo terminal-error aggregation.
*
* #10314 — combo error aggregation mixes quality and auth. Prior to this module
* the combo terminal message was built as a single `lastError` string (last
* writer wins — it can only ever represent ONE target's reason) concatenated
* with a raw `[model (status)]` suffix. A quality-failure reason from one
* target and a sibling's 401 were collapsed into one client-facing sentence
* (`invalid_api_key [openai/proxy-account-b (401)]`) and a quality reason that
* was not the final failing target was dropped entirely.
*
* This module gives each per-target failure a structured {model, status, error,
* kind} entry, so the terminal message can list every distinct reason
* separately (and classification-labelled) instead of mashing them, and it
* redacts connection/account identifiers that, on openai-compatible proxy
* connections, used to surface verbatim in client-visible and shared-warn
* strings (ops/PII leak).
*/
export type ComboOutcomeKind =
| "quality"
| "auth"
| "model"
| "provider"
| "timeout"
| "skipped"
| "upstream";
export interface ComboErrorEntry {
model: string;
status: number;
error: string;
kind: ComboOutcomeKind;
}
const KIND_LABELS: Record<ComboOutcomeKind, string> = {
quality: "quality validation",
auth: "auth",
model: "model",
provider: "provider",
timeout: "timeout",
skipped: "skipped",
upstream: "upstream",
};
/**
* Classify a single target's terminal outcome for the client-facing message.
* Auth-class errors (401/403 or auth-sounding text) are kept distinct from
* model-class (400/422) and provider-class (5xx) so a sibling's 401 is never
* presented as "quality failed". Fall through to `model` for everything else.
*/
export function classifyComboOutcome(status: number, errorText: string): ComboOutcomeKind {
const text = typeof errorText === "string" ? errorText : "";
if (
status === 401 ||
status === 403 ||
/(invalid.?api.?key|unauthorized|not.?authorized|auth(entication|orization)?)/i.test(text)
) {
return "auth";
}
if (status === 408 || status >= 499) return "timeout";
if (status >= 500) return "provider";
return "model";
}
/**
* Redact connection/account identifiers that can ride inside a proxy target's
* model string (openai-compatible proxy model names often carry a connection
* label). UUIDs and long hex hashes are truncated to a short `conn:` prefix.
* Provider/model names operators need for debugging are left intact.
*/
export function redactConnectionLabel(modelStr: string | null | undefined): string {
const label = typeof modelStr === "string" && modelStr ? modelStr : "unknown";
return label
.replace(
/\b[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}\b/g,
(m) => `conn:${m.slice(0, 8)}`
)
.replace(/\b[0-9a-fA-F]{16,}\b/g, (m) => `conn:${m.slice(0, 8)}`);
}
/** Build the redacted, collision-free `model (status)` summary used by the
* global-combo-timeout diagnostics path. */
export function buildRedactedSummary(
entries: Array<{ model: string; status: number }> | ReadonlyArray<{ model: string; status: number }>
): string {
const slice = entries.slice(0, 5);
const parts = slice.map((e) => `${redactConnectionLabel(e.model)} (${e.status})`).join(", ");
return entries.length > 5 ? `${parts}... (+${entries.length - 5})` : parts;
}
/**
* Format per-target terminal outcomes into one client-facing sentence that keeps
* every distinct reason separate (and classification-labelled) instead of
* mashing a single `lastError` with raw status markers. Always redacts
* connection identifiers unless `{ redact: false }` is explicitly passed.
*/
export function formatComboOutcomes(
entries: ReadonlyArray<{ model: string; status: number; error: string; kind?: ComboOutcomeKind }>,
opts?: { redact?: boolean }
): string {
if (!entries.length) return "";
const redact = opts?.redact !== false;
const slice = entries.slice(0, 5);
const parts = slice.map((e) => {
const label = redact ? redactConnectionLabel(e.model) : e.model;
const kind = e.kind ? KIND_LABELS[e.kind] ?? e.kind : null;
const reason = e.error || `HTTP ${e.status}`;
const statusTxt = ` (HTTP ${e.status})`;
return kind ? `${label}: ${kind}${reason}${statusTxt}` : `${label}: ${reason}${statusTxt}`;
});
return entries.length > 5
? `${parts.join("; ")}... (+${entries.length - 5} more)`
: parts.join("; ");
}

View File

@@ -71,6 +71,7 @@ import { getFirecrawlUsage } from "./usage/firecrawl.ts";
import { getCommandCodeUsage } from "./usage/command-code.ts";
import { getQwenTokenPlanUsage } from "./usage/qwen-token-plan.ts";
import { getConolUsage } from "./conolUsage.ts";
import { getAgentrouterUsage } from "./usage/agentrouter.ts";
type JsonRecord = Record<string, unknown>;
type UsageProviderConnection = JsonRecord & {
@@ -138,6 +139,8 @@ export const USAGE_FETCHER_PROVIDERS = [
"command-code",
"conol-web",
"cnl",
// AgentRouter (New-API) console balance (GET /api/user/self)
"agentrouter",
] as const;
export type UsageFetcherProvider = (typeof USAGE_FETCHER_PROVIDERS)[number];
@@ -244,6 +247,8 @@ export async function getUsageForProvider(
case "conol-web":
case "cnl":
return await getConolUsage(apiKey || accessToken, providerSpecificData);
case "agentrouter":
return await getAgentrouterUsage(id, connection);
default:
return { message: `Usage API not implemented for ${provider}` };
}

View File

@@ -0,0 +1,73 @@
/**
* usage/agentrouter.ts — AgentRouter (New-API) balance quota shapes the Provider
* Limits dashboard expects.
*
* Reuses the already-registered preflight/monitor fetcher (OpenAI-style routing
* apiKey vs console System Access Token + New-Api-User id) instead of re-implementing
* the HTTP call, so the 60s in-memory cache in agentrouterQuotaFetcher.ts is shared.
*
* AgentRouter exposes a raw New-API credit balance, not a real grant to divide by —
* so, following the DeepSeek boolean-availability precedent, `remainingPercentage` is
* only a two-state signal (100 = has balance, 0 = exhausted) used for the quota-card
* bar color. The human-meaningful number — the actual USD balance (rawQuota /
* QUOTA_PER_UNIT) — MUST travel inside `quotas.balance.remaining` so the Dashboard
* Quota UI's credits-row renderer (quotaParsing.ts::parseAgentrouterQuota, which reads
* `quota.remaining`/`quota.currency`) can format it with a currency symbol instead of
* dropping it: `getUsageForProvider()`'s top-level `remainingUsd`/`availableUsd`/
* `balance` sibling fields exist for API/CLI consumers only — parseQuotaData() (the
* Dashboard renderer) never reads them, only `data.quotas` (#10078 follow-up).
*/
import { fetchAgentrouterQuota, type AgentrouterQuota } from "../agentrouterQuotaFetcher.ts";
import { type UsageQuota } from "./quota.ts";
type JsonRecord = Record<string, unknown>;
/**
* AgentRouter balance → dashboard usage shape.
*
* Returns `{ message }` when the fetch returns null (no console credentials, an
* upstream error, or a rejected token), which the Provider Limits UI renders as a
* graceful per-row status instead of crashing the whole page. Otherwise shapes the
* balance into a single USD `quotas.balance` entry whose `remaining` field carries
* the exact dollar amount (never negative, exactly 0 when the wallet is exhausted).
*/
export async function getAgentrouterUsage(
connectionId: string | undefined,
connection: JsonRecord
) {
const quota = (await fetchAgentrouterQuota(
connectionId || "",
connection
)) as AgentrouterQuota | null;
if (!quota) {
return {
message:
"AgentRouter balance not available. Add the Console API Key + New-API User ID to the connection to view usage.",
};
}
// `dollarBalance` is already `rawQuota / QUOTA_PER_UNIT` (agentrouterQuotaFetcher.ts);
// clamp defensively so an exhausted/mis-parsed wallet never surfaces as negative.
const remainingUsd = Math.max(0, quota.dollarBalance);
const remainingPercentage = quota.limitReached ? 0 : 100;
const balance: UsageQuota = {
used: 0,
total: 0,
remaining: remainingUsd,
remainingPercentage,
resetAt: quota.resetAt ?? null,
unlimited: true,
currency: "USD",
displayName: "Wallet Balance (USD)",
};
return {
plan: "AgentRouter",
quotas: { balance },
remainingUsd,
availableUsd: remainingUsd,
balance: remainingUsd,
};
}

View File

@@ -217,6 +217,27 @@ function parseDeepseek(data: any) {
return quotaEntries(data).map(([quotaKey, quota]) => parseDeepseekQuota(quotaKey, quota));
}
// #10078 follow-up: AgentRouter's `quotas.balance` entry (open-sse/services/usage/agentrouter.ts)
// carries a real USD amount in `remaining` + `currency: "USD"`. The generic path
// (normalizeQuotaEntry via parseGeneric) drops `currency` entirely and never sets
// `isCredits`/`creditCount`, so QuotaCardBody/QuotaCardExpanded's dollar-formatted
// renderer (which only activates on `q.isCredits`) never triggers — the balance was
// rendered as a bare "100%/0% left" percentage instead of "$X.XX". Route it through
// buildCreditsQuota() (same shape DeepSeek/Claude-extra-usage credits rows use) so the
// dollar figure — and an exhausted ($0.00) balance — render unambiguously as USD.
function parseAgentrouterQuota(quotaKey: string, quota: any) {
if (quotaKey !== "balance") return normalizeQuotaEntry(quotaKey, quota);
const remaining = Math.max(0, Number(quota?.remaining ?? 0));
const currency = quota?.currency || "USD";
const remainingPercentage =
safePercentage(quota?.remainingPercentage) ?? (remaining > 0 ? 100 : 0);
return buildCreditsQuota(currency, remaining, remainingPercentage, { currency });
}
function parseAgentrouter(data: any) {
return quotaEntries(data).map(([quotaKey, quota]) => parseAgentrouterQuota(quotaKey, quota));
}
function parseProviderQuotas(providerId: string, data: any) {
if (providerId === "github") return parseGithub(data);
if (["glm", "glm-cn", "glmt", "opencode-go"].includes(providerId)) return parseGlmFamily(data);
@@ -224,6 +245,7 @@ function parseProviderQuotas(providerId: string, data: any) {
if (providerId === "codex") return parseCodex(data);
if (providerId === "claude") return parseClaude(data);
if (providerId === "deepseek") return parseDeepseek(data);
if (providerId === "agentrouter") return parseAgentrouter(data);
return parseGeneric(data);
}

View File

@@ -100,6 +100,8 @@ const PROVIDER_LIMITS_APIKEY_PROVIDERS = new Set([
// Alibaba Coding Plan (console API key) + Qwen personal Token Plan (console cookie) — #9603
"bailian-coding-plan",
"qwen-cloud-token-plan",
// AgentRouter (New-API) console System Access Token + New-Api-User id (providerSpecificData)
"agentrouter",
]);
const DEFAULT_PROVIDER_LIMITS_SYNC_INTERVAL_MINUTES = 70;
const PROVIDER_LIMITS_AUTO_SYNC_SETTING_KEY = "provider_limits_auto_sync_last_run";

View File

@@ -504,6 +504,8 @@ export const USAGE_SUPPORTED_PROVIDERS = [
"bailian-coding-plan",
// Qwen Cloud / Model Studio personal Token Plan (cookie-authenticated console gateway)
"qwen-cloud-token-plan",
// AgentRouter (New-API) console balance quota (consoleApiKey + newApiUserId)
"agentrouter",
];
// ── Zod validation at module load (Phase 7.2) ──

View File

@@ -0,0 +1,117 @@
import test from "node:test";
import assert from "node:assert/strict";
import { getAgentrouterUsage } from "../../open-sse/services/usage/agentrouter.ts";
import { parseQuotaData } from "../../src/app/(dashboard)/dashboard/usage/components/ProviderLimits/quotaParsing.ts";
const originalFetch = globalThis.fetch;
test.afterEach(() => {
globalThis.fetch = originalFetch;
});
function mockAgentrouterFetch(rawQuota: number) {
globalThis.fetch = (async () =>
new Response(JSON.stringify({ data: { quota: rawQuota } }), {
status: 200,
headers: { "content-type": "application/json" },
})) as typeof fetch;
}
/**
* #10078 follow-up — the original fix wired AgentRouter's balance into
* getUsageForProvider()/USAGE_SUPPORTED_PROVIDERS (visibility + data path), but the
* Dashboard Quota UI *renderer* (QuotaCardBody / QuotaCardExpanded under
* src/app/(dashboard)/dashboard/usage/components/ProviderLimits/) only formats a
* quota row as a dollar amount ("$X.XX") when the row carries `isCredits: true` +
* `currency` + `creditCount` — fields the generic quota-parsing path
* (parseGeneric -> normalizeQuotaEntry in quotaParsing.ts) never sets, and never
* copies `currency` through at all. Because "agentrouter" wasn't special-cased in
* parseProviderQuotas(), a configured balance rendered as a bare "100% left"
* percentage (not USD), and the real dollar figure (`dollarBalance`) was only
* exposed as a top-level `remainingUsd`/`availableUsd`/`balance` sibling field that
* parseQuotaData() never reads (it only walks `data.quotas`).
*
* This test drives the real producer (getAgentrouterUsage) through the real
* Dashboard adapter (parseQuotaData) end-to-end — the same path the Provider
* Limits UI takes — and asserts the row the renderer actually consumes
* (`isCredits`, `currency`, `creditCount`) instead of just the wire-shape fields
* asserted by tests/unit/agentrouter-quota-visibility.test.ts.
*/
test("#10078: a configured AgentRouter balance renders as a USD credits row in the Dashboard Quota UI", async () => {
const connectionId = `agentrouter-dash-configured-${Date.now()}`;
mockAgentrouterFetch(250_000); // 250_000 / 500_000 QUOTA_PER_UNIT = $0.50
const usage = await getAgentrouterUsage(connectionId, {
provider: "agentrouter",
providerSpecificData: { consoleApiKey: "system-access-token", newApiUserId: "42" },
});
const rows = parseQuotaData("agentrouter", usage) as Array<{
isCredits?: boolean;
currency?: string;
creditCount?: number;
remainingPercentage?: number;
}>;
assert.equal(rows.length, 1, `expected exactly one quota row, got: ${JSON.stringify(rows)}`);
const [row] = rows;
// These are exactly the fields QuotaCardBody.tsx / QuotaCardExpanded.tsx branch on
// to render a dollar-formatted amount ("$0.50") instead of a bare percentage.
assert.equal(row.isCredits, true, "renderer only formats USD when isCredits is true");
assert.equal(row.currency, "USD", "renderer looks up CURRENCY_SYMBOLS[q.currency]");
assert.equal(row.creditCount, 0.5, "renderer displays q.creditCount as the dollar amount");
assert.equal(row.remainingPercentage, 100, "a funded wallet must not read as exhausted");
});
test("#10078: an exhausted AgentRouter balance renders as exactly $0, not negative or NaN", async () => {
const connectionId = `agentrouter-dash-exhausted-${Date.now()}`;
mockAgentrouterFetch(0);
const usage = await getAgentrouterUsage(connectionId, {
provider: "agentrouter",
providerSpecificData: { consoleApiKey: "system-access-token", newApiUserId: "42" },
});
const rows = parseQuotaData("agentrouter", usage) as Array<{
isCredits?: boolean;
currency?: string;
creditCount?: number;
remainingPercentage?: number;
}>;
assert.equal(rows.length, 1);
const [row] = rows;
assert.equal(row.isCredits, true);
assert.equal(row.currency, "USD");
assert.equal(row.creditCount, 0, "exhausted balance must render as exactly zero");
assert.equal(Number.isFinite(row.creditCount), true, "must never render NaN");
assert.ok((row.creditCount ?? -1) >= 0, "must never render negative");
assert.equal(row.remainingPercentage, 0, "exhausted wallet must read as 0% remaining (critical color)");
});
test("#10078: parseQuotaData never drops a raw negative/garbage remaining as -$X — clamps to 0", () => {
// Defends the Math.max(0, ...) clamp in both getAgentrouterUsage() and
// parseAgentrouterQuota() against a malformed/negative upstream `remaining`.
const data = {
plan: "AgentRouter",
quotas: {
balance: {
used: 0,
total: 0,
remaining: -5,
remainingPercentage: 0,
resetAt: null,
unlimited: true,
currency: "USD",
displayName: "Wallet Balance (USD)",
},
},
};
const rows = parseQuotaData("agentrouter", data) as Array<{ creditCount?: number }>;
assert.equal(rows.length, 1);
assert.equal(rows[0].creditCount, 0);
});

View File

@@ -0,0 +1,86 @@
import test from "node:test";
import assert from "node:assert/strict";
import { USAGE_SUPPORTED_PROVIDERS } from "../../src/shared/constants/providers.ts";
import { supportsProviderQuota } from "../../src/shared/utils/providerQuotaVisibility.ts";
import {
USAGE_FETCHER_PROVIDERS,
getUsageForProvider,
} from "../../open-sse/services/usage.ts";
import {
getAgentrouterUsage,
} from "../../open-sse/services/usage/agentrouter.ts";
import {
invalidateAgentrouterQuotaCache,
type AgentrouterQuota,
} from "../../open-sse/services/agentrouterQuotaFetcher.ts";
const originalFetch = globalThis.fetch;
test.afterEach(() => {
globalThis.fetch = originalFetch;
});
/**
* #10078 — AgentRouter quota was missing from the dashboard:
* - USAGE_SUPPORTED_PROVIDERS (visibility gate) omitted "agentrouter", and
* - USAGE_FETCHER_PROVIDERS + getUsageForProvider (the provider-limits data
* path) had no "agentrouter" case, so /api/usage/provider-limits fell back
* to the generic "Usage API not implemented" message.
* These three assertions are the permanent regression guard (RED before the
* fix, GREEN after).
*/
test("#10078: agentrouter is present in USAGE_SUPPORTED_PROVIDERS", () => {
assert.equal(
USAGE_SUPPORTED_PROVIDERS.includes("agentrouter" as (typeof USAGE_SUPPORTED_PROVIDERS)[number]),
true
);
});
test("#10078: supportsProviderQuota('agentrouter') is true", () => {
assert.equal(supportsProviderQuota("agentrouter"), true);
});
test("#10078: agentrouter is present in USAGE_FETCHER_PROVIDERS", () => {
assert.equal(
USAGE_FETCHER_PROVIDERS.includes("agentrouter" as (typeof USAGE_FETCHER_PROVIDERS)[number]),
true
);
});
test("#10078: getUsageForProvider shapes the AgentRouter balance into a USD quota", async () => {
const connectionId = `agentrouter-vis-${Date.now()}`;
globalThis.fetch = (async () => {
return new Response(JSON.stringify({ data: { quota: 250_000 } }), {
status: 200,
headers: { "content-type": "application/json" },
});
}) as typeof fetch;
const usage = (await getUsageForProvider({
id: connectionId,
provider: "agentrouter",
providerSpecificData: { consoleApiKey: "system-access-token", newApiUserId: "42" },
})) as {
plan?: string;
quotas?: Record<string, { displayName?: string; remainingPercentage?: number }>;
remainingUsd?: number;
};
assert.equal(usage.plan, "AgentRouter");
assert.ok(usage.quotas);
const balance = usage.quotas.balance;
assert.ok(balance, "expected a `balance` quota entry");
assert.equal(balance.displayName, "Wallet Balance (USD)");
assert.equal(usage.remainingUsd, 0.5);
});
test("#10078: getAgentrouterUsage returns a graceful message when console credentials are missing", async () => {
const usage = (await getAgentrouterUsage(`missing-${Date.now()}`, {
provider: "agentrouter",
})) as { message?: string; quotas?: unknown };
assert.equal(typeof usage.message, "string");
assert.ok(/not available/i.test(usage.message || ""));
assert.equal(usage.quotas, undefined);
});

View File

@@ -1,54 +0,0 @@
import test from "node:test";
import assert from "node:assert/strict";
import {
classifyComboOutcome,
formatComboOutcomes,
redactConnectionLabel,
buildRedactedSummary,
} from "../../open-sse/services/combo/comboErrorAggregation.ts";
// #10314 — combo error aggregation mixes quality and auth.
// Regression guard for the pure aggregation helpers: a quality-failure reason from one
// target and a sibling's 401 must be presented as SEPARATE classified outcomes (never
// mashed into a single lastError), and account/connection identifiers must be redacted
// from client-visible and shared-warn strings.
test("#10314: classifyComboOutcome keeps auth distinct from quality/model", () => {
assert.equal(classifyComboOutcome(401, "invalid_api_key"), "auth");
assert.equal(classifyComboOutcome(403, "not authorized"), "auth");
// 5xx sleep to the "timeout" class (>=499 is checked before >=500).
assert.equal(classifyComboOutcome(503, "upstream unavailable"), "timeout");
assert.equal(classifyComboOutcome(408, "timeout"), "timeout");
assert.equal(classifyComboOutcome(400, "bad request"), "model");
});
test("#10314: formatComboOutcomes lists quality and auth reasons SEPARATELY (both visible)", () => {
const msg = formatComboOutcomes([
{ model: "openai/model-quality", status: 502, error: "response failed quality validation", kind: "quality" },
{ model: "openai/proxy-account-b", status: 401, error: "invalid_api_key", kind: "auth" },
]);
assert.match(msg, /quality validation/);
assert.match(msg, /invalid_api_key/);
assert.match(msg, /auth/);
assert.ok(msg.indexOf("quality validation") < msg.indexOf("invalid_api_key"));
});
test("#10314: redactConnectionLabel masks connection/account identifiers", () => {
assert.equal(
redactConnectionLabel("openai/proxy-account-b"),
"openai/proxy-account-b"
);
const withUuid = redactConnectionLabel("openai/8a4f0c6e-3b27-4c51-9d88-1f2a3b4c5d6e");
assert.equal(withUuid, "openai/conn:8a4f0c6e");
const withHex = redactConnectionLabel("openai/0f1e2d3c4b5a69788796170a1b2c3d4e5f607182");
assert.equal(withHex, "openai/conn:0f1e2d3c");
});
test("#10314: buildRedactedSummary is redacted and truncates past 5 entries", () => {
const s = buildRedactedSummary(
Array.from({ length: 6 }, (_, i) => ({ model: `openai/8a4f0c6e-3b27-4c51-9d88-1f2a3b4c5d6e-${i}`, status: 401 + i }))
);
assert.ok(!s.includes("8a4f0c6e-3b27"), "summary must not leak a full UUID");
assert.match(s, /conn:8a4f0c6e/);
assert.match(s, /\(\+1\)/);
});