Check if ollama is running

Author: Victor Goubet

README.md

@@ -55,10 +55,13 @@ configure-sage
 
 After setting up, launch the script with admin rights. If no configuration is provided, the default configuration will be used:
 
+
 ```sh
 launch-sage
 ```
 
+**Note**: Be sure to have **ollama running** if you intend to use local models
+
 ### Launch Options:
 
 - `--streamlit <true or false>`: If `true`, the Streamlit interface will be used; otherwise, a shell interface will appear.

poetry.lock

@@ -17,6 +17,7 @@ duckduckgo-search = "^6.1.6"
 httpcore = "^1.0.5"
 httpx = "^0.27.0"
 openai = "^1.34.0"
+psutil = "^6.0.0"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -6,6 +6,7 @@
 from typing import Optional
 
 import click
+import psutil
 
 from techsage.utils.constants import APP_FOLDER, DEFAULT_CONFIG
 
@@ -18,6 +19,23 @@ def create_app_folder() -> None:
     os.makedirs(f"{APP_FOLDER}/models", exist_ok=True)
 
 
+def check_ollama_running() -> None:
+    """
+    Check if 'ollama' is currently running.
+
+    :raises RuntimeError: If 'ollama' is not running.
+    """
+    try:
+        for process in psutil.process_iter(["name"]):
+            if process.info["name"].lower() == "ollama":
+                print(" ✅ Ollama is running.")
+                return
+        raise RuntimeError("Ollama is not running.")
+    except Exception as e:
+        print(f" ❌ {str(e)}")
+        sys.exit(1)
+
+
 def check_ollama_installed() -> None:
     """
     Check if 'ollama' is installed.
@@ -43,14 +61,15 @@ def pull_model(model_name: str) -> None:
     """
     try:
         output = None if VERBOSE else subprocess.DEVNULL
+        print(" ⏳ Pulling model...")
         subprocess.check_call(
             ["ollama", "pull", model_name],
             stdout=output,
             stderr=output,
         )
         print(" ✅ Base model is ready")
     except subprocess.CalledProcessError:
-        print(f" ❌ Failed to pull model {model_name}, check if it exists and if ollama is launched")
+        print(f" ❌ Failed to pull model {model_name}, check if it exists on ollama website")
         sys.exit(1)
 
 
@@ -175,6 +194,7 @@ def configure(
     create_app_folder()
     if local:
         check_ollama_installed()
+        check_ollama_running()
         model_config_file_path = create_model_file(model, f"{APP_FOLDER}/models/")
         pull_model(model)
         crewai_model_name = create_model(f"{model}_crewai", model_config_file_path)