fix name getting wiped

Mobile-Artificial-Intelligence · Aug 4, 2024 · b4cb278 · b4cb278
1 parent cbb06af
commit b4cb278
Show file tree

Hide file tree

Showing 8 changed files with 37 additions and 33 deletions.
diff --git a/lib/classes/providers/large_language_model.dart b/lib/classes/providers/large_language_model.dart
@@ -89,7 +89,6 @@ class LargeLanguageModel extends ChangeNotifier {
   }
 
   set uri(String value) {
-    _name = '';
     _uri = value;
     notifyListeners();
   }

diff --git a/lib/classes/providers/large_language_models/anthropic_claude_model.dart b/lib/classes/providers/large_language_models/anthropic_claude_model.dart
@@ -139,6 +139,7 @@ class ClaudeModel extends LargeLanguageModel {
   @override
   Future<void> resetUri() async {
     uri = defaultUrl;
+    name = '';
     notifyListeners();
   }
 

diff --git a/lib/classes/providers/large_language_models/google_gemini_model.dart b/lib/classes/providers/large_language_models/google_gemini_model.dart
@@ -114,6 +114,7 @@ class GoogleGeminiModel extends LargeLanguageModel {
   @override
   Future<void> resetUri() async {
     uri = ''; // No URI to reset
+    name = '';
     notifyListeners();
   }
 

diff --git a/lib/classes/providers/large_language_models/llama_cpp_model.dart b/lib/classes/providers/large_language_models/llama_cpp_model.dart
@@ -231,6 +231,7 @@ class LlamaCppModel extends LargeLanguageModel {
   @override
   Future<void> resetUri() async {
     uri = '';
+    name = '';
     notifyListeners();
   }
 

diff --git a/lib/classes/providers/large_language_models/mistral_ai_model.dart b/lib/classes/providers/large_language_models/mistral_ai_model.dart
@@ -136,6 +136,7 @@ class MistralAiModel extends LargeLanguageModel {
   @override
   Future<void> resetUri() async {
     uri = defaultUrl;
+    name = '';
     notifyListeners();
   }
 

diff --git a/lib/classes/providers/large_language_models/ollama_model.dart b/lib/classes/providers/large_language_models/ollama_model.dart
@@ -93,6 +93,8 @@ class OllamaModel extends LargeLanguageModel {
 
   @override
   Future<void> resetUri() async {
+    name = "";
+
     // Check Localhost
     if ((await _checkIpForOllama('127.0.0.1')).isNotEmpty) {
       uri = 'http://127.0.0.1:11434';

diff --git a/lib/classes/providers/large_language_models/open_ai_model.dart b/lib/classes/providers/large_language_models/open_ai_model.dart
@@ -163,6 +163,7 @@ class OpenAiModel extends LargeLanguageModel {
   @override
   Future<void> resetUri() async {
     uri = defaultUrl;
+    name = "";
     notifyListeners();
   }
 

diff --git a/lib/ui/shared/dialogs/load_model_dialog.dart b/lib/ui/shared/dialogs/load_model_dialog.dart
@@ -102,37 +102,6 @@ class _LoadModelDialogState extends State<LoadModelDialog> {
     );
   }
 
-  List<Widget> get importedOptions {
-    return [
-      FilledButton(
-        onPressed: () {
-          LlamaCppModel.of(context).name = selected!;
-          LlamaCppModel.of(context).uri = importedModels![selected!]!;
-          Navigator.of(context).pop();
-        },
-        child: Text(
-          "Select",
-          style: Theme.of(context).textTheme.labelLarge,
-        ),
-      ),
-      const SizedBox(width: 8),
-      FilledButton(
-        onPressed: () {
-          File(importedModels![selected!]!).delete();
-          setState(() {
-            importedModels!.remove(selected);
-            selected = null;
-          });
-          saveImportedModels();
-        },
-        child: Text(
-          "Delete",
-          style: Theme.of(context).textTheme.labelLarge,
-        ),
-      ),
-    ];
-  }
-
   Widget buildDropdown() {
     return Column(
       mainAxisSize: MainAxisSize.min,
@@ -159,7 +128,36 @@ class _LoadModelDialogState extends State<LoadModelDialog> {
         if (selected != null)
         Row(
           mainAxisAlignment: MainAxisAlignment.center,
-          children: importedOptions,
+          children: [
+            FilledButton(
+              onPressed: () {
+                print("Selected: $selected");
+                LlamaCppModel.of(context).name = selected!;
+                print("Selected: ${importedModels![selected!]}");
+                LlamaCppModel.of(context).uri = importedModels![selected!]!;
+                Navigator.of(context).pop();
+              },
+              child: Text(
+                "Select",
+                style: Theme.of(context).textTheme.labelLarge,
+              ),
+            ),
+            const SizedBox(width: 8),
+            FilledButton(
+              onPressed: () {
+                File(importedModels![selected!]!).delete();
+                setState(() {
+                  importedModels!.remove(selected);
+                  selected = null;
+                });
+                saveImportedModels();
+              },
+              child: Text(
+                "Delete",
+                style: Theme.of(context).textTheme.labelLarge,
+              ),
+            ),
+          ],
         ),
       ],
     );