125 lines
3.2 KiB
JavaScript
125 lines
3.2 KiB
JavaScript
/**
|
|
* Google Provider
|
|
*
|
|
* Google quota provider implementation.
|
|
* @module quota/providers/google
|
|
*/
|
|
|
|
export {
|
|
resolveGoogleOAuthClient,
|
|
resolveGeminiCliAuth,
|
|
resolveAntigravityAuth,
|
|
resolveGoogleAuthSources,
|
|
DEFAULT_PROJECT_ID
|
|
} from './auth.js';
|
|
|
|
export {
|
|
resolveGoogleWindow,
|
|
transformQuotaBucket,
|
|
transformModelData
|
|
} from './transforms.js';
|
|
|
|
export {
|
|
refreshGoogleAccessToken,
|
|
fetchGoogleQuotaBuckets,
|
|
fetchGoogleModels
|
|
} from './api.js';
|
|
|
|
import { buildResult } from '../../utils/index.js';
|
|
import {
|
|
resolveGoogleAuthSources,
|
|
resolveGoogleOAuthClient,
|
|
DEFAULT_PROJECT_ID
|
|
} from './auth.js';
|
|
import { transformQuotaBucket, transformModelData } from './transforms.js';
|
|
import {
|
|
refreshGoogleAccessToken,
|
|
fetchGoogleQuotaBuckets,
|
|
fetchGoogleModels
|
|
} from './api.js';
|
|
|
|
export const fetchGoogleQuota = async () => {
|
|
const authSources = resolveGoogleAuthSources();
|
|
if (!authSources.length) {
|
|
return buildResult({
|
|
providerId: 'google',
|
|
providerName: 'Google',
|
|
ok: false,
|
|
configured: false,
|
|
error: 'Not configured'
|
|
});
|
|
}
|
|
|
|
const models = {};
|
|
const sourceErrors = [];
|
|
|
|
for (const source of authSources) {
|
|
const now = Date.now();
|
|
let accessToken = source.accessToken;
|
|
|
|
if (!accessToken || (typeof source.expires === 'number' && source.expires <= now)) {
|
|
if (!source.refreshToken) {
|
|
sourceErrors.push(`${source.sourceLabel}: Missing refresh token`);
|
|
continue;
|
|
}
|
|
const { clientId, clientSecret } = resolveGoogleOAuthClient(source.sourceId);
|
|
accessToken = await refreshGoogleAccessToken(source.refreshToken, clientId, clientSecret);
|
|
}
|
|
|
|
if (!accessToken) {
|
|
sourceErrors.push(`${source.sourceLabel}: Failed to refresh OAuth token`);
|
|
continue;
|
|
}
|
|
|
|
const projectId = source.projectId ?? DEFAULT_PROJECT_ID;
|
|
let mergedAnyModel = false;
|
|
|
|
if (source.sourceId === 'gemini') {
|
|
const quotaPayload = await fetchGoogleQuotaBuckets(accessToken, projectId);
|
|
const buckets = Array.isArray(quotaPayload?.buckets) ? quotaPayload.buckets : [];
|
|
|
|
for (const bucket of buckets) {
|
|
const transformed = transformQuotaBucket(bucket, source.sourceId);
|
|
if (transformed) {
|
|
Object.assign(models, transformed);
|
|
mergedAnyModel = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
const payload = await fetchGoogleModels(accessToken, projectId);
|
|
if (payload) {
|
|
for (const [modelName, modelData] of Object.entries(payload.models ?? {})) {
|
|
const transformed = transformModelData(modelName, modelData, source.sourceId);
|
|
Object.assign(models, transformed);
|
|
mergedAnyModel = true;
|
|
}
|
|
}
|
|
|
|
if (!mergedAnyModel) {
|
|
sourceErrors.push(`${source.sourceLabel}: Failed to fetch models`);
|
|
}
|
|
}
|
|
|
|
if (!Object.keys(models).length) {
|
|
return buildResult({
|
|
providerId: 'google',
|
|
providerName: 'Google',
|
|
ok: false,
|
|
configured: true,
|
|
error: sourceErrors[0] ?? 'Failed to fetch models'
|
|
});
|
|
}
|
|
|
|
return buildResult({
|
|
providerId: 'google',
|
|
providerName: 'Google',
|
|
ok: true,
|
|
configured: true,
|
|
usage: {
|
|
windows: {},
|
|
models: Object.keys(models).length ? models : undefined
|
|
}
|
|
});
|
|
};
|