feat: add Bearer token authentication support for MCP

This commit is contained in:
sha
2026-03-20 19:05:44 +02:00
parent f36d779346
commit b73fa63c0e
6 changed files with 57 additions and 4 deletions
+16 -3
View File
@@ -8,9 +8,11 @@ Environment variables:
MCP_TRANSPORT — "stdio" (default) or "http"
MCP_HOST — bind host for HTTP transport (default: 0.0.0.0)
MCP_PORT — bind port for HTTP transport (default: 8000)
MCP_AUTH_TOKEN — Bearer token for HTTP transport auth (optional)
"""
import os
import uvicorn
from app.config import validate
from app import mcp # imports entities as a side effect via __init__
@@ -18,8 +20,19 @@ if __name__ == "__main__":
validate()
transport = os.environ.get("MCP_TRANSPORT", "stdio")
if transport == "http":
mcp.settings.host = os.environ.get("MCP_HOST", "0.0.0.0")
mcp.settings.port = int(os.environ.get("MCP_PORT", "8000"))
mcp.run(transport="streamable-http")
host = os.environ.get("MCP_HOST", "0.0.0.0")
port = int(os.environ.get("MCP_PORT", "8000"))
token = os.environ.get("MCP_AUTH_TOKEN", "")
app = mcp.streamable_http_app()
if token:
from app.auth import BearerAuthMiddleware
app = BearerAuthMiddleware(app, token)
print("Auth: Bearer token enabled.", flush=True)
else:
print("Auth: disabled (MCP_AUTH_TOKEN not set).", flush=True)
uvicorn.run(app, host=host, port=port)
else:
mcp.run(transport="stdio")