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 = ({ model }) => { +const ModelTags: FC = ({ model, showFree = true }) => { const { t } = useTranslation() return ( <> {isVisionModel(model) && } {isWebSearchModel(model) && } - {isFreeModel(model) && ( + {showFree && isFreeModel(model) && ( {t('models.free')} 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 = ({ assistant }) => { {model ? model.name : t('button.select_model')} {providerName ? '| ' + providerName : ''} - + )