diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index 251082de75..762b821d7b 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -2161,9 +2161,15 @@ export const GEMINI_SEARCH_MODELS = [ 'gemini-2.0-pro-exp-02-05', 'gemini-2.0-pro-exp', 'gemini-2.5-pro-exp', - 'gemini-2.5-pro-exp-03-25' + 'gemini-2.5-pro-exp-03-25', + 'gemini-2.5-pro-preview', + 'gemini-2.5-pro-preview-03-25', + 'gemini-2.5-flash-preview', + 'gemini-2.5-flash-preview-04-17' ] +export const PERPLEXITY_SEARCH_MODELS = ['sonar-pro', 'sonar', 'sonar-reasoning', 'sonar-reasoning-pro'] + export function isTextToImageModel(model: Model): boolean { return TEXT_TO_IMAGE_REGEX.test(model.id) } @@ -2319,6 +2325,10 @@ export function isWebSearchModel(model: Model): boolean { return false } + if (provider.id === 'perplexity') { + return PERPLEXITY_SEARCH_MODELS.includes(model?.id) + } + if (provider.id === 'aihubmix') { const models = ['gemini-2.0-flash-search', 'gemini-2.0-flash-exp-search', 'gemini-2.0-pro-exp-02-05-search'] return models.includes(model?.id)