From ada375353402b6143d92b8be290c706f389c7580 Mon Sep 17 00:00:00 2001
From: Daulet Amirkhanov <damirkhanov01@gmail.com>
Date: Tue, 7 Oct 2025 21:18:00 +0100
Subject: [PATCH] fix: fix broken check_llm_provider healthcheck method

---
 cognee/api/health.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/cognee/api/health.py b/cognee/api/health.py
index 1b1e45efa..14ebb8043 100644
--- a/cognee/api/health.py
+++ b/cognee/api/health.py
@@ -193,8 +193,17 @@ class HealthChecker:
 
             config = get_llm_config()
 
-            # Test actual API connection with minimal request
-            LLMGateway.show_prompt("test", "test.txt")
+            from pydantic import BaseModel
+
+            class TestResponse(BaseModel):
+                status: str
+
+            # Use a simple test to verify LLM connectivity
+            await LLMGateway.acreate_structured_output(
+                text_input="Respond with status: ok",
+                system_prompt="You are a test assistant. Respond with a simple status.",
+                response_model=TestResponse,
+            )
 
             response_time = int((time.time() - start_time) * 1000)
             return ComponentHealth(