update: add Pro 3.1

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
+1 -1
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', () => {
+8 -2
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.