Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-19 14:41:24 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: add explicit thinking token support for gemini-3-pro-image (#11744)

Browse Source
This commit is contained in:
defi-failure 2025-12-08 11:31:05 +08:00 committed by GitHub
parent 4c4f832bc7
commit ef5b97813c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/renderer/src/config/models/__tests__/reasoning.test.ts
View File

@ -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',

4
src/renderer/src/config/models/reasoning.ts
View File

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