fix failing tests, update default config for 3.1 flash lite

This commit is contained in:
davidapierce
2026-05-15 19:38:17 +00:00
parent 2d4cfac3a4
commit 07fb3efe08
3 changed files with 8 additions and 24 deletions

View File

@@ -108,7 +108,7 @@ export const DEFAULT_MODEL_CONFIGS: ModelConfigServiceConfig = {
},
},
'gemini-3.1-flash-lite': {
extends: 'chat-base-2.5',
extends: 'chat-base-3',
modelConfig: {
model: 'gemini-3.1-flash-lite',
},
@@ -501,12 +501,6 @@ export const DEFAULT_MODEL_CONFIGS: ModelConfigServiceConfig = {
},
'gemini-3.1-flash-lite': {
default: 'gemini-3.1-flash-lite',
contexts: [
{
condition: { useGemini3_1FlashLite: false },
target: 'gemini-2.5-flash-lite',
},
],
},
flash: {
default: 'gemini-3-flash-preview',
@@ -519,12 +513,6 @@ export const DEFAULT_MODEL_CONFIGS: ModelConfigServiceConfig = {
},
'flash-lite': {
default: 'gemini-3.1-flash-lite',
contexts: [
{
condition: { useGemini3_1FlashLite: false },
target: 'gemini-2.5-flash-lite',
},
],
},
'auto-gemini-3': {
default: 'gemini-3-pro-preview',

View File

@@ -609,7 +609,6 @@ describe('isActiveModel', () => {
it('should return false for Gemini 3.1 models when Gemini 3.1 is not launched', () => {
expect(isActiveModel(PREVIEW_GEMINI_3_1_MODEL)).toBe(false);
expect(isActiveModel(PREVIEW_GEMINI_FLASH_LITE_MODEL)).toBe(false);
});
it('should return true for unknown models and aliases', () => {
@@ -625,11 +624,11 @@ describe('isActiveModel', () => {
expect(isActiveModel(DEFAULT_GEMINI_MODEL, true)).toBe(true);
});
it('should return true for PREVIEW_GEMINI_FLASH_LITE_MODEL and DEFAULT_GEMINI_FLASH_LITE_MODEL only when useGemini3_1FlashLite is true', () => {
it('should return true for PREVIEW_GEMINI_FLASH_LITE_MODEL only when useGemini3_1FlashLite is true, and always true for DEFAULT_GEMINI_FLASH_LITE_MODEL', () => {
expect(isActiveModel(PREVIEW_GEMINI_FLASH_LITE_MODEL, false, true)).toBe(
true,
);
expect(isActiveModel(DEFAULT_GEMINI_FLASH_LITE_MODEL, false, true)).toBe(
expect(isActiveModel(DEFAULT_GEMINI_FLASH_LITE_MODEL, false, false)).toBe(
true,
);
expect(isActiveModel(PREVIEW_GEMINI_FLASH_LITE_MODEL, true, true)).toBe(
@@ -642,7 +641,7 @@ describe('isActiveModel', () => {
false,
);
expect(isActiveModel(DEFAULT_GEMINI_FLASH_LITE_MODEL, true, false)).toBe(
false,
true,
);
});
@@ -664,7 +663,7 @@ describe('isActiveModel', () => {
).toBe(false);
});
it('should return false for Gemini 3.1 models when useGemini3_1 and useGemini3_1FlashLite are false', () => {
it('should return false for Gemini 3.1 preview models when useGemini3_1 and useGemini3_1FlashLite are false', () => {
expect(isActiveModel(PREVIEW_GEMINI_3_1_MODEL, false, false, true)).toBe(
false,
);
@@ -681,7 +680,7 @@ describe('isActiveModel', () => {
false,
);
expect(isActiveModel(DEFAULT_GEMINI_FLASH_LITE_MODEL, false, false)).toBe(
false,
true,
);
});
});

View File

@@ -57,7 +57,7 @@ export const PREVIEW_GEMINI_3_1_MODEL = 'gemini-3.1-pro-preview';
export const PREVIEW_GEMINI_3_1_CUSTOM_TOOLS_MODEL =
'gemini-3.1-pro-preview-customtools';
export const PREVIEW_GEMINI_FLASH_MODEL = 'gemini-3-flash-preview';
export const PREVIEW_GEMINI_FLASH_LITE_MODEL = 'gemini-3.1-flash-lite';
export const PREVIEW_GEMINI_FLASH_LITE_MODEL = 'gemini-3.1-flash-lite-preview';
export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro';
export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash';
export const DEFAULT_GEMINI_FLASH_LITE_MODEL = 'gemini-3.1-flash-lite';
@@ -500,10 +500,7 @@ export function isActiveModel(
if (model === GEMMA_4_31B_IT_MODEL || model === GEMMA_4_26B_A4B_IT_MODEL) {
return experimentalGemma;
}
if (
model === PREVIEW_GEMINI_FLASH_LITE_MODEL ||
model === DEFAULT_GEMINI_FLASH_LITE_MODEL
) {
if (model === PREVIEW_GEMINI_FLASH_LITE_MODEL) {
return useGemini3_1FlashLite;
}
if (useGemini3_1) {