feat(codebuddy-cn): add API key auth + credit quota tracker
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>
This commit is contained in:
@@ -16,6 +16,8 @@ export default {
|
||||
},
|
||||
},
|
||||
category: "oauth",
|
||||
authModes: ["oauth", "apikey"],
|
||||
hasOAuth: true,
|
||||
transport: {
|
||||
baseUrl: "https://copilot.tencent.com/v2/chat/completions",
|
||||
forceStream: true,
|
||||
@@ -36,6 +38,11 @@ export default {
|
||||
header: "Authorization",
|
||||
scheme: "bearer",
|
||||
},
|
||||
// Quota endpoint differs from the chat gateway: POST returns nested Tencent
|
||||
// billing payload (data.Response.Data.Accounts[]). See services/usage/codebuddy-cn.js.
|
||||
usage: {
|
||||
url: "https://copilot.tencent.com/v2/billing/meter/get-user-resource",
|
||||
},
|
||||
},
|
||||
models: [
|
||||
{ id: "glm-5.2", name: "GLM-5.2" },
|
||||
@@ -63,4 +70,8 @@ export default {
|
||||
platform: "CLI",
|
||||
pollInterval: 5000,
|
||||
},
|
||||
features: {
|
||||
usage: true,
|
||||
usageApikey: true,
|
||||
},
|
||||
};
|
||||
|
||||
@@ -10,6 +10,7 @@ import { getCodexUsage, consumeCodexRateLimitResetCredit } from "./usage/codex.j
|
||||
export { consumeCodexRateLimitResetCredit };
|
||||
import { getKiroUsage } from "./usage/kiro.js";
|
||||
import { getMiniMaxUsage } from "./usage/minimax.js";
|
||||
import { getCodeBuddyCnUsage } from "./usage/codebuddy-cn.js";
|
||||
import {
|
||||
getQwenUsage,
|
||||
getIflowUsage,
|
||||
@@ -41,6 +42,7 @@ const USAGE_HANDLERS = {
|
||||
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) {
|
||||
|
||||
131
open-sse/services/usage/codebuddy-cn.js
Normal file
131
open-sse/services/usage/codebuddy-cn.js
Normal file
@@ -0,0 +1,131 @@
|
||||
/**
|
||||
* CodeBuddy CN usage handler
|
||||
*
|
||||
* Scoped to the "codebuddy-cn" provider specifically — a future "codebuddy-intl"
|
||||
* variant would get its own handler/endpoint, so keep this CN-only.
|
||||
*
|
||||
* Quota lives behind a Tencent billing endpoint (POST, payload wrapped twice
|
||||
* under data.Response.Data). It mixes two credit types that must NOT be merged:
|
||||
*
|
||||
* - Refill / base ("基础体验包"): a recurring allowance whose cycle resets long
|
||||
* before the resource itself expires (CycleEndTime << DeductionEndTime). The
|
||||
* live numbers live in the *Cycle* fields (e.g. CycleCapacityUsed 6.54 / 500)
|
||||
* and resetAt is the next monthly refresh.
|
||||
* - Bonus ("活动赠送包"): one-shot credits that run a single cycle and then
|
||||
* expire for good (CycleEndTime == DeductionEndTime). Numbers live in the
|
||||
* plain Capacity fields.
|
||||
*
|
||||
* We surface one quota row per package — a cadence label (Monthly/Weekly/Daily)
|
||||
* for refill packs, "Bonus Pack N" for bonus packs (soonest-expiring first).
|
||||
*/
|
||||
|
||||
import { proxyAwareFetch } from "../../utils/proxyFetch.js";
|
||||
import { PROVIDERS } from "../../providers/index.js";
|
||||
import { U, parseResetTime } from "./shared.js";
|
||||
|
||||
const PROVIDER_ID = "codebuddy-cn";
|
||||
|
||||
// Prefer the *Precise string fields (exact), fall back to the numeric ones.
|
||||
function num(precise, plain) {
|
||||
const n = Number(precise ?? plain);
|
||||
return Number.isFinite(n) ? n : 0;
|
||||
}
|
||||
|
||||
// Label a refill pack by its cycle length (Monthly is the common CodeBuddy case).
|
||||
function refillCadence(acc) {
|
||||
const start = parseResetTime(acc.CycleStartTime);
|
||||
const end = parseResetTime(acc.CycleEndTime);
|
||||
if (start && end) {
|
||||
const days = (new Date(end).getTime() - new Date(start).getTime()) / 86400000;
|
||||
if (days <= 1.5) return "Daily";
|
||||
if (days <= 10) return "Weekly";
|
||||
}
|
||||
return "Monthly";
|
||||
}
|
||||
|
||||
export async function getCodeBuddyCnUsage(accessToken, apiKey, providerSpecificData, proxyOptions = null) {
|
||||
const token = accessToken || apiKey;
|
||||
if (!token) {
|
||||
return { message: "CodeBuddy CN credential not available." };
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await proxyAwareFetch(U(PROVIDER_ID).url, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
...(PROVIDERS[PROVIDER_ID]?.headers || {}),
|
||||
Authorization: `Bearer ${token}`,
|
||||
"Content-Type": "application/json",
|
||||
Accept: "application/json",
|
||||
},
|
||||
body: "{}",
|
||||
}, proxyOptions);
|
||||
|
||||
if (response.status === 401 || response.status === 403) {
|
||||
return { message: "CodeBuddy CN credential invalid or expired." };
|
||||
}
|
||||
if (!response.ok) {
|
||||
return { message: `CodeBuddy CN quota API error (${response.status}).` };
|
||||
}
|
||||
|
||||
const json = await response.json();
|
||||
if (json?.code !== 0) {
|
||||
return { message: `CodeBuddy CN quota error: ${json?.msg || "unknown"}` };
|
||||
}
|
||||
|
||||
const data = json?.data?.Response?.Data || {};
|
||||
const accounts = Array.isArray(data.Accounts) ? data.Accounts : [];
|
||||
if (accounts.length === 0) {
|
||||
return { message: "CodeBuddy CN connected. No credit package found." };
|
||||
}
|
||||
|
||||
const cycleEndMs = (acc) => {
|
||||
const r = parseResetTime(acc.CycleEndTime);
|
||||
return r ? new Date(r).getTime() : Number.POSITIVE_INFINITY;
|
||||
};
|
||||
// Refill packs roll into a new cycle before the resource expires; bonus packs
|
||||
// end exactly at expiry. >2d gap between cycle end and validity end = refill.
|
||||
const REFILL_GAP_MS = 2 * 24 * 60 * 60 * 1000;
|
||||
const isRefill = (acc) => {
|
||||
const ce = cycleEndMs(acc);
|
||||
const de = Number(acc.DeductionEndTime);
|
||||
return Number.isFinite(ce) && Number.isFinite(de) && de - ce > REFILL_GAP_MS;
|
||||
};
|
||||
const byExpiry = (a, b) => cycleEndMs(a) - cycleEndMs(b);
|
||||
|
||||
const refills = accounts.filter(isRefill).sort(byExpiry);
|
||||
const bonuses = accounts.filter((a) => !isRefill(a)).sort(byExpiry);
|
||||
|
||||
const quotas = {};
|
||||
// Refill packs first: cadence-labelled, using the *Cycle* balance and
|
||||
// resetting at the next refresh.
|
||||
const seenRefill = {};
|
||||
refills.forEach((acc) => {
|
||||
const base = refillCadence(acc);
|
||||
seenRefill[base] = (seenRefill[base] || 0) + 1;
|
||||
const name = seenRefill[base] > 1 ? `${base} ${seenRefill[base]}` : base;
|
||||
quotas[name] = {
|
||||
used: num(acc.CycleCapacityUsedPrecise, acc.CycleCapacityUsed),
|
||||
total: num(acc.CycleCapacitySizePrecise, acc.CycleCapacitySize),
|
||||
resetAt: parseResetTime(acc.CycleEndTime),
|
||||
unlimited: false,
|
||||
};
|
||||
});
|
||||
// Bonus packs: use the lifetime Capacity balance; resetAt is the expiry.
|
||||
bonuses.forEach((acc, i) => {
|
||||
quotas[`Bonus Pack ${i + 1}`] = {
|
||||
used: num(acc.CapacityUsedPrecise, acc.CapacityUsed),
|
||||
total: num(acc.CapacitySizePrecise, acc.CapacitySize),
|
||||
resetAt: parseResetTime(acc.CycleEndTime),
|
||||
unlimited: false,
|
||||
};
|
||||
});
|
||||
|
||||
const basePkg = refills[0] || accounts[0] || {};
|
||||
const plan = basePkg.PackageName || basePkg.SubProductName || "CodeBuddy CN";
|
||||
|
||||
return { plan, quotas };
|
||||
} catch (error) {
|
||||
return { message: `CodeBuddy CN error: ${error.message}` };
|
||||
}
|
||||
}
|
||||
@@ -102,8 +102,12 @@ export async function POST(request) {
|
||||
|
||||
// Validation
|
||||
const isWebCookieProvider = !!WEB_COOKIE_PROVIDERS[provider];
|
||||
// Dual-auth providers (e.g. codebuddy-cn, xai) live under category "oauth" but also
|
||||
// accept an API key via authModes — they aren't in APIKEY_PROVIDERS, so allow them here.
|
||||
const supportsApiKeyMode = !!AI_PROVIDERS[provider]?.authModes?.includes("apikey");
|
||||
const isValidProvider = APIKEY_PROVIDERS[provider] ||
|
||||
FREE_TIER_PROVIDERS[provider] ||
|
||||
supportsApiKeyMode ||
|
||||
isWebCookieProvider ||
|
||||
isOpenAICompatibleProvider(provider) ||
|
||||
isAnthropicCompatibleProvider(provider) ||
|
||||
|
||||
Reference in New Issue
Block a user