From 1400ef053c43147a1e1fc2b800b22b3c358752ed Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Fri, 5 Jan 2024 13:50:17 +0700
Subject: [PATCH] fix: model migration stopped working (#1378)

---
 extensions/model-extension/src/index.ts | 28 +++++++------------------
 1 file changed, 8 insertions(+), 20 deletions(-)

diff --git a/extensions/model-extension/src/index.ts b/extensions/model-extension/src/index.ts
index b0fc995a1..90ebbe62b 100644
--- a/extensions/model-extension/src/index.ts
+++ b/extensions/model-extension/src/index.ts
@@ -50,17 +50,16 @@ export default class JanModelExtension implements ModelExtension {
 
   private async copyModelsToHomeDir() {
     try {
-      // list all of the files under the home directory
-
-      if (await fs.existsSync(JanModelExtension._homeDir)) {
-        // ignore if the model is already downloaded
+      
+      // Check for migration conditions
+      if (
+        localStorage.getItem(`${EXTENSION_NAME}-version`) === VERSION &&
+        (await fs.existsSync(JanModelExtension._homeDir))
+      ) {
+        // ignore if the there is no need to migrate
         console.debug('Models already persisted.')
         return
       }
-
-      // Get available models
-      const readyModels = (await this.getDownloadedModels()).map((e) => e.id)
-
       // copy models folder from resources to home directory
       const resourePath = await getResourcePath()
       const srcPath = await joinPath([resourePath, 'models'])
@@ -72,18 +71,7 @@ export default class JanModelExtension implements ModelExtension {
 
       console.debug('Finished syncing models')
 
-      const reconfigureModels = (await this.getConfiguredModels()).filter((e) =>
-        readyModels.includes(e.id)
-      )
-      console.debug('Finished updating downloaded models')
-
-      // update back the status
-      await Promise.all(
-        reconfigureModels.map(async (model) => this.saveModel(model))
-      )
-
       // Finished migration
-
       localStorage.setItem(`${EXTENSION_NAME}-version`, VERSION)
     } catch (err) {
       console.error(err)
@@ -203,7 +191,7 @@ export default class JanModelExtension implements ModelExtension {
   ): Promise<Model[]> {
     try {
       if (!(await fs.existsSync(JanModelExtension._homeDir))) {
-        console.error('Model folder not found')
+        console.debug('Model folder not found')
         return []
       }