Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-04 11:49:02 +08:00
Code Issues Actions 4 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 }> = { export const THINKING_TOKEN_MAP: Record<string, { min: number; max: number }> = {
// Gemini models // Gemini models
'gemini-.*$': { min: 0, max: 24576 }, 'gemini-.*-flash.*$': { min: 0, max: 24576 },
'gemini-.*-pro.*$': { min: 128, max: 32768 },
// Qwen models // Qwen models
'qwen-plus-.*$': { min: 0, max: 38912 }, '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) { private getBudgetToken(assistant: Assistant, model: Model) {
if (isGeminiReasoningModel(model)) { if (isGeminiReasoningModel(model)) {
const reasoningEffort = assistant?.settings?.reasoning_effort const reasoningEffort = assistant?.settings?.reasoning_effort
const GEMINI_FLASH_MODEL_REGEX = new RegExp('gemini-.*-flash.*$')
// 如果thinking_budget是undefined，不思考 // 如果thinking_budget是undefined，不思考
if (reasoningEffort === undefined) { if (reasoningEffort === undefined) {
return { return {
thinkingConfig: { thinkingConfig: {
includeThoughts: false includeThoughts: false,
...(GEMINI_FLASH_MODEL_REGEX.test(model.id) ? { thinkingBudget: 0 } : {})
} as ThinkingConfig } as ThinkingConfig
} }
} }