From e6e20d2d7238b269965fc97d7f494546c659877e Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Thu, 30 Oct 2025 17:10:25 +0800
Subject: [PATCH] refactor(models): restructure default model configuration and
 usage

- Replace SYSTEM_MODELS.defaultModel array with DEFAULT_MODEL_MAP object for clearer model assignments
- Update related store and hook implementations to use new model map
- Add reset functions for default models in useAssistant hook
---
 .../InputEmbeddingDimension.test.tsx.snap     |  4 +--
 src/renderer/src/config/models/default.ts     | 18 ++++-------
 src/renderer/src/hooks/useAssistant.ts        | 32 ++++++++++++++++---
 src/renderer/src/store/llm.ts                 | 10 +++---
 src/renderer/src/store/migrate.ts             |  3 +-
 5 files changed, 44 insertions(+), 23 deletions(-)

diff --git a/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap b/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
index f8ddaa407d..f056130030 100644
--- a/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
+++ b/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
@@ -14,10 +14,10 @@ exports[`InputEmbeddingDimension > basic rendering > should match snapshot with
   />
   <div
     data-testid="tooltip"
+    data-title="自动设置维度"
   >
     <button
       aria-label="Get embedding dimension"
-      data-testid="button"
       role="button"
       type="button"
     >
@@ -49,10 +49,10 @@ exports[`InputEmbeddingDimension > basic rendering > should match snapshot with
   />
   <div
     data-testid="tooltip"
+    data-title="自动设置维度"
   >
     <button
       aria-label="Get embedding dimension"
-      data-testid="button"
       disabled=""
       role="button"
       type="button"
diff --git a/src/renderer/src/config/models/default.ts b/src/renderer/src/config/models/default.ts
index 5b242f297d..4a567712d6 100644
--- a/src/renderer/src/config/models/default.ts
+++ b/src/renderer/src/config/models/default.ts
@@ -14,17 +14,13 @@ export const qwen38bModel: Model = {
   group: 'Qwen'
 }
 
-export const SYSTEM_MODELS: Record<SystemProviderId | 'defaultModel', Model[]> = {
-  defaultModel: [
-    // Default assistant model
-    glm45FlashModel,
-    // Default topic naming model
-    qwen38bModel,
-    // Default translation model
-    glm45FlashModel,
-    // Default quick assistant model
-    glm45FlashModel
-  ],
+export const DEFAULT_MODEL_MAP = {
+  assistant: glm45FlashModel,
+  quick: qwen38bModel,
+  translate: glm45FlashModel
+} as const
+
+export const SYSTEM_MODELS: Record<SystemProviderId, Model[]> = {
   cherryin: [],
   vertexai: [],
   '302ai': [
diff --git a/src/renderer/src/hooks/useAssistant.ts b/src/renderer/src/hooks/useAssistant.ts
index dc0f5c28e3..183dd02422 100644
--- a/src/renderer/src/hooks/useAssistant.ts
+++ b/src/renderer/src/hooks/useAssistant.ts
@@ -1,5 +1,6 @@
 import { loggerService } from '@logger'
 import {
+  DEFAULT_MODEL_MAP,
   getThinkModelType,
   isSupportedReasoningEffortModel,
   isSupportedThinkingTokenModel,
@@ -24,7 +25,11 @@ import {
   updateTopic,
   updateTopics
 } from '@renderer/store/assistants'
-import { setDefaultModel, setQuickModel, setTranslateModel } from '@renderer/store/llm'
+import {
+  setDefaultModel as setDefaultModelAction,
+  setQuickModel as setQuickModelAction,
+  setTranslateModel as setTranslateModelAction
+} from '@renderer/store/llm'
 import type { Assistant, AssistantSettings, Model, ThinkingOption, Topic } from '@renderer/types'
 import { uuid } from '@renderer/utils'
 import { useCallback, useEffect, useMemo, useRef } from 'react'
@@ -198,12 +203,31 @@ export function useDefaultModel() {
   const { defaultModel, quickModel, translateModel } = useAppSelector((state) => state.llm)
   const dispatch = useAppDispatch()
 
+  const setDefaultModel = useCallback((model: Model) => dispatch(setDefaultModelAction({ model })), [dispatch])
+  const setQuickModel = useCallback((model: Model) => dispatch(setQuickModelAction({ model })), [dispatch])
+  const setTranslateModel = useCallback((model: Model) => dispatch(setTranslateModelAction({ model })), [dispatch])
+
+  const resetDefaultAssistantModel = useCallback(() => {
+    setDefaultModel(DEFAULT_MODEL_MAP.assistant)
+  }, [setDefaultModel])
+
+  const resetTranslateModel = useCallback(() => {
+    setTranslateModel(DEFAULT_MODEL_MAP.translate)
+  }, [setTranslateModel])
+
+  const resetQuickModel = useCallback(() => {
+    setTranslateModel(DEFAULT_MODEL_MAP.quick)
+  }, [setTranslateModel])
+
   return {
     defaultModel,
+    setDefaultModel,
+    resetDefaultAssistantModel,
     quickModel,
+    resetQuickModel,
+    setQuickModel,
     translateModel,
-    setDefaultModel: (model: Model) => dispatch(setDefaultModel({ model })),
-    setQuickModel: (model: Model) => dispatch(setQuickModel({ model })),
-    setTranslateModel: (model: Model) => dispatch(setTranslateModel({ model }))
+    setTranslateModel,
+    resetTranslateModel
   }
 }
diff --git a/src/renderer/src/store/llm.ts b/src/renderer/src/store/llm.ts
index 737de4440b..5b535c9b3f 100644
--- a/src/renderer/src/store/llm.ts
+++ b/src/renderer/src/store/llm.ts
@@ -1,7 +1,7 @@
 import type { PayloadAction } from '@reduxjs/toolkit'
 import { createSlice } from '@reduxjs/toolkit'
 import { isLocalAi } from '@renderer/config/env'
-import { SYSTEM_MODELS } from '@renderer/config/models'
+import { DEFAULT_MODEL_MAP } from '@renderer/config/models'
 import { SYSTEM_PROVIDERS } from '@renderer/config/providers'
 import type { Model, Provider } from '@renderer/types'
 import { uniqBy } from 'lodash'
@@ -43,10 +43,10 @@ export interface LlmState {
 }
 
 export const initialState: LlmState = {
-  defaultModel: SYSTEM_MODELS.defaultModel[0],
-  topicNamingModel: SYSTEM_MODELS.defaultModel[1],
-  quickModel: SYSTEM_MODELS.defaultModel[1],
-  translateModel: SYSTEM_MODELS.defaultModel[2],
+  defaultModel: DEFAULT_MODEL_MAP.assistant,
+  topicNamingModel: DEFAULT_MODEL_MAP.quick,
+  quickModel: DEFAULT_MODEL_MAP.quick,
+  translateModel: DEFAULT_MODEL_MAP.translate,
   quickAssistantId: '',
   providers: SYSTEM_PROVIDERS,
   settings: {
diff --git a/src/renderer/src/store/migrate.ts b/src/renderer/src/store/migrate.ts
index e2763ea0b6..9722d6bc15 100644
--- a/src/renderer/src/store/migrate.ts
+++ b/src/renderer/src/store/migrate.ts
@@ -6,6 +6,7 @@ import { nanoid } from '@reduxjs/toolkit'
 import { DEFAULT_CONTEXTCOUNT, DEFAULT_TEMPERATURE, isMac } from '@renderer/config/constant'
 import { DEFAULT_MIN_APPS } from '@renderer/config/minapps'
 import {
+  DEFAULT_MODEL_MAP,
   glm45FlashModel,
   isFunctionCallingModel,
   isNotSupportedTextDelta,
@@ -1667,7 +1668,7 @@ const migrateConfig = {
         state.llm.translateModel.provider === 'silicon' &&
         state.llm.translateModel.id === 'meta-llama/Llama-3.3-70B-Instruct'
       ) {
-        state.llm.translateModel = SYSTEM_MODELS.defaultModel[2]
+        state.llm.translateModel = DEFAULT_MODEL_MAP.translate
       }
 
       // add selection_assistant_toggle and selection_assistant_select_text shortcuts after mini_window