From aa5c9c09e4c4766d7ec0a52d5d7aec74929b1592 Mon Sep 17 00:00:00 2001
From: Ruben Kaiser <ruben@kaiser.fyi>
Date: Wed, 14 Aug 2024 09:57:35 +0200
Subject: [PATCH] feat: re-added is_up which was lost for some reason

---
 src/ai/LLM/Strategies/OLLAMAService.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/ai/LLM/Strategies/OLLAMAService.py b/src/ai/LLM/Strategies/OLLAMAService.py
index 4b3ae2b..d08816f 100644
--- a/src/ai/LLM/Strategies/OLLAMAService.py
+++ b/src/ai/LLM/Strategies/OLLAMAService.py
@@ -23,6 +23,15 @@
 
 logger = logging.getLogger(__name__)
 
+def is_up() -> bool:
+    # res = requests.post(os.getenv('OLLAMA_URL') + '/api/show', json={'name': os.getenv('OLLAMA_MODEL', 'llama3')})
+    res = httpx.post(
+        config.ollama_url + "/api/show",
+        json={"name": config.ollama_model},
+    )
+    if res.status_code == 200:
+        return True
+
 
 class OLLAMAService(BaseLLMService, LLMServiceMixin):
     def __init__(