fcfcced4ab
GLM quota parsing only accepted TOKENS_LIMIT and wrote every limit to a single "session" key, so credit-based plans showed nothing and later intervals overwrote earlier ones. Accept CREDIT_LIMIT too and derive the quota key from the limit unit (5h session, 7d weekly, tokens, custom). Moves the parser into its own usage/glm.js, re-exported from misc.js.
89 lines
2.6 KiB
JavaScript
89 lines
2.6 KiB
JavaScript
/**
|
|
* GLM Coding Plan usage (international + China regions)
|
|
*/
|
|
|
|
import { proxyAwareFetch } from "../../utils/proxyFetch.js";
|
|
import { U } from "./shared.js";
|
|
|
|
// GLM quota endpoints (region-aware) — url from registry transport.usage
|
|
const GLM_QUOTA_URLS = {
|
|
international: U("glm").url,
|
|
china: U("glm-cn").url,
|
|
};
|
|
|
|
/**
|
|
* GLM Coding Plan usage (international + China regions)
|
|
* Supports both TOKENS_LIMIT and CREDIT_LIMIT and dynamic intervals (e.g. session 5h, weekly 7d).
|
|
*/
|
|
export async function getGlmUsage(apiKey, provider, proxyOptions = null) {
|
|
if (!apiKey) {
|
|
return { message: "GLM API key not available." };
|
|
}
|
|
|
|
const region = provider === "glm-cn" ? "china" : "international";
|
|
const quotaUrl = GLM_QUOTA_URLS[region];
|
|
|
|
try {
|
|
const response = await proxyAwareFetch(
|
|
quotaUrl,
|
|
{
|
|
headers: {
|
|
Authorization: `Bearer ${apiKey}`,
|
|
Accept: "application/json",
|
|
},
|
|
},
|
|
proxyOptions,
|
|
);
|
|
|
|
if (!response.ok) {
|
|
if (response.status === 401) {
|
|
return { message: "GLM API key invalid or expired." };
|
|
}
|
|
return { message: `GLM quota API error (${response.status}).` };
|
|
}
|
|
|
|
const json = await response.json();
|
|
const data = json?.data && typeof json.data === "object" ? json.data : {};
|
|
const limits = Array.isArray(data.limits) ? data.limits : [];
|
|
const quotas = {};
|
|
|
|
for (const limit of limits) {
|
|
// 1. Accept both TOKENS_LIMIT and CREDIT_LIMIT from GLM API
|
|
if (!limit || (limit.type !== "TOKENS_LIMIT" && limit.type !== "CREDIT_LIMIT")) continue;
|
|
const usedPercent = Number(limit.percentage) || 0;
|
|
const resetMs = Number(limit.nextResetTime) || 0;
|
|
const remaining = Math.max(0, 100 - usedPercent);
|
|
|
|
// 2. Map key dynamically based on type and period (unit) to avoid overwriting
|
|
let key = "session";
|
|
if (limit.unit === 3) {
|
|
key = `Session (${limit.number}h)`;
|
|
} else if (limit.unit === 6) {
|
|
key = "Weekly (7d)";
|
|
} else if (limit.type === "TOKENS_LIMIT") {
|
|
key = "Tokens";
|
|
} else {
|
|
key = `Limit (${limit.number})`;
|
|
}
|
|
|
|
quotas[key] = {
|
|
used: usedPercent,
|
|
total: 100,
|
|
remaining,
|
|
remainingPercentage: remaining,
|
|
resetAt: resetMs > 0 ? new Date(resetMs).toISOString() : null,
|
|
unlimited: false,
|
|
};
|
|
}
|
|
|
|
const levelRaw = typeof data.level === "string" ? data.level : "";
|
|
const plan = levelRaw
|
|
? levelRaw.charAt(0).toUpperCase() + levelRaw.slice(1).toLowerCase()
|
|
: "Unknown";
|
|
|
|
return { plan, quotas };
|
|
} catch (error) {
|
|
return { message: `GLM error: ${error.message}` };
|
|
}
|
|
}
|