Pouzor
e41dbe579c
- Replace BaseHTTPMiddleware with pure ASGI middleware in auth.py to fix the "Unexpected message: http.response.start" crash on SSE streams - Migrate from SseServerTransport to StreamableHTTPSessionManager in main.py - Add _slim_canvas() in tools.py to strip React Flow layout fields from get_canvas responses (60-80% payload reduction) - Update tests: assert canvas slimming, mock session_manager.handle_request in auth tests to avoid uninitialized task group errors
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI, Request
|
|
from mcp.server import Server
|
|
from mcp.server.streamable_http_manager import StreamableHTTPSessionManager
|
|
|
|
from .auth import ApiKeyMiddleware
|
|
from .backend_client import backend
|
|
from .resources import register_resources
|
|
from .tools import register_tools
|
|
|
|
|
|
mcp_server = Server("homelable")
|
|
register_resources(mcp_server)
|
|
register_tools(mcp_server)
|
|
|
|
session_manager = StreamableHTTPSessionManager(
|
|
app=mcp_server,
|
|
json_response=False,
|
|
stateless=True,
|
|
)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
await backend.start()
|
|
async with session_manager.run():
|
|
yield
|
|
await backend.stop()
|
|
|
|
|
|
app = FastAPI(title="Homelable MCP", lifespan=lifespan)
|
|
app.add_middleware(ApiKeyMiddleware)
|
|
|
|
|
|
@app.api_route("/mcp", methods=["GET", "POST", "DELETE"])
|
|
async def mcp_endpoint(request: Request):
|
|
await session_manager.handle_request(request.scope, request.receive, request._send)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|