73cb89143c
Adds the Desktop-exclusive Preview models and the Xiaomi account-session
route to the existing xiaomi-mimo provider instead of a separate
xiaomi-desktop provider, so the dashboard shows one MiMo entry rather than
three overlapping ones.
Dual auth, same pattern as kimi — API key (sk-) covers the cloud API,
Desktop/OAuth adds the account session used by the Preview models:
- registry: category oauth, authModes [oauth, apikey], oauth block, the two
mimo-x-*-preview models, and the invite signupUrl
- executor: routes Preview models to the account-service route with a Cookie
session, everything else keeps the sourceFormat-matched transport
- oauth: custom ECDH encrypted-callback flow (X25519 -> SHA256 -> AES-256-GCM)
with a loopback callback proxy, plus one-click import of the local Desktop
auth.json
- usage: weekly quota from the account session
Fixes found while merging:
- the OAuth browser flow was dead: poll-status cleared the session before the
client could POST /exchange, so every exchange returned 400
- a Claude-format client was sent to /v1/chat/completions instead of the
declared /anthropic/v1/messages transport, because buildUrl ignored
runtimeTransport
- stopXiaomiMimoProxy leaked every pending session (each holding an X25519
private key) for the process lifetime
- the OAuth exchange did not persist the Desktop passToken, so the Preview
models could never work after a browser sign-in
Removes dead code: the local engine token minting (mimoEngine, never called
on the request path), the model-catalog and usage routes, engineToken/
engineUrl plumbing, and an unread top-level usage block.
Adds tests/unit/xiaomi-mimo-{executor,oauth-session,oauth-proxy}.test.js —
the provider previously had none.
86 lines
4.1 KiB
JavaScript
86 lines
4.1 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, getCodexRateLimitResetCredits } from "./usage/codex.js";
|
|
|
|
export { consumeCodexRateLimitResetCredit, getCodexRateLimitResetCredits };
|
|
import { getKiroUsage } from "./usage/kiro.js";
|
|
import { getMiniMaxUsage } from "./usage/minimax.js";
|
|
import { getCodeBuddyCnUsage, getCodeBuddyIntlUsage } from "./usage/codebuddy-cn.js";
|
|
import { getGrokCliUsage } from "./usage/grok-cli.js";
|
|
import { getKimiUsage } from "./usage/kimi.js";
|
|
import { getDeepseekUsage } from "./usage/deepseek.js";
|
|
import { getOpenCodeGoUsage } from "./usage/opencode-go.js";
|
|
import { getGroqUsage } from "./usage/groq.js";
|
|
import { getZedUsage } from "./usage/zed.js";
|
|
import { getXiaomiMimoUsage } from "./usage/xiaomi-mimo.js";
|
|
import { resolveQoderCredentials } from "./qoderModels.js";
|
|
import { getGlmUsage } from "./usage/glm.js";
|
|
import {
|
|
getIflowUsage,
|
|
getOllamaUsage,
|
|
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, { force: c.force }),
|
|
codex: (c) => getCodexUsage(c.accessToken, c.proxyOptions),
|
|
kiro: (c) => getKiroUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
qoder: async (c) => {
|
|
// PAT (pt-...) connections must be exchanged to a job token before the
|
|
// quota endpoint accepts them.
|
|
const resolved = await resolveQoderCredentials(c, c.proxyOptions).catch(() => null);
|
|
return getQoderUsage(resolved?.accessToken || c.accessToken, c.proxyOptions);
|
|
},
|
|
iflow: (c) => getIflowUsage(c.accessToken),
|
|
ollama: (c) => getOllamaUsage(c.apiKey, c.providerSpecificData, c.proxyOptions),
|
|
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),
|
|
"codebuddy-intl": (c) => getCodeBuddyIntlUsage(c.accessToken, c.apiKey, c.providerSpecificData, c.proxyOptions),
|
|
"grok-cli": (c) => getGrokCliUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
kimi: (c) => getKimiUsage(c.accessToken, c.apiKey, c.proxyOptions, c.providerSpecificData),
|
|
"opencode-go": (c) => getOpenCodeGoUsage(c.apiKey, c.proxyOptions),
|
|
deepseek: (c) => getDeepseekUsage(c.apiKey, c.proxyOptions),
|
|
groq: (c) => getGroqUsage(c.apiKey, c.proxyOptions),
|
|
zed: (c) => getZedUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
"xiaomi-mimo": (c) => getXiaomiMimoUsage(c.accessToken, c.providerSpecificData, c.proxyOptions),
|
|
};
|
|
|
|
export async function getUsageForProvider(connection, proxyOptions = null, options = {}) {
|
|
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,
|
|
force: options.force === true,
|
|
});
|
|
}
|