🗃️ fix: fix db migration

lobehub · Apr 10, 2024 · 4e75074 · 4e75074
1 parent ecd1d25
commit 4e75074
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 11 deletions.
diff --git a/src/database/core/db.ts b/src/database/core/db.ts
@@ -147,7 +147,7 @@ export class LocalDB extends Dexie {
 
   upgradeToV8 = async (trans: Transaction) => {
     const users = trans.table('users');
-    users.toCollection().modify((user: DB_User) => {
+    await users.toCollection().modify((user: DB_User) => {
       if (user.settings) {
         user.settings = MigrationLLMSettings.migrateSettings(user.settings as any);
       }

diff --git a/src/migrations/FromV3ToV4/index.ts b/src/migrations/FromV3ToV4/index.ts
@@ -33,17 +33,23 @@ export class MigrationV3ToV4 implements Migration {
       languageModel: {
         ...res,
         azure,
-        ollama: this.migrateProvider(ollama),
+        ollama: ollama && this.migrateProvider(ollama),
         openai,
-        openrouter: this.migrateProvider(openrouter),
-        togetherai: this.migrateProvider(togetherai),
+        openrouter: openrouter && this.migrateProvider(openrouter),
+        togetherai: togetherai && this.migrateProvider(togetherai),
       },
     };
   };
 
   static migrateOpenAI = (
-    openai: V3OpenAIConfig,
+    openai?: V3OpenAIConfig,
   ): { azure: V4AzureOpenAIConfig; openai: V4ProviderConfig } => {
+    if (!openai)
+      return {
+        azure: { apiKey: '', enabled: false },
+        openai: { apiKey: '', enabled: true },
+      };
+
     if (openai.useAzure) {
       return {
         azure: {

diff --git a/src/migrations/FromV3ToV4/types/v3.ts b/src/migrations/FromV3ToV4/types/v3.ts
@@ -44,7 +44,7 @@ export interface V3Settings {
   defaultAgent: any;
   fontSize: number;
   language: string;
-  languageModel: V3LLMConfig;
+  languageModel?: Partial<V3LLMConfig>;
   neutralColor?: string;
   password: string;
   primaryColor?: string;

diff --git a/src/migrations/FromV3ToV4/types/v4.ts b/src/migrations/FromV3ToV4/types/v4.ts
@@ -17,12 +17,12 @@ export interface V4AzureOpenAIConfig extends V4ProviderConfig {
 }
 
 export interface V4lLLMConfig
-  extends Omit<V3LLMConfig, 'ollama' | 'openAI' | 'openrouter' | 'togetherai'> {
-  azure: V4AzureOpenAIConfig;
-  ollama: V4ProviderConfig;
+  extends Omit<Partial<V3LLMConfig>, 'ollama' | 'openAI' | 'openrouter' | 'togetherai'> {
+  azure?: V4AzureOpenAIConfig;
+  ollama?: V4ProviderConfig;
   openai: V4ProviderConfig;
-  openrouter: V4ProviderConfig;
-  togetherai: V4ProviderConfig;
+  openrouter?: V4ProviderConfig;
+  togetherai?: V4ProviderConfig;
 }
 
 /**