From b7441f81cdf6775110cc050491fd4c1beb52e545 Mon Sep 17 00:00:00 2001
From: Daulet Amirkhanov <damirkhanov01@gmail.com>
Date: Fri, 26 Sep 2025 16:29:14 +0100
Subject: [PATCH] feat: add health check endpoint to MCP server

---
 cognee-mcp/src/server.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cognee-mcp/src/server.py b/cognee-mcp/src/server.py
index 9393fe71b..7670db9f4 100755
--- a/cognee-mcp/src/server.py
+++ b/cognee-mcp/src/server.py
@@ -19,6 +19,7 @@ from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline
 from cognee.modules.search.types import SearchType
 from cognee.shared.data_models import KnowledgeGraph
 from cognee.modules.storage.utils import JSONEncoder
+from starlette.responses import JSONResponse
 
 
 try:
@@ -37,6 +38,9 @@ mcp = FastMCP("Cognee")
 
 logger = get_logger()
 
+@mcp.custom_route("/health", methods=["GET"])
+async def health_check(request) -> dict:
+    return JSONResponse({"status": "ok"})
 
 @mcp.tool()
 async def cognee_add_developer_rules(