whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(UI): exclude rerank models from mention dropdown (#3958)

Browse Source
This commit is contained in:
purefkh 2025-03-26 16:17:07 +08:00 committed by GitHub
parent 3cbe45fc8d
commit 0d320120a4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx
View File

@ -1,6 +1,6 @@
import { PushpinOutlined } from '@ant-design/icons' import { PushpinOutlined } from '@ant-design/icons'
import ModelTags from '@renderer/components/ModelTags' import ModelTags from '@renderer/components/ModelTags'
import { getModelLogo, isEmbeddingModel } from '@renderer/config/models' import { getModelLogo, isEmbeddingModel, isRerankModel } from '@renderer/config/models'
import db from '@renderer/databases' import db from '@renderer/databases'
import { useProviders } from '@renderer/hooks/useProvider' import { useProviders } from '@renderer/hooks/useProvider'
import { EVENT_NAMES, EventEmitter } from '@renderer/services/EventService' import { EVENT_NAMES, EventEmitter } from '@renderer/services/EventService'
@ -67,6 +67,7 @@ const MentionModelsButton: FC<Props> = ({ mentionModels, onMentionModel: onSelec
.map((p) => { .map((p) => {
const filteredModels = sortBy(p.models, ['group', 'name']) const filteredModels = sortBy(p.models, ['group', 'name'])
.filter((m) => !isEmbeddingModel(m)) .filter((m) => !isEmbeddingModel(m))
.filter((m) => !isRerankModel(m))
// Filter out pinned models from regular groups // Filter out pinned models from regular groups
.filter((m) => !pinnedModels.includes(getModelUniqId(m))) .filter((m) => !pinnedModels.includes(getModelUniqId(m)))
// Filter by search text // Filter by search text