From 0d320120a454934603228ace84514cfa6a1d7cfc Mon Sep 17 00:00:00 2001 From: purefkh Date: Wed, 26 Mar 2025 16:17:07 +0800 Subject: [PATCH] fix(UI): exclude rerank models from mention dropdown (#3958) --- src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx b/src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx index 71eb504c..7cffbaca 100644 --- a/src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx +++ b/src/renderer/src/pages/home/Inputbar/MentionModelsButton.tsx @@ -1,6 +1,6 @@ import { PushpinOutlined } from '@ant-design/icons' 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 { useProviders } from '@renderer/hooks/useProvider' import { EVENT_NAMES, EventEmitter } from '@renderer/services/EventService' @@ -67,6 +67,7 @@ const MentionModelsButton: FC = ({ mentionModels, onMentionModel: onSelec .map((p) => { const filteredModels = sortBy(p.models, ['group', 'name']) .filter((m) => !isEmbeddingModel(m)) + .filter((m) => !isRerankModel(m)) // Filter out pinned models from regular groups .filter((m) => !pinnedModels.includes(getModelUniqId(m))) // Filter by search text