From f273621082ad5952a962d7fd0cfdf8ae1101f2b7 Mon Sep 17 00:00:00 2001 From: SuYao Date: Wed, 27 Aug 2025 00:50:47 +0800 Subject: [PATCH] =?UTF-8?q?fix(models):=20enhance=20model=20type=20checks?= =?UTF-8?q?=20for=20function=20calling=20and=20web=20s=E2=80=A6=20(#9579)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit fix(models): enhance model type checks for function calling and web search models --- src/renderer/src/config/models.ts | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index fb2ee8dd67..19247e63d3 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -359,7 +359,7 @@ export const getThinkModelType = (model: Model): ThinkingModelType => { } export function isFunctionCallingModel(model?: Model): boolean { - if (!model || isEmbeddingModel(model) || isRerankModel(model)) { + if (!model || isEmbeddingModel(model) || isRerankModel(model) || isTextToImageModel(model)) { return false } @@ -2723,7 +2723,14 @@ export function isGeminiReasoningModel(model?: Model): boolean { export const isSupportedThinkingTokenGeminiModel = (model: Model): boolean => { const modelId = getLowerBaseModelName(model.id, '/') - return modelId.includes('gemini-2.5') + if (modelId.includes('gemini-2.5')) { + if (modelId.includes('image') || modelId.includes('tts')) { + return false + } + return true + } else { + return false + } } /** 是否为Qwen推理模型 */ @@ -2957,7 +2964,7 @@ export function isNotSupportTemperatureAndTopP(model: Model): boolean { } export function isWebSearchModel(model: Model): boolean { - if (!model || isEmbeddingModel(model) || isRerankModel(model)) { + if (!model || isEmbeddingModel(model) || isRerankModel(model) || isTextToImageModel(model)) { return false }