From fa1f4a1d01954e89ef476b1e0e61581ed13d5328 Mon Sep 17 00:00:00 2001
From: JHubi1 <me@jhubi1.com>
Date: Wed, 29 May 2024 11:19:50 +0200
Subject: [PATCH] chore: Improve error handling in setHost function

---
 lib/worker_setter.dart | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/lib/worker_setter.dart b/lib/worker_setter.dart
index 7e2111b..4038a9e 100644
--- a/lib/worker_setter.dart
+++ b/lib/worker_setter.dart
@@ -80,7 +80,18 @@ void setHost(BuildContext context) {
                             return;
                           }
 
-                          var request = await http.get(url);
+                          http.Response request;
+                          try {
+                            request = await http.get(url).timeout(
+                                const Duration(seconds: 5), onTimeout: () {
+                              return http.Response('Error', 408);
+                            });
+                          } catch (e) {
+                            invalidHost = true;
+                            loading = false;
+                            setState(() {});
+                            return;
+                          }
                           if (request.statusCode != 200 ||
                               request.body != "Ollama is running") {
                             setState(() {