[mob] Skip deleting downloaded models

ente-io · Nov 7, 2024 · 19ccc53 · 19ccc53
1 parent 27f55b1
commit 19ccc53
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/mobile/lib/services/machine_learning/ml_indexing_isolate.dart b/mobile/lib/services/machine_learning/ml_indexing_isolate.dart
@@ -193,21 +193,23 @@ class MLIndexingIsolate extends SuperIsolate {
     }
   }
 
-  Future<void> cleanupLocalIndexingModels() async {
+  Future<void> cleanupLocalIndexingModels({bool delete = false}) async {
     if (!areModelsDownloaded) return;
     await _releaseModels();
 
-    final List<String> remoteModelPaths = [];
+    if (delete) {
+      final List<String> remoteModelPaths = [];
 
-    for (final model in MLModels.values) {
-      if (!model.isIndexingModel) continue;
-      final mlModel = model.model;
-      remoteModelPaths.add(mlModel.modelRemotePath);
-    }
-
-    await RemoteAssetsService.instance.cleanupSelectedModels(remoteModelPaths);
+      for (final model in MLModels.values) {
+        if (!model.isIndexingModel) continue;
+        final mlModel = model.model;
+        remoteModelPaths.add(mlModel.modelRemotePath);
+      }
+      await RemoteAssetsService.instance
+          .cleanupSelectedModels(remoteModelPaths);
 
-    areModelsDownloaded = false;
+      areModelsDownloaded = false;
+    }
   }
 
   Future<void> _releaseModels() async {