feat(quota): add Zhipu AI Coding Plan usage tracking (#756)

- Add zhipuai-coding-plan provider for quota monitoring
- Support TOKENS_LIMIT (5-hour window) and TIME_LIMIT (MCP tools monthly)
- Update UI provider list and TypeScript types
- Register provider in server quota registry
This commit is contained in:
Xiangfang Chen
2026-04-11 22:15:19 +03:00
committed by GitHub
parent bdcff3a5b7
commit a3d73a8b67
6 changed files with 268 additions and 0 deletions
+124
View File
@@ -71,6 +71,36 @@ type ZaiPayload = {
};
};
type ZhipuaiTokensLimit = {
type: 'TOKENS_LIMIT';
unit?: number;
number?: number;
nextResetTime?: number;
percentage?: number;
};
type ZhipuaiMcpTimeLimit = {
type: 'TIME_LIMIT';
unit?: number;
number?: number;
usage?: number;
currentValue?: number;
remaining?: number;
percentage?: number;
nextResetTime?: number;
usageDetails?: Array<{
modelCode?: string;
usage?: number;
}>;
};
type ZhipuaiPayload = {
data?: {
limits?: Array<ZhipuaiTokensLimit | ZhipuaiMcpTimeLimit>;
level?: string;
};
};
export type ProviderResult = {
providerId: string;
providerName: string;
@@ -355,6 +385,11 @@ export const listConfiguredQuotaProviders = () => {
configured.add('zai-coding-plan');
}
const zhipuaiAuth = normalizeAuthEntry(getAuthEntry(auth, ['zhipuai-coding-plan']));
if (zhipuaiAuth && ((zhipuaiAuth as Record<string, unknown>).key || (zhipuaiAuth as Record<string, unknown>).token)) {
configured.add('zhipuai-coding-plan');
}
const kimiAuth = normalizeAuthEntry(getAuthEntry(auth, ['kimi-for-coding', 'kimi']));
if (kimiAuth && ((kimiAuth as Record<string, unknown>).key || (kimiAuth as Record<string, unknown>).token)) {
configured.add('kimi-for-coding');
@@ -1259,6 +1294,93 @@ export const fetchZaiQuota = async (): Promise<ProviderResult> => {
}
};
export const fetchZhipuaiCodingPlanQuota = async (): Promise<ProviderResult> => {
const auth = readAuthFile();
const entry = normalizeAuthEntry(getAuthEntry(auth, ['zhipuai-coding-plan'])) as Record<string, unknown> | null;
const apiKey = (entry?.key as string | undefined) ?? (entry?.token as string | undefined);
if (!apiKey) {
return buildResult({
providerId: 'zhipuai-coding-plan',
providerName: 'Zhipu AI Coding Plan',
ok: false,
configured: false,
error: 'Not configured',
});
}
try {
const response = await fetch('https://open.bigmodel.cn/api/monitor/usage/quota/limit', {
method: 'GET',
headers: {
Authorization: `Bearer ${apiKey}`,
'Content-Type': 'application/json',
},
});
if (!response.ok) {
return buildResult({
providerId: 'zhipuai-coding-plan',
providerName: 'Zhipu AI Coding Plan',
ok: false,
configured: true,
error: `API error: ${response.status}`,
});
}
const payload = await response.json() as ZhipuaiPayload;
const limits = Array.isArray(payload?.data?.limits) ? payload.data.limits : [];
const tokensLimit = limits.find((limit): limit is ZhipuaiTokensLimit => limit?.type === 'TOKENS_LIMIT');
const mcpToolsTimeLimit = limits.find((limit): limit is ZhipuaiMcpTimeLimit => limit?.type === 'TIME_LIMIT');
const windows: Record<string, UsageWindow> = {};
// Handle TOKENS_LIMIT (5-hour window for token usage)
if (tokensLimit) {
const windowSeconds = resolveWindowSeconds(tokensLimit);
const resetAt = tokensLimit?.nextResetTime ? normalizeTimestamp(tokensLimit.nextResetTime) : null;
const usedPercent = typeof tokensLimit?.percentage === 'number' ? tokensLimit.percentage : null;
windows['Tokens'] = toUsageWindow({
usedPercent,
windowSeconds,
resetAt,
});
}
// Handle TIME_LIMIT (MCP tools monthly window)
if (mcpToolsTimeLimit) {
// TIME_LIMIT unit=5 means 1 month (30 days)
const monthSeconds = 30 * 24 * 60 * 60;
const resetAt = mcpToolsTimeLimit?.nextResetTime ? normalizeTimestamp(mcpToolsTimeLimit.nextResetTime) : null;
const usedPercent = typeof mcpToolsTimeLimit?.percentage === 'number' ? mcpToolsTimeLimit.percentage : null;
windows['MCP Tools'] = toUsageWindow({
usedPercent,
windowSeconds: monthSeconds,
resetAt,
});
}
return buildResult({
providerId: 'zhipuai-coding-plan',
providerName: 'Zhipu AI Coding Plan',
ok: true,
configured: true,
usage: { windows },
});
} catch (error) {
return buildResult({
providerId: 'zhipuai-coding-plan',
providerName: 'Zhipu AI Coding Plan',
ok: false,
configured: true,
error: error instanceof Error ? error.message : 'Request failed',
});
}
};
const NANO_GPT_DAILY_WINDOW_SECONDS = 86400;
export const fetchNanoGptQuota = async (): Promise<ProviderResult> => {
@@ -1384,6 +1506,8 @@ export const fetchQuotaForProvider = async (providerId: string): Promise<Provide
return fetchOpenRouterQuota();
case 'zai-coding-plan':
return fetchZaiQuota();
case 'zhipuai-coding-plan':
return fetchZhipuaiCodingPlanQuota();
default:
return buildResult({
providerId,