Initial commit: restructure to flat layout with ui/ and web/ at root
This commit is contained in:
152
web/server/lib/quota/providers/index.js
Normal file
152
web/server/lib/quota/providers/index.js
Normal file
@@ -0,0 +1,152 @@
|
||||
/**
|
||||
* Quota Providers Registry
|
||||
*
|
||||
* Implements quota fetching for various AI providers using a registry pattern.
|
||||
* @module quota/providers
|
||||
*/
|
||||
|
||||
import { buildResult } from '../utils/index.js';
|
||||
|
||||
import * as claude from './claude.js';
|
||||
import * as codex from './codex.js';
|
||||
import * as copilot from './copilot.js';
|
||||
import * as google from './google/index.js';
|
||||
import * as kimi from './kimi.js';
|
||||
import * as nanogpt from './nanogpt.js';
|
||||
import * as openai from './openai.js';
|
||||
import * as openrouter from './openrouter.js';
|
||||
import * as zai from './zai.js';
|
||||
import * as minimaxCodingPlan from './minimax-coding-plan.js';
|
||||
import * as minimaxCnCodingPlan from './minimax-cn-coding-plan.js';
|
||||
import * as ollamaCloud from './ollama-cloud.js';
|
||||
|
||||
const registry = {
|
||||
claude: {
|
||||
providerId: claude.providerId,
|
||||
providerName: claude.providerName,
|
||||
isConfigured: claude.isConfigured,
|
||||
fetchQuota: claude.fetchQuota
|
||||
},
|
||||
codex: {
|
||||
providerId: codex.providerId,
|
||||
providerName: codex.providerName,
|
||||
isConfigured: codex.isConfigured,
|
||||
fetchQuota: codex.fetchQuota
|
||||
},
|
||||
google: {
|
||||
providerId: 'google',
|
||||
providerName: 'Google',
|
||||
isConfigured: () => google.resolveGoogleAuthSources().length > 0,
|
||||
fetchQuota: google.fetchGoogleQuota
|
||||
},
|
||||
'zai-coding-plan': {
|
||||
providerId: zai.providerId,
|
||||
providerName: zai.providerName,
|
||||
isConfigured: zai.isConfigured,
|
||||
fetchQuota: zai.fetchQuota
|
||||
},
|
||||
'kimi-for-coding': {
|
||||
providerId: kimi.providerId,
|
||||
providerName: kimi.providerName,
|
||||
isConfigured: kimi.isConfigured,
|
||||
fetchQuota: kimi.fetchQuota
|
||||
},
|
||||
openrouter: {
|
||||
providerId: openrouter.providerId,
|
||||
providerName: openrouter.providerName,
|
||||
isConfigured: openrouter.isConfigured,
|
||||
fetchQuota: openrouter.fetchQuota
|
||||
},
|
||||
'nano-gpt': {
|
||||
providerId: nanogpt.providerId,
|
||||
providerName: nanogpt.providerName,
|
||||
isConfigured: nanogpt.isConfigured,
|
||||
fetchQuota: nanogpt.fetchQuota
|
||||
},
|
||||
'github-copilot': {
|
||||
providerId: copilot.providerId,
|
||||
providerName: copilot.providerName,
|
||||
isConfigured: copilot.isConfigured,
|
||||
fetchQuota: copilot.fetchQuota
|
||||
},
|
||||
'github-copilot-addon': {
|
||||
providerId: copilot.providerIdAddon,
|
||||
providerName: copilot.providerNameAddon,
|
||||
isConfigured: copilot.isConfigured,
|
||||
fetchQuota: copilot.fetchQuotaAddon
|
||||
},
|
||||
'minimax-coding-plan': {
|
||||
providerId: minimaxCodingPlan.providerId,
|
||||
providerName: minimaxCodingPlan.providerName,
|
||||
isConfigured: minimaxCodingPlan.isConfigured,
|
||||
fetchQuota: minimaxCodingPlan.fetchQuota
|
||||
},
|
||||
'minimax-cn-coding-plan': {
|
||||
providerId: minimaxCnCodingPlan.providerId,
|
||||
providerName: minimaxCnCodingPlan.providerName,
|
||||
isConfigured: minimaxCnCodingPlan.isConfigured,
|
||||
fetchQuota: minimaxCnCodingPlan.fetchQuota
|
||||
},
|
||||
'ollama-cloud': {
|
||||
providerId: ollamaCloud.providerId,
|
||||
providerName: ollamaCloud.providerName,
|
||||
isConfigured: ollamaCloud.isConfigured,
|
||||
fetchQuota: ollamaCloud.fetchQuota
|
||||
}
|
||||
};
|
||||
|
||||
export const listConfiguredQuotaProviders = () => {
|
||||
const configured = [];
|
||||
|
||||
for (const [id, provider] of Object.entries(registry)) {
|
||||
try {
|
||||
if (provider.isConfigured()) {
|
||||
configured.push(id);
|
||||
}
|
||||
} catch {
|
||||
// Ignore provider-specific config errors in list API.
|
||||
}
|
||||
}
|
||||
|
||||
return configured;
|
||||
};
|
||||
|
||||
export const fetchQuotaForProvider = async (providerId) => {
|
||||
const provider = registry[providerId];
|
||||
|
||||
if (!provider) {
|
||||
return buildResult({
|
||||
providerId,
|
||||
providerName: providerId,
|
||||
ok: false,
|
||||
configured: false,
|
||||
error: 'Unsupported provider'
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
return await provider.fetchQuota();
|
||||
} catch (error) {
|
||||
return buildResult({
|
||||
providerId: provider.providerId,
|
||||
providerName: provider.providerName,
|
||||
ok: false,
|
||||
configured: true,
|
||||
error: error instanceof Error ? error.message : 'Request failed'
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
export const fetchClaudeQuota = claude.fetchQuota;
|
||||
export const fetchOpenaiQuota = openai.fetchQuota;
|
||||
export const fetchGoogleQuota = google.fetchGoogleQuota;
|
||||
export const fetchCodexQuota = codex.fetchQuota;
|
||||
export const fetchCopilotQuota = copilot.fetchQuota;
|
||||
export const fetchCopilotAddonQuota = copilot.fetchQuotaAddon;
|
||||
export const fetchKimiQuota = kimi.fetchQuota;
|
||||
export const fetchOpenRouterQuota = openrouter.fetchQuota;
|
||||
export const fetchZaiQuota = zai.fetchQuota;
|
||||
export const fetchNanoGptQuota = nanogpt.fetchQuota;
|
||||
export const fetchMinimaxCodingPlanQuota = minimaxCodingPlan.fetchQuota;
|
||||
export const fetchMinimaxCnCodingPlanQuota = minimaxCnCodingPlan.fetchQuota;
|
||||
export const fetchOllamaCloudQuota = ollamaCloud.fetchQuota;
|
||||
Reference in New Issue
Block a user