From 24e46efa0cc165d4fb4e4f85fbfa9a57cd1defb3 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Sun, 13 Apr 2025 22:41:41 +0800
Subject: [PATCH] feat(migrate, websearch): enable enhanceMode in websearch and
 update migration logic

---
 .../src/pages/home/Inputbar/NewContextButton.tsx     |  4 ++--
 src/renderer/src/pages/home/Tabs/AssistantItem.tsx   | 12 +++++++++++-
 src/renderer/src/store/migrate.ts                    | 10 +++-------
 src/renderer/src/store/websearch.ts                  |  2 +-
 4 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/src/renderer/src/pages/home/Inputbar/NewContextButton.tsx b/src/renderer/src/pages/home/Inputbar/NewContextButton.tsx
index 0dcdbc2f..2cf0ba2d 100644
--- a/src/renderer/src/pages/home/Inputbar/NewContextButton.tsx
+++ b/src/renderer/src/pages/home/Inputbar/NewContextButton.tsx
@@ -1,6 +1,6 @@
 import { useShortcut, useShortcutDisplay } from '@renderer/hooks/useShortcuts'
 import { Tooltip } from 'antd'
-import { CircleFadingPlus } from 'lucide-react'
+import { Eraser } from 'lucide-react'
 import { FC } from 'react'
 import { useTranslation } from 'react-i18next'
 import styled from 'styled-components'
@@ -20,7 +20,7 @@ const NewContextButton: FC<Props> = ({ onNewContext, ToolbarButton }) => {
     <Container>
       <Tooltip placement="top" title={t('chat.input.new.context', { Command: newContextShortcut })} arrow>
         <ToolbarButton type="text" onClick={onNewContext}>
-          <CircleFadingPlus size={18} />
+          <Eraser size={18} />
         </ToolbarButton>
       </Tooltip>
     </Container>
diff --git a/src/renderer/src/pages/home/Tabs/AssistantItem.tsx b/src/renderer/src/pages/home/Tabs/AssistantItem.tsx
index 1af70ecc..79eb7755 100644
--- a/src/renderer/src/pages/home/Tabs/AssistantItem.tsx
+++ b/src/renderer/src/pages/home/Tabs/AssistantItem.tsx
@@ -172,7 +172,17 @@ const AssistantItem: FC<AssistantItemProps> = ({ assistant, isActive, onSwitch,
         }
       }
     ],
-    [addAgent, addAssistant, onSwitch, removeAllTopics, t, onDelete, sortByPinyinAsc, sortByPinyinDesc]
+    [
+      addAgent,
+      addAssistant,
+      onDelete,
+      onSwitch,
+      removeAllTopics,
+      setAssistantIconType,
+      sortByPinyinAsc,
+      sortByPinyinDesc,
+      t
+    ]
   )
 
   const handleSwitch = useCallback(async () => {
diff --git a/src/renderer/src/store/migrate.ts b/src/renderer/src/store/migrate.ts
index f8038bcb..c734299a 100644
--- a/src/renderer/src/store/migrate.ts
+++ b/src/renderer/src/store/migrate.ts
@@ -1220,14 +1220,10 @@ const migrateConfig = {
       state.settings.assistantIconType = state.settings?.showAssistantIcon ? 'model' : 'emoji'
       // @ts-ignore eslint-disable-next-line
       delete state.settings.showAssistantIcon
-      return state
-    } catch (error) {
-      return state
-    }
-  },
-  '97': (state: RootState) => {
-    try {
       state.settings.enableBackspaceDeleteModel = true
+      if (state.websearch) {
+        state.websearch.enhanceMode = true
+      }
       return state
     } catch (error) {
       return state
diff --git a/src/renderer/src/store/websearch.ts b/src/renderer/src/store/websearch.ts
index 3c7b2902..cd092023 100644
--- a/src/renderer/src/store/websearch.ts
+++ b/src/renderer/src/store/websearch.ts
@@ -64,7 +64,7 @@ const initialState: WebSearchState = {
   maxResults: 5,
   excludeDomains: [],
   subscribeSources: [],
-  enhanceMode: false,
+  enhanceMode: true,
   overwrite: false
 }