diff --git a/src/renderer/src/pages/settings/ModelSettings/ModelSettings.tsx b/src/renderer/src/pages/settings/ModelSettings/ModelSettings.tsx
index bf5ec1d3d4..a8c64f810a 100644
--- a/src/renderer/src/pages/settings/ModelSettings/ModelSettings.tsx
+++ b/src/renderer/src/pages/settings/ModelSettings/ModelSettings.tsx
@@ -1,8 +1,7 @@
-import { RedoOutlined } from '@ant-design/icons'
-import { Button, InfoTooltip, RowFlex, Tooltip } from '@cherrystudio/ui'
+import { Button, InfoTooltip, ResetIcon, RowFlex, Tooltip } from '@cherrystudio/ui'
import { usePreference } from '@data/hooks/usePreference'
import ModelSelector from '@renderer/components/ModelSelector'
-import { isEmbeddingModel, isRerankModel, isTextToImageModel } from '@renderer/config/models'
+import { DEFAULT_MODEL_MAP, isEmbeddingModel, isRerankModel, isTextToImageModel } from '@renderer/config/models'
import { useTheme } from '@renderer/context/ThemeProvider'
import { useDefaultModel } from '@renderer/hooks/useAssistant'
import { useProviders } from '@renderer/hooks/useProvider'
@@ -21,8 +20,17 @@ import DefaultAssistantSettings from './DefaultAssistantSettings'
import TopicNamingModalPopup from './QuickModelPopup'
const ModelSettings: FC = () => {
- const { defaultModel, quickModel, translateModel, setDefaultModel, setQuickModel, setTranslateModel } =
- useDefaultModel()
+ const {
+ defaultModel,
+ quickModel,
+ translateModel,
+ setDefaultModel,
+ setQuickModel,
+ setTranslateModel,
+ resetDefaultAssistantModel,
+ resetQuickModel,
+ resetTranslateModel
+ } = useDefaultModel()
const { providers } = useProviders()
const allModels = providers.map((p) => p.models).flat()
const { theme } = useTheme()
@@ -73,6 +81,11 @@ const ModelSettings: FC = () => {
+ {defaultModelValue !== getModelUniqId(DEFAULT_MODEL_MAP.assistant) && (
+
+ } style={{ marginLeft: 8 }} onClick={resetDefaultAssistantModel} />
+
+ )}
{t('settings.models.default_assistant_model_description')}
@@ -97,6 +110,11 @@ const ModelSettings: FC = () => {
+ {defaultQuickModel !== getModelUniqId(DEFAULT_MODEL_MAP.quick) && (
+
+ } style={{ marginLeft: 8 }} onClick={resetQuickModel} />
+
+ )}
{t('settings.models.quick_model.description')}
@@ -120,11 +138,9 @@ const ModelSettings: FC = () => {
- {translateModelPrompt !== TRANSLATE_PROMPT && (
-
-
+ {defaultTranslateModel !== getModelUniqId(DEFAULT_MODEL_MAP.translate) && (
+
+ } style={{ marginLeft: 8 }} onClick={resetTranslateModel} />
)}