From e6003463ac240dce491c1e6f230f312c4d65926c Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Mon, 8 Dec 2025 13:02:20 +0800
Subject: [PATCH] feat: enhance update dialog functionality and state
 management

- Added ability to ignore updates in the UpdateDialogPopup, updating the state accordingly.
- Updated UpdateAppButton to conditionally render based on the ignore state.
- Refactored runtime state to include an ignore flag for better update management.
- Minor UI adjustments in UpdateAppButton for improved user experience.
---
 src/main/services/agents/BaseService.ts               |  2 +-
 .../src/components/Popups/UpdateDialogPopup.tsx       | 11 +++++++++--
 .../src/pages/home/components/UpdateAppButton.tsx     |  6 +++++-
 src/renderer/src/services/AssistantService.ts         |  3 +--
 src/renderer/src/store/runtime.ts                     |  4 +++-
 5 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/src/main/services/agents/BaseService.ts b/src/main/services/agents/BaseService.ts
index 78bf72a952..461fdab96d 100644
--- a/src/main/services/agents/BaseService.ts
+++ b/src/main/services/agents/BaseService.ts
@@ -78,7 +78,7 @@ export abstract class BaseService {
    * Get database instance
    * Automatically waits for initialization to complete
    */
-  protected async getDatabase() {
+  public async getDatabase() {
     const dbManager = await DatabaseManager.getInstance()
     return dbManager.getDatabase()
   }
diff --git a/src/renderer/src/components/Popups/UpdateDialogPopup.tsx b/src/renderer/src/components/Popups/UpdateDialogPopup.tsx
index 29afcc0d24..593c882bf5 100644
--- a/src/renderer/src/components/Popups/UpdateDialogPopup.tsx
+++ b/src/renderer/src/components/Popups/UpdateDialogPopup.tsx
@@ -1,6 +1,7 @@
 import { loggerService } from '@logger'
 import { TopView } from '@renderer/components/TopView'
-import { handleSaveData } from '@renderer/store'
+import { handleSaveData, useAppDispatch } from '@renderer/store'
+import { setUpdateState } from '@renderer/store/runtime'
 import { Button, Modal } from 'antd'
 import type { ReleaseNoteInfo, UpdateInfo } from 'builder-util-runtime'
 import { useEffect, useState } from 'react'
@@ -22,6 +23,7 @@ const PopupContainer: React.FC<Props> = ({ releaseInfo, resolve }) => {
   const { t } = useTranslation()
   const [open, setOpen] = useState(true)
   const [isInstalling, setIsInstalling] = useState(false)
+  const dispatch = useAppDispatch()
 
   useEffect(() => {
     if (releaseInfo) {
@@ -50,6 +52,11 @@ const PopupContainer: React.FC<Props> = ({ releaseInfo, resolve }) => {
     resolve({})
   }
 
+  const onIgnore = () => {
+    dispatch(setUpdateState({ ignore: true }))
+    setOpen(false)
+  }
+
   UpdateDialogPopup.hide = onCancel
 
   const releaseNotes = releaseInfo?.releaseNotes
@@ -69,7 +76,7 @@ const PopupContainer: React.FC<Props> = ({ releaseInfo, resolve }) => {
       centered
       width={720}
       footer={[
-        <Button key="later" onClick={onCancel} disabled={isInstalling}>
+        <Button key="later" onClick={onIgnore} disabled={isInstalling}>
           {t('update.later')}
         </Button>,
         <Button key="install" type="primary" onClick={handleInstall} loading={isInstalling}>
diff --git a/src/renderer/src/pages/home/components/UpdateAppButton.tsx b/src/renderer/src/pages/home/components/UpdateAppButton.tsx
index 72c46c9fa8..b83dcea068 100644
--- a/src/renderer/src/pages/home/components/UpdateAppButton.tsx
+++ b/src/renderer/src/pages/home/components/UpdateAppButton.tsx
@@ -20,6 +20,10 @@ const UpdateAppButton: FC = () => {
     return null
   }
 
+  if (update.ignore) {
+    return null
+  }
+
   const handleOpenUpdateDialog = () => {
     UpdateDialogPopup.show({ releaseInfo: update.info || null })
   }
@@ -30,7 +34,7 @@ const UpdateAppButton: FC = () => {
         className="nodrag"
         onClick={handleOpenUpdateDialog}
         icon={<SyncOutlined />}
-        color="orange"
+        color="primary"
         variant="outlined"
         size="small">
         {t('button.update_available')}
diff --git a/src/renderer/src/services/AssistantService.ts b/src/renderer/src/services/AssistantService.ts
index 589fd7b6c8..5ae3710e87 100644
--- a/src/renderer/src/services/AssistantService.ts
+++ b/src/renderer/src/services/AssistantService.ts
@@ -7,7 +7,6 @@ import {
   UNLIMITED_CONTEXT_COUNT
 } from '@renderer/config/constant'
 import { isQwenMTModel } from '@renderer/config/models/qwen'
-import { CHERRYAI_PROVIDER } from '@renderer/config/providers'
 import { UNKNOWN } from '@renderer/config/translate'
 import { getStoreProviders } from '@renderer/hooks/useStore'
 import i18n from '@renderer/i18n'
@@ -142,7 +141,7 @@ export function getProviderByModel(model?: Model): Provider {
 
   if (!provider) {
     const defaultProvider = providers.find((p) => p.id === getDefaultModel()?.provider)
-    return defaultProvider || CHERRYAI_PROVIDER || providers[0]
+    return defaultProvider || providers[0]
   }
 
   return provider
diff --git a/src/renderer/src/store/runtime.ts b/src/renderer/src/store/runtime.ts
index fcb87905f3..2ee7719469 100644
--- a/src/renderer/src/store/runtime.ts
+++ b/src/renderer/src/store/runtime.ts
@@ -34,6 +34,7 @@ export interface UpdateState {
   downloaded: boolean
   downloadProgress: number
   available: boolean
+  ignore: boolean
 }
 
 export interface RuntimeState {
@@ -79,7 +80,8 @@ const initialState: RuntimeState = {
     downloading: false,
     downloaded: false,
     downloadProgress: 0,
-    available: false
+    available: false,
+    ignore: false
   },
   export: {
     isExporting: false