diff --git a/src/renderer/src/config/models/__tests__/reasoning.test.ts b/src/renderer/src/config/models/__tests__/reasoning.test.ts index e8ee19adf6..350758c3e2 100644 --- a/src/renderer/src/config/models/__tests__/reasoning.test.ts +++ b/src/renderer/src/config/models/__tests__/reasoning.test.ts @@ -1016,7 +1016,7 @@ describe('Gemini Models', () => { provider: '', group: '' }) - ).toBe(false) + ).toBe(true) expect( isSupportedThinkingTokenGeminiModel({ id: 'gemini-3.0-flash-image-preview', @@ -1224,7 +1224,7 @@ describe('Gemini Models', () => { provider: '', group: '' }) - ).toBe(false) + ).toBe(true) expect( isGeminiReasoningModel({ id: 'gemini-3.5-flash-image-preview', diff --git a/src/renderer/src/config/models/reasoning.ts b/src/renderer/src/config/models/reasoning.ts index c85029db8f..86d3fe97d2 100644 --- a/src/renderer/src/config/models/reasoning.ts +++ b/src/renderer/src/config/models/reasoning.ts @@ -277,6 +277,10 @@ export const GEMINI_THINKING_MODEL_REGEX = export const isSupportedThinkingTokenGeminiModel = (model: Model): boolean => { const modelId = getLowerBaseModelName(model.id, '/') if (GEMINI_THINKING_MODEL_REGEX.test(modelId)) { + // ref: https://docs.cloud.google.com/vertex-ai/generative-ai/docs/models/gemini/3-pro-image + if (modelId.includes('gemini-3-pro-image')) { + return true + } if (modelId.includes('image') || modelId.includes('tts')) { return false }