From 6d91c6cd450bf88e01f5712a8d70495287931688 Mon Sep 17 00:00:00 2001
From: Anibal Angulo <anibal.angulo.cardoza@banorte.com>
Date: Sat, 27 Sep 2025 19:10:40 +0000
Subject: [PATCH] Add health endpoint

---
 pyproject.toml                       | 2 +-
 src/searchbox/mcp_server/__init__.py | 6 ++----
 src/searchbox/mcp_server/server.py   | 8 ++++++++
 uv.lock                              | 2 +-
 4 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index cb53168..2dd1046 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "searchbox"
-version = "0.1.0"
+version = "0.1.1"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
diff --git a/src/searchbox/mcp_server/__init__.py b/src/searchbox/mcp_server/__init__.py
index b1a6d2b..c1f7b80 100644
--- a/src/searchbox/mcp_server/__init__.py
+++ b/src/searchbox/mcp_server/__init__.py
@@ -17,12 +17,10 @@ Example:
 
 """
 
-from fastmcp.server.server import Transport
-
 from .server import mcp
 
 
-def run(transport: Transport = "sse"):  # pragma: no cover
+def run():  # pragma: no cover
     """Run the vector search MCP server with the specified transport.
 
     Args:
@@ -34,4 +32,4 @@ def run(transport: Transport = "sse"):  # pragma: no cover
         >>> run("stdio")  # Start with stdio transport
 
     """
-    mcp.run(transport=transport)
+    mcp.run(transport="sse", host="0.0.0.0", port=8000)
diff --git a/src/searchbox/mcp_server/server.py b/src/searchbox/mcp_server/server.py
index 44b6535..b58e3fa 100644
--- a/src/searchbox/mcp_server/server.py
+++ b/src/searchbox/mcp_server/server.py
@@ -19,6 +19,8 @@ Example:
 from typing import Annotated
 
 from fastmcp import FastMCP
+from starlette.requests import Request
+from starlette.responses import JSONResponse
 
 from ..engine import get_engine
 
@@ -49,3 +51,9 @@ async def get_information(
     )
 
     return result
+
+
+@mcp.custom_route("/health", methods=["GET"])
+async def health_check(_request: Request):
+    """Health check endpoint."""
+    return JSONResponse({"status": "ok", "service": "searchbox-mcp"})
diff --git a/uv.lock b/uv.lock
index bd2efdd..1eb08ee 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1527,7 +1527,7 @@ wheels = [
 
 [[package]]
 name = "searchbox"
-version = "0.1.0"
+version = "0.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "qdrant-client" },