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

fix: set thinking budget to 0 for gemini-2.5-flash when reasoning effort is off (#6917)

Browse Source
This commit is contained in:
purefkh 2025-06-06 22:11:18 +08:00 committed by GitHub
parent 9ba2dea148
commit 7f12c2f8b8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/renderer/src/config/models.ts
View File

@ -2631,7 +2631,8 @@ export function groupQwenModels(models: Model[]): Record<string, Model[]> {
export const THINKING_TOKEN_MAP: Record<string, { min: number; max: number }> = {
// Gemini models
'gemini-.*$': { min: 0, max: 24576 },
'gemini-.*-flash.*$': { min: 0, max: 24576 },
'gemini-.*-pro.*$': { min: 128, max: 32768 },
// Qwen models
'qwen-plus-.*$': { min: 0, max: 38912 },

4
src/renderer/src/providers/AiProvider/GeminiProvider.ts
View File

@ -286,12 +286,14 @@ export default class GeminiProvider extends BaseProvider {
private getBudgetToken(assistant: Assistant, model: Model) {
if (isGeminiReasoningModel(model)) {
const reasoningEffort = assistant?.settings?.reasoning_effort
const GEMINI_FLASH_MODEL_REGEX = new RegExp('gemini-.*-flash.*$')
// 如果thinking_budget是undefined，不思考
if (reasoningEffort === undefined) {
return {
thinkingConfig: {
includeThoughts: false
includeThoughts: false,
...(GEMINI_FLASH_MODEL_REGEX.test(model.id) ? { thinkingBudget: 0 } : {})
} as ThinkingConfig
}
}