Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-11 08:19:01 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

Browse Source 
* 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 2833c377fa
commit 0744e42be9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/renderer/src/config/models.ts
View File

@ -2161,9 +2161,15 @@ export const GEMINI_SEARCH_MODELS = [
'gemini-2.0-pro-exp-02-05', 'gemini-2.0-pro-exp-02-05',
'gemini-2.0-pro-exp', 'gemini-2.0-pro-exp',
'gemini-2.5-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 { export function isTextToImageModel(model: Model): boolean {
return TEXT_TO_IMAGE_REGEX.test(model.id) return TEXT_TO_IMAGE_REGEX.test(model.id)
} }
@ -2319,6 +2325,10 @@ export function isWebSearchModel(model: Model): boolean {
return false return false
} }
if (provider.id === 'perplexity') {
return PERPLEXITY_SEARCH_MODELS.includes(model?.id)
}
if (provider.id === 'aihubmix') { 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'] 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) return models.includes(model?.id)