From 635e125ef4183f3d0378efea9ee1dea09f460a88 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 12 Feb 2025 14:52:45 +0800
Subject: [PATCH] chore: update store migration and LLM providers

---
 src/renderer/src/store/index.ts   |  2 +-
 src/renderer/src/store/llm.ts     | 50 +++++++++++++------------------
 src/renderer/src/store/migrate.ts | 26 ++++++++++++++++
 3 files changed, 47 insertions(+), 31 deletions(-)

diff --git a/src/renderer/src/store/index.ts b/src/renderer/src/store/index.ts
index 76a2630be6..b0690fba25 100644
--- a/src/renderer/src/store/index.ts
+++ b/src/renderer/src/store/index.ts
@@ -30,7 +30,7 @@ const persistedReducer = persistReducer(
   {
     key: 'cherry-studio',
     storage,
-    version: 65,
+    version: 66,
     blacklist: ['runtime'],
     migrate
   },
diff --git a/src/renderer/src/store/llm.ts b/src/renderer/src/store/llm.ts
index c965188769..b84a4245f8 100644
--- a/src/renderer/src/store/llm.ts
+++ b/src/renderer/src/store/llm.ts
@@ -43,16 +43,6 @@ const initialState: LlmState = {
       isSystem: true,
       enabled: false
     },
-    {
-      id: 'gitee-ai',
-      name: 'gitee ai',
-      type: 'openai',
-      apiKey: '',
-      apiHost: 'https://ai.gitee.com',
-      models: SYSTEM_MODELS['gitee-ai'],
-      isSystem: true,
-      enabled: false
-    },
     {
       id: 'deepseek',
       name: 'deepseek',
@@ -63,16 +53,6 @@ const initialState: LlmState = {
       isSystem: true,
       enabled: false
     },
-    {
-      id: 'ppio',
-      name: 'PPIO',
-      type: 'openai',
-      apiKey: '',
-      apiHost: 'https://api.ppinfra.com/v3/openai',
-      models: SYSTEM_MODELS.ppio,
-      isSystem: true,
-      enabled: false
-    },
     {
       id: 'ocoolai',
       name: 'ocoolAI',
@@ -234,16 +214,6 @@ const initialState: LlmState = {
       isSystem: true,
       enabled: false
     },
-    {
-      id: 'graphrag-kylin-mountain',
-      name: 'GraphRAG',
-      type: 'openai',
-      apiKey: '',
-      apiHost: '',
-      models: [],
-      isSystem: true,
-      enabled: false
-    },
     {
       id: 'openrouter',
       name: 'OpenRouter',
@@ -353,6 +323,26 @@ const initialState: LlmState = {
       models: SYSTEM_MODELS.jina,
       isSystem: true,
       enabled: false
+    },
+    {
+      id: 'gitee-ai',
+      name: 'gitee ai',
+      type: 'openai',
+      apiKey: '',
+      apiHost: 'https://ai.gitee.com',
+      models: SYSTEM_MODELS['gitee-ai'],
+      isSystem: true,
+      enabled: false
+    },
+    {
+      id: 'ppio',
+      name: 'PPIO',
+      type: 'openai',
+      apiKey: '',
+      apiHost: 'https://api.ppinfra.com/v3/openai',
+      models: SYSTEM_MODELS.ppio,
+      isSystem: true,
+      enabled: false
     }
   ],
   settings: {
diff --git a/src/renderer/src/store/migrate.ts b/src/renderer/src/store/migrate.ts
index f646300596..b28cb053f0 100644
--- a/src/renderer/src/store/migrate.ts
+++ b/src/renderer/src/store/migrate.ts
@@ -925,6 +925,32 @@ const migrateConfig = {
   '65': (state: RootState) => {
     state.settings.targetLanguage = 'english'
     return state
+  },
+  '66': (state: RootState) => {
+    state.llm.providers.push(
+      {
+        id: 'gitee-ai',
+        name: 'gitee ai',
+        type: 'openai',
+        apiKey: '',
+        apiHost: 'https://ai.gitee.com',
+        models: SYSTEM_MODELS['gitee-ai'],
+        isSystem: true,
+        enabled: false
+      },
+      {
+        id: 'ppio',
+        name: 'PPIO',
+        type: 'openai',
+        apiKey: '',
+        apiHost: 'https://api.ppinfra.com/v3/openai',
+        models: SYSTEM_MODELS.ppio,
+        isSystem: true,
+        enabled: false
+      }
+    )
+    state.llm.providers = state.llm.providers.filter((provider) => provider.id !== 'graphrag-kylin-mountain')
+    return state
   }
 }