gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add health check endpoint to MCP server

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-09-26 16:29:14 +01:00
parent 056da96995
commit b7441f81cd
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee-mcp/src/server.py
View file

@ -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.modules.search.types import SearchType
from cognee.shared.data_models import KnowledgeGraph from cognee.shared.data_models import KnowledgeGraph
from cognee.modules.storage.utils import JSONEncoder from cognee.modules.storage.utils import JSONEncoder
from starlette.responses import JSONResponse
try: try:
@ -37,6 +38,9 @@ mcp = FastMCP("Cognee")
logger = get_logger() logger = get_logger()
@mcp.custom_route("/health", methods=["GET"])
async def health_check(request) -> dict:
return JSONResponse({"status": "ok"})
@mcp.tool() @mcp.tool()
async def cognee_add_developer_rules( async def cognee_add_developer_rules(