Update server.py

server.py

@@ -1,11 +1,9 @@
-import os
 import random
 import string
+import uvicorn
 
-from fastmcp import FastMCP
+from mcp.server.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")
@@ -89,9 +87,6 @@ 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):
@@ -175,8 +170,4 @@ def file_ticket(user_request: str, user_profile: UserProfile):
 
 
 if __name__ == "__main__":
-    mcp.run(
-        transport="streamable-http",
-        host="0.0.0.0",
-        port=8000
-    )
+    uvicorn.run(mcp.streamable_http_app, host="0.0.0.0", port=8000)