From ad25d5a50f6dfdc010ee5f4761169a68cd5b7b29 Mon Sep 17 00:00:00 2001
From: jeffser <jeffrywalfa@gmail.com>
Date: Thu, 30 May 2024 19:15:19 -0600
Subject: [PATCH] Fixed: Ollama instance wasn't shutting down

---
 src/local_instance.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/local_instance.py b/src/local_instance.py
index 4a02586..46e8e78 100644
--- a/src/local_instance.py
+++ b/src/local_instance.py
@@ -7,14 +7,19 @@ port = 11435
 data_dir = os.getenv("XDG_DATA_HOME")
 
 def start():
+    global instance
     instance = subprocess.Popen(["/app/bin/ollama", "serve"], env={**os.environ, 'OLLAMA_HOST': f"127.0.0.1:{port}", "HOME": data_dir}, stderr=subprocess.PIPE, text=True)
     print("Starting Alpaca's Ollama instance...")
     sleep(1)
     print("Started Alpaca's Ollama instance")
 
 def stop():
-    if instance: instance.kill()
-    print("Stopped Alpaca's Ollama instance")
+    global instance
+    if instance:
+        instance.kill()
+        instance.wait()
+        instance = None
+        print("Stopped Alpaca's Ollama instance")
 
 def reset():
     stop()