Skip to content

Commit b36788e

Browse files
anishs1207anishs1207
authored
fix(core): add aliases and thinking config for gemini-3.1 models (#27007)
1 parent d32c9b7 commit b36788e

6 files changed

Lines changed: 178 additions & 2 deletions

File tree

‎docs/reference/configuration.md‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -550,6 +550,24 @@ their corresponding top-level category object in your `settings.json` file.
550550
"model": "gemini-3-flash-preview"
551551
}
552552
},
553+
"gemini-3.1-pro-preview": {
554+
"extends": "chat-base-3",
555+
"modelConfig": {
556+
"model": "gemini-3.1-pro-preview"
557+
}
558+
},
559+
"gemini-3.1-pro-preview-customtools": {
560+
"extends": "chat-base-3",
561+
"modelConfig": {
562+
"model": "gemini-3.1-pro-preview-customtools"
563+
}
564+
},
565+
"gemini-3.1-flash-lite-preview": {
566+
"extends": "chat-base-3",
567+
"modelConfig": {
568+
"model": "gemini-3.1-flash-lite-preview"
569+
}
570+
},
553571
"gemini-2.5-pro": {
554572
"extends": "chat-base-2.5",
555573
"modelConfig": {

‎packages/core/src/config/defaultModelConfigs.ts‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,24 @@ export const DEFAULT_MODEL_CONFIGS: ModelConfigServiceConfig = {
7171
model: 'gemini-3-flash-preview',
7272
},
7373
},
74+
'gemini-3.1-pro-preview': {
75+
extends: 'chat-base-3',
76+
modelConfig: {
77+
model: 'gemini-3.1-pro-preview',
78+
},
79+
},
80+
'gemini-3.1-pro-preview-customtools': {
81+
extends: 'chat-base-3',
82+
modelConfig: {
83+
model: 'gemini-3.1-pro-preview-customtools',
84+
},
85+
},
86+
'gemini-3.1-flash-lite-preview': {
87+
extends: 'chat-base-3',
88+
modelConfig: {
89+
model: 'gemini-3.1-flash-lite-preview',
90+
},
91+
},
7492
'gemini-2.5-pro': {
7593
extends: 'chat-base-2.5',
7694
modelConfig: {

‎packages/core/src/config/models.test.ts‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -672,3 +672,35 @@ describe('isActiveModel', () => {
672672
).toBe(false);
673673
});
674674
});
675+
676+
describe('Gemini 3.1 Config Resolution', () => {
677+
it('PREVIEW_GEMINI_3_1_MODEL should resolve to chat-base-3 config (including thinkingLevel)', () => {
678+
const resolved = modelConfigService.getResolvedConfig({
679+
model: PREVIEW_GEMINI_3_1_MODEL,
680+
isChatModel: true,
681+
});
682+
expect(
683+
resolved.generateContentConfig?.thinkingConfig?.thinkingLevel,
684+
).toBeDefined();
685+
});
686+
687+
it('PREVIEW_GEMINI_3_1_CUSTOM_TOOLS_MODEL should resolve to chat-base-3 config (including thinkingLevel)', () => {
688+
const resolved = modelConfigService.getResolvedConfig({
689+
model: PREVIEW_GEMINI_3_1_CUSTOM_TOOLS_MODEL,
690+
isChatModel: true,
691+
});
692+
expect(
693+
resolved.generateContentConfig?.thinkingConfig?.thinkingLevel,
694+
).toBeDefined();
695+
});
696+
697+
it('PREVIEW_GEMINI_3_1_FLASH_LITE_MODEL should resolve to chat-base-3 config (including thinkingLevel)', () => {
698+
const resolved = modelConfigService.getResolvedConfig({
699+
model: PREVIEW_GEMINI_3_1_FLASH_LITE_MODEL,
700+
isChatModel: true,
701+
});
702+
expect(
703+
resolved.generateContentConfig?.thinkingConfig?.thinkingLevel,
704+
).toBeDefined();
705+
});
706+
});

‎packages/core/src/services/test-data/resolved-aliases-retry.golden.json‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,42 @@
6161
"topK": 64
6262
}
6363
},
64+
"gemini-3.1-pro-preview": {
65+
"model": "gemini-3.1-pro-preview",
66+
"generateContentConfig": {
67+
"temperature": 1,
68+
"topP": 0.95,
69+
"thinkingConfig": {
70+
"includeThoughts": true,
71+
"thinkingLevel": "HIGH"
72+
},
73+
"topK": 64
74+
}
75+
},
76+
"gemini-3.1-pro-preview-customtools": {
77+
"model": "gemini-3.1-pro-preview-customtools",
78+
"generateContentConfig": {
79+
"temperature": 1,
80+
"topP": 0.95,
81+
"thinkingConfig": {
82+
"includeThoughts": true,
83+
"thinkingLevel": "HIGH"
84+
},
85+
"topK": 64
86+
}
87+
},
88+
"gemini-3.1-flash-lite-preview": {
89+
"model": "gemini-3.1-flash-lite-preview",
90+
"generateContentConfig": {
91+
"temperature": 1,
92+
"topP": 0.95,
93+
"thinkingConfig": {
94+
"includeThoughts": true,
95+
"thinkingLevel": "HIGH"
96+
},
97+
"topK": 64
98+
}
99+
},
64100
"gemini-2.5-pro": {
65101
"model": "gemini-2.5-pro",
66102
"generateContentConfig": {

‎packages/core/src/services/test-data/resolved-aliases.golden.json‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,42 @@
6161
"topK": 64
6262
}
6363
},
64+
"gemini-3.1-pro-preview": {
65+
"model": "gemini-3.1-pro-preview",
66+
"generateContentConfig": {
67+
"temperature": 1,
68+
"topP": 0.95,
69+
"thinkingConfig": {
70+
"includeThoughts": true,
71+
"thinkingLevel": "HIGH"
72+
},
73+
"topK": 64
74+
}
75+
},
76+
"gemini-3.1-pro-preview-customtools": {
77+
"model": "gemini-3.1-pro-preview-customtools",
78+
"generateContentConfig": {
79+
"temperature": 1,
80+
"topP": 0.95,
81+
"thinkingConfig": {
82+
"includeThoughts": true,
83+
"thinkingLevel": "HIGH"
84+
},
85+
"topK": 64
86+
}
87+
},
88+
"gemini-3.1-flash-lite-preview": {
89+
"model": "gemini-3.1-flash-lite-preview",
90+
"generateContentConfig": {
91+
"temperature": 1,
92+
"topP": 0.95,
93+
"thinkingConfig": {
94+
"includeThoughts": true,
95+
"thinkingLevel": "HIGH"
96+
},
97+
"topK": 64
98+
}
99+
},
64100
"gemini-2.5-pro": {
65101
"model": "gemini-2.5-pro",
66102
"generateContentConfig": {

‎schemas/settings.schema.json‎

Lines changed: 38 additions & 2 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)