feat(models): add new search models (#5349)

* feat(models): add new Perplexity search models and update Gemini search models

* fix(models): update Perplexity search model names for consistency
This commit is contained in:
SuYao 2025-04-25 22:57:39 +08:00 committed by GitHub
parent a58f23f68e
commit f17fe3adb5

View File

@ -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)