From a051f9fa442ee6f9e2e26009129218f1f0d4b892 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Tue, 7 Jan 2025 11:23:32 +0800
Subject: [PATCH] feat: add optional free model tag display

---
 src/renderer/src/components/ModelTags.tsx                    | 5 +++--
 src/renderer/src/pages/home/components/SelectModelButton.tsx | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/components/ModelTags.tsx b/src/renderer/src/components/ModelTags.tsx
index b6a2d145..e798b9be 100644
--- a/src/renderer/src/components/ModelTags.tsx
+++ b/src/renderer/src/components/ModelTags.tsx
@@ -10,15 +10,16 @@ import WebSearchIcon from './Icons/WebSearchIcon'
 
 interface ModelTagsProps {
   model: Model
+  showFree?: boolean
 }
 
-const ModelTags: FC<ModelTagsProps> = ({ model }) => {
+const ModelTags: FC<ModelTagsProps> = ({ model, showFree = true }) => {
   const { t } = useTranslation()
   return (
     <>
       {isVisionModel(model) && <VisionIcon />}
       {isWebSearchModel(model) && <WebSearchIcon />}
-      {isFreeModel(model) && (
+      {showFree && isFreeModel(model) && (
         <Tag style={{ marginLeft: 10 }} color="green">
           {t('models.free')}
         </Tag>
diff --git a/src/renderer/src/pages/home/components/SelectModelButton.tsx b/src/renderer/src/pages/home/components/SelectModelButton.tsx
index d17688f6..137f9b48 100644
--- a/src/renderer/src/pages/home/components/SelectModelButton.tsx
+++ b/src/renderer/src/pages/home/components/SelectModelButton.tsx
@@ -39,7 +39,7 @@ const SelectModelButton: FC<Props> = ({ assistant }) => {
         <ModelName>
           {model ? model.name : t('button.select_model')} {providerName ? '| ' + providerName : ''}
         </ModelName>
-        <ModelTags model={model} />
+        <ModelTags model={model} showFree={false} />
       </ButtonContent>
     </DropdownButton>
   )