mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-04-27 13:34:15 -07:00
Gaurav Ghosh
6836f0e1b2
Adds end-to-end support for Google One AI credits in quota exhaustion flows: - New billing module (packages/core/src/billing/) with credit balance checking, overage strategy management, and G1 URL construction - OverageMenuDialog and EmptyWalletDialog UI components for quota exhaustion with credit purchase options - Credits flow handler extracted to creditsFlowHandler.ts with overage menu, empty wallet, and auto-use-credits logic - Server-side credit tracking: enabledCreditTypes on requests, consumed/remaining credits from streaming responses - Billing telemetry events (overage menu shown, option selected, credits used, credit purchase click, API key updated) - OpenTelemetry metrics for overage option and credit purchase counters - Credit balance display in /stats command with refresh support - Settings: general.overageStrategy (ask/always/never) for credit usage - Error handling: INSUFFICIENT_G1_CREDITS_BALANCE as terminal error regardless of domain field presence - Persistent info message after
31 lines
1.0 KiB
TypeScript
31 lines
1.0 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import type { Attributes } from '@opentelemetry/api';
|
|
import type { Config } from '../config/config.js';
|
|
import { InstallationManager } from '../utils/installationManager.js';
|
|
import { UserAccountManager } from '../utils/userAccountManager.js';
|
|
|
|
const userAccountManager = new UserAccountManager();
|
|
const installationManager = new InstallationManager();
|
|
|
|
export function getCommonAttributes(config: Config): Attributes {
|
|
const email = userAccountManager.getCachedGoogleAccount();
|
|
const experiments = config.getExperiments();
|
|
const authType = config.getContentGeneratorConfig()?.authType;
|
|
return {
|
|
'session.id': config.getSessionId(),
|
|
'installation.id': installationManager.getInstallationId(),
|
|
interactive: config.isInteractive(),
|
|
...(email && { 'user.email': email }),
|
|
...(authType && { auth_type: authType }),
|
|
...(experiments &&
|
|
experiments.experimentIds.length > 0 && {
|
|
'experiments.ids': experiments.experimentIds,
|
|
}),
|
|
};
|
|
}
|