From 6276890e5bfa155a40b800967f1c29b860ac96bb Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 15 Jan 2025 11:56:51 +0800
Subject: [PATCH] feat: replaced visionicon with modeltags

---
 src/renderer/src/components/Popups/SelectModelPopup.tsx | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/components/Popups/SelectModelPopup.tsx b/src/renderer/src/components/Popups/SelectModelPopup.tsx
index 1ca48bc6..6f249692 100644
--- a/src/renderer/src/components/Popups/SelectModelPopup.tsx
+++ b/src/renderer/src/components/Popups/SelectModelPopup.tsx
@@ -1,7 +1,6 @@
 import { PushpinOutlined, SearchOutlined } from '@ant-design/icons'
-import VisionIcon from '@renderer/components/Icons/VisionIcon'
 import { TopView } from '@renderer/components/TopView'
-import { getModelLogo, isEmbeddingModel, isVisionModel } from '@renderer/config/models'
+import { getModelLogo, isEmbeddingModel } from '@renderer/config/models'
 import db from '@renderer/databases'
 import { useProviders } from '@renderer/hooks/useProvider'
 import { getModelUniqId } from '@renderer/services/ModelService'
@@ -118,7 +117,7 @@ const PopupContainer: React.FC<PopupContainerProps> = ({ model, resolve }) => {
         key: getModelUniqId(m) + '_pinned',
         label: (
           <ModelItem>
-            {m?.name} {isVisionModel(m) && <VisionIcon />}
+            {m?.name} <ModelTags model={m} />
             <PinIcon
               onClick={(e) => {
                 e.stopPropagation()