MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-03-12 23:21:27 -07:00
Code Issues Packages Projects Releases Wiki Activity

update: add Pro 3.1

Browse Source
This commit is contained in:
Gaurav Ghosh
2026-02-25 22:26:18 -08:00
parent 6836f0e1b2
commit 5cc9644a16
8 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/cli/src/config/settingsSchema.ts
View File

@@ -835,7 +835,7 @@ const SETTINGS_SCHEMA = {
requiresRestart: false,
default: {},
description: 'Billing and AI credits settings.',
showInDialog: true,
showInDialog: false,
properties: {
overageStrategy: {
type: 'enum',

2
packages/cli/src/ui/commands/statsCommand.test.ts
View File

@@ -62,7 +62,7 @@ describe('statsCommand', () => {
tier: undefined,
userEmail: 'mock@example.com',
currentModel: undefined,
creditBalance: null,
creditBalance: undefined,
});
});

2
packages/cli/src/ui/commands/statsCommand.ts
View File

@@ -31,7 +31,7 @@ function getUserIdentity(context: CommandContext) {
const tier = context.services.config?.getUserTierName();
const paidTier = context.services.config?.getUserPaidTier();
const creditBalance = getG1CreditBalance(paidTier);
const creditBalance = getG1CreditBalance(paidTier) ?? undefined;
return { selectedAuthType, userEmail, tier, creditBalance };
}

3
packages/cli/src/ui/components/StatsDisplay.tsx
View File

@@ -395,8 +395,7 @@ interface StatsDisplayProps {
tier?: string;
currentModel?: string;
quotaStats?: QuotaStats;
/** G1 AI Credits balance, null if not eligible */
creditBalance?: number | null;
creditBalance?: number;
}
export const StatsDisplay: React.FC<StatsDisplayProps> = ({

4
packages/cli/src/ui/hooks/creditsFlowHandler.ts
View File

@@ -74,8 +74,8 @@ export async function handleCreditsFlow(
const { overageStrategy } = args;
// If credits are already auto-enabled (strategy='always'), the request
// that just failed already included credits — they didn't help.
// Fall through to ProQuotaDialog which offers the Flash downgrade.
// that just failed already included enabledCreditTypes — credits didn't
// help. Fall through to ProQuotaDialog which offers the Flash downgrade.
if (shouldAutoUseCredits(overageStrategy, creditBalance)) {
return null;
}

3
packages/cli/src/ui/types.ts
View File

@@ -201,8 +201,7 @@ export type HistoryItemStats = HistoryItemQuotaBase & {
type: 'stats';
duration: string;
quotas?: RetrieveUserQuotaResponse;
/** G1 AI Credits balance, null if not eligible */
creditBalance?: number | null;
creditBalance?: number;
};
export type HistoryItemModelStats = HistoryItemQuotaBase & {

2
packages/core/src/billing/billing.test.ts
View File

@@ -226,7 +226,7 @@ describe('billing', () => {
});
it('should return true for gemini-3.1-pro-preview', () => {
expect(isOverageEligibleModel('gemini-3.1-pro-preview')).toBe(false);
expect(isOverageEligibleModel('gemini-3.1-pro-preview')).toBe(true);
});
it('should return true for gemini-3.1-pro-preview-customtools', () => {

10
packages/core/src/billing/billing.ts
View File

@@ -9,7 +9,10 @@ import type {
CreditType,
GeminiUserTier,
} from '../code_assist/types.js';
import { PREVIEW_GEMINI_MODEL } from '../config/models.js';
import {
PREVIEW_GEMINI_MODEL,
PREVIEW_GEMINI_3_1_MODEL,
} from '../config/models.js';
/**
* Strategy for handling quota exhaustion when AI credits are available.
@@ -26,7 +29,10 @@ export const G1_CREDIT_TYPE: CreditType = 'GOOGLE_ONE_AI';
* The set of models that support AI credits overage billing.
* Only these models are eligible for the credits-based retry flow.
*/
export const OVERAGE_ELIGIBLE_MODELS = new Set([PREVIEW_GEMINI_MODEL]);
export const OVERAGE_ELIGIBLE_MODELS = new Set([
PREVIEW_GEMINI_MODEL,
PREVIEW_GEMINI_3_1_MODEL,
]);
/**
* Checks if a model is eligible for AI credits overage billing.