check localhost

Mobile-Artificial-Intelligence · Apr 24, 2024 · acd299a · acd299a
1 parent 679df8c
commit acd299a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/classes/ollama_model.dart b/lib/classes/ollama_model.dart
@@ -74,6 +74,14 @@ class OllamaModel extends LargeLanguageModel {
 
   @override
   Future<void> resetUri() async {
+    // Check Localhost
+    if ((await _checkIpForOllama('127.0.0.1')).isNotEmpty) {
+      uri = 'http://127.0.0.1:11434';
+      notifyListeners();
+      return;
+    }
+
+    // If ip is set, check if it's valid
     if (_ip.isNotEmpty && (await _checkIpForOllama(_ip)).isNotEmpty) {
       uri = 'http://$_ip:11434';
       notifyListeners();