feat: add 01-yi provider

src/renderer/src/config/models.ts

@@ -201,6 +201,56 @@ export const SYSTEM_MODELS: Record<string, SystemModel[]> = {
       defaultEnabled: true
     }
   ],
+  yi: [
+    {
+      id: 'yi-large',
+      provider: 'yi',
+      name: 'yi-large',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: false
+    },
+    {
+      id: 'yi-large-turbo',
+      provider: 'yi',
+      name: 'yi-large-turbo',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: true
+    },
+    {
+      id: 'yi-large-rag',
+      provider: 'yi',
+      name: 'yi-large-rag',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: false
+    },
+    {
+      id: 'yi-medium',
+      provider: 'yi',
+      name: 'yi-medium',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: true
+    },
+    {
+      id: 'yi-medium-200k',
+      provider: 'yi',
+      name: 'yi-medium-200k',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: false
+    },
+    {
+      id: 'yi-spark',
+      provider: 'yi',
+      name: 'yi-spark',
+      group: 'Yi',
+      temperature: 0.7,
+      defaultEnabled: false
+    }
+  ],
   groq: [
     {
       id: 'llama3-8b-8192',

src/renderer/src/store/index.ts

@@ -6,6 +6,7 @@ import assistants from './assistants'
 import settings from './settings'
 import llm from './llm'
 import runtime from './runtime'
+import migrate from './migrate'
 
 const rootReducer = combineReducers({
   assistants,
@@ -18,8 +19,9 @@ const persistedReducer = persistReducer(
   {
     key: 'cherry-studio',
     storage,
-    version: 1,
+    version: 2,
-    blacklist: ['runtime']
+    blacklist: ['runtime'],
+    migrate
   },
   rootReducer
 )

src/renderer/src/store/llm.ts

@@ -37,6 +37,14 @@ const initialState: LlmState = {
       isSystem: true,
       models: SYSTEM_MODELS.deepseek.filter((m) => m.defaultEnabled)
     },
+    {
+      id: 'yi',
+      name: 'Yi',
+      apiKey: '',
+      apiHost: 'https://api.lingyiwanwu.com',
+      isSystem: true,
+      models: SYSTEM_MODELS.yi.filter((m) => m.defaultEnabled)
+    },
     {
       id: 'groq',
       name: 'Groq',

src/renderer/src/store/migrate.ts

@@ -0,0 +1,28 @@
+import { createMigrate } from 'redux-persist'
+import { RootState } from '.'
+import { SYSTEM_MODELS } from '@renderer/config/models'
+
+const migrate = createMigrate({
+  // @ts-ignore store type is unknown
+  '2': (state: RootState) => {
+    return {
+      ...state,
+      llm: {
+        ...state.llm,
+        providers: [
+          ...state.llm.providers,
+          {
+            id: 'yi',
+            name: 'Yi',
+            apiKey: '',
+            apiHost: 'https://api.lingyiwanwu.com',
+            isSystem: true,
+            models: SYSTEM_MODELS.yi.filter((m) => m.defaultEnabled)
+          }
+        ]
+      }
+    }
+  }
+})
+
+export default migrate