File size: 2,327 Bytes
11f438b f09ddb7 11f438b 84ae379 11f438b 84ae379 f09ddb7 11f438b f09ddb7 1ca7ef3 6015c25 f09ddb7 84ae379 f09ddb7 11f438b f09ddb7 11f438b f09ddb7 84ae379 f09ddb7 11f438b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import requests
import time
from utils.config import config
def check_ollama_status():
"""
Checks if Ollama is running and which model is loaded.
Returns:
dict: {
"running": True/False,
"model_loaded": "mistral-7b" or None,
"ngrok_url": "https://f943b91f0a0c.ngrok-free.app/",
"local_url": "http://localhost:11434/"
}
"""
# Use the working ngrok URL from your logs
ngrok_url = config.ollama_host
local_url = "http://localhost:11434/" # Always check localhost as fallback
def _get_model_from_url(base_url, retries=3, delay=1):
"""Try to get model info with retry logic"""
for attempt in range(retries):
try:
# Add headers to skip ngrok browser warning
headers = {
"ngrok-skip-browser-warning": "true",
"User-Agent": "AI-Life-Coach-App"
}
response = requests.get(f"{base_url}/api/tags", timeout=10, headers=headers)
if response.status_code == 200:
models = response.json().get("models", [])
if models:
return models[0].get("name")
elif response.status_code == 404:
# Try alternative endpoint
response2 = requests.get(f"{base_url}", timeout=10, headers=headers)
if response2.status_code == 200:
# Server is running but might not have /api/tags endpoint
return "unknown-model"
except Exception as e:
if attempt < retries - 1: # Don't sleep on last attempt
time.sleep(delay * (2 ** attempt)) # Exponential backoff
continue
return None
# First try remote ngrok URL, then fall back to local
remote_model = _get_model_from_url(ngrok_url)
local_model = None
if not remote_model: # Only check local if remote failed
local_model = _get_model_from_url(local_url)
model_loaded = remote_model or local_model
running = bool(model_loaded)
return {
"running": running,
"model_loaded": model_loaded,
"ngrok_url": ngrok_url,
"local_url": local_url,
}
|