Update server.py

server.py

@@ -4,6 +4,8 @@ import string
 
 from fastmcp import FastMCP
 from pydantic import BaseModel
+from starlette.requests import Request
+from starlette.responses import PlainTextResponse
 
 # Create an MCP server
 mcp = FastMCP("Airline Agent")
@@ -87,6 +89,9 @@ flight_database = {
 itinery_database = {}
 ticket_database = {}
 
+@mcp.custom_route("/health", methods=["GET"])
+async def health_check(request: Request) -> PlainTextResponse:
+    return PlainTextResponse("OK")
 
 @mcp.tool()
 def fetch_flight_info(date: Date, origin: str, destination: str):