feat: re-added is_up which was lost for some reason

DigitalProductInnovationAndDevelopment · Aug 14, 2024 · aa5c9c0 · aa5c9c0
1 parent 66eec00
commit aa5c9c0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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__(