diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index c9e6174c..0556c0f6 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -1009,5 +1009,13 @@ export function isWebSearchModel(model: Model): boolean { return false } - return (provider.id === 'gemini' || provider?.type === 'gemini') && model?.id === 'gemini-2.0-flash-exp' + if (provider.id === 'gemini' || provider?.type === 'gemini') { + return model?.id === 'gemini-2.0-flash-exp' + } + + if (provider.id === 'hunyuan') { + return model?.id !== 'hunyuan-lite' + } + + return false } diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts index 3c19be48..6110ff7d 100644 --- a/src/renderer/src/providers/OpenAIProvider.ts +++ b/src/renderer/src/providers/OpenAIProvider.ts @@ -1,4 +1,4 @@ -import { isEmbeddingModel, isSupportedModel, isVisionModel } from '@renderer/config/models' +import { isEmbeddingModel, isSupportedModel, isVisionModel, isWebSearchModel } from '@renderer/config/models' import { getStoreSetting } from '@renderer/hooks/useSettings' import i18n from '@renderer/i18n' import { getAssistantSettings, getDefaultModel, getTopNamingModel } from '@renderer/services/AssistantService' @@ -149,6 +149,7 @@ export default class OpenAIProvider extends BaseProvider { max_tokens: maxTokens, keep_alive: this.keepAliveTime, stream: isSupportStreamOutput, + ...(isWebSearchModel(model) ? { enable_enhancement: true } : {}), ...this.getCustomParameters(assistant) })