8321032e36
Two additions on top of the merged CodeBuddy CN OAuth provider:
1. API key auth — connect with a direct API key (sent as
Authorization: Bearer), not only via OAuth device-code flow.
- registry: add authModes ["oauth","apikey"] + hasOAuth; combined Bearer
auth already forwards the key, token-refresh skips key connections.
- providers POST: accept dual-auth providers (authModes includes
"apikey") that live under category "oauth" — previously rejected as
"Invalid provider". Also fixes the same latent gap for xai.
2. Quota tracker — surface CodeBuddy CN credit balance on the usage
dashboard for both OAuth and API-key connections.
- registry: add transport.usage.url (Tencent billing endpoint) +
features.usage/usageApikey so the connection is quota-eligible.
- new CN-scoped handler services/usage/codebuddy-cn.js: POST the billing
meter endpoint, unwrap data.Response.Data.Accounts[]. The payload mixes
two credit types that must not be merged:
* refill/base ("基础体验包") — recurring allowance; cycle resets well
before the resource expires (CycleEndTime << DeductionEndTime).
Reads the *Cycle* balance, resetAt = next refresh. Cadence-labelled.
* bonus ("活动赠送包") — one-shot credits that expire at CycleEndTime.
Reads the plain Capacity balance. Labelled "Bonus Pack N".
One quota row per package, soonest-expiring first.
- register handler under "codebuddy-cn" in USAGE_HANDLERS.
Frontend needs no change — USAGE_SUPPORTED_PROVIDERS/USAGE_APIKEY_PROVIDERS
and the generic parseQuotaData branch already cover this shape.
Co-authored-by: Cursor <cursoragent@cursor.com>
59 lines
2.6 KiB
JavaScript
59 lines
2.6 KiB
JavaScript
/**
|
|
* Usage Fetcher - Get usage data from provider APIs
|
|
*/
|
|
|
|
import { getGitHubUsage } from "./usage/github.js";
|
|
import { getGeminiUsage, getAntigravityUsage } from "./usage/google.js";
|
|
import { getClaudeUsage } from "./usage/claude.js";
|
|
import { getCodexUsage, consumeCodexRateLimitResetCredit } from "./usage/codex.js";
|
|
|
|
export { consumeCodexRateLimitResetCredit };
|
|
import { getKiroUsage } from "./usage/kiro.js";
|
|
import { getMiniMaxUsage } from "./usage/minimax.js";
|
|
import { getCodeBuddyCnUsage } from "./usage/codebuddy-cn.js";
|
|
import {
|
|
getQwenUsage,
|
|
getIflowUsage,
|
|
getOllamaUsage,
|
|
getGlmUsage,
|
|
getVercelAiGatewayUsage,
|
|
getQoderUsage,
|
|
} from "./usage/misc.js";
|
|
|
|
/**
|
|
* Get usage data for a provider connection
|
|
* @param {Object} connection - Provider connection with accessToken
|
|
* @returns {Object} Usage data with quotas
|
|
*/
|
|
// provider → usage handler (ctx carries every arg each handler needs)
|
|
const USAGE_HANDLERS = {
|
|
github: (c) => getGitHubUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
"gemini-cli": (c) => getGeminiUsage(c.accessToken, c.providerDataWithProjectId, c.proxyOptions),
|
|
antigravity: (c) => getAntigravityUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
claude: (c) => getClaudeUsage(c.accessToken, c.proxyOptions),
|
|
codex: (c) => getCodexUsage(c.accessToken, c.proxyOptions),
|
|
kiro: (c) => getKiroUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
qoder: (c) => getQoderUsage(c.accessToken, c.proxyOptions),
|
|
qwen: (c) => getQwenUsage(c.accessToken, c.providerSpecificData),
|
|
iflow: (c) => getIflowUsage(c.accessToken),
|
|
ollama: (c) => getOllamaUsage(c.accessToken),
|
|
glm: (c) => getGlmUsage(c.apiKey, c.provider, c.proxyOptions),
|
|
"glm-cn": (c) => getGlmUsage(c.apiKey, c.provider, c.proxyOptions),
|
|
minimax: (c) => getMiniMaxUsage(c.apiKey, c.provider, c.proxyOptions),
|
|
"minimax-cn": (c) => getMiniMaxUsage(c.apiKey, c.provider, c.proxyOptions),
|
|
"vercel-ai-gateway": (c) => getVercelAiGatewayUsage(c.apiKey, c.proxyOptions),
|
|
"codebuddy-cn": (c) => getCodeBuddyCnUsage(c.accessToken, c.apiKey, c.providerSpecificData, c.proxyOptions),
|
|
};
|
|
|
|
export async function getUsageForProvider(connection, proxyOptions = null) {
|
|
const { provider, accessToken, apiKey, providerSpecificData, projectId } = connection;
|
|
const providerDataWithProjectId = {
|
|
...(providerSpecificData || {}),
|
|
...(projectId ? { projectId } : {}),
|
|
};
|
|
|
|
const handler = USAGE_HANDLERS[provider];
|
|
if (!handler) return { message: `Usage API not implemented for ${provider}` };
|
|
return await handler({ provider, accessToken, apiKey, providerSpecificData, providerDataWithProjectId, proxyOptions });
|
|
}
|