error handling

Mobile-Artificial-Intelligence · Jun 27, 2024 · 4dcccb2 · 4dcccb2
1 parent 4abf284
commit 4dcccb2
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/lib/classes/providers/large_language_models/llama_cpp_model.dart b/lib/classes/providers/large_language_models/llama_cpp_model.dart
@@ -20,6 +20,8 @@ class LlamaCppModel extends LargeLanguageModel {
 
   String _template = '';
 
+  bool _downloading = false;
+
   String get template => _template;
 
   set template(String value) {
@@ -38,6 +40,10 @@ class LlamaCppModel extends LargeLanguageModel {
     if (!File(uri).existsSync()) {
       missing.add('- The file provided does not exist.\n');
     }
+
+    if (_downloading) {
+      missing.add('- The model is currently downloading.\n');
+    }
 
     return missing;
   }
@@ -206,9 +212,19 @@ class LlamaCppModel extends LargeLanguageModel {
   }
 
   void setModelWithFuture(Future<(String, String)> future) async {
-    final (filePath, tag) = await future;
-    uri = filePath;
-    name = tag;
+    _downloading = true;
+    notifyListeners();
+
+    try {
+      final (filePath, tag) = await future;
+      uri = filePath;
+      name = tag;
+    } 
+    catch (e) {
+      Logger.log("Error setting model: $e");
+    }
+
+    _downloading = false;
     notifyListeners();
   }