update: add Pro 3.1

Author: gsquared94

packages/cli/src/config/settingsSchema.ts

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

packages/cli/src/ui/commands/statsCommand.test.ts

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

packages/cli/src/ui/commands/statsCommand.ts

@@ -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 };
 }

packages/cli/src/ui/components/StatsDisplay.tsx

@@ -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> = ({

packages/cli/src/ui/hooks/creditsFlowHandler.ts

@@ -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;
   }

packages/cli/src/ui/types.ts

@@ -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 & {

packages/core/src/billing/billing.test.ts

@@ -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', () => {

packages/core/src/billing/billing.ts

@@ -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.