feat(ingestion): add bearer token auth middleware for MCP endpoint

This commit is contained in:
Mathias Bergqvist
2026-05-07 20:58:16 +02:00
parent bee4bb3c1f
commit 228ee57d4c
3 changed files with 86 additions and 1 deletions

View File

@@ -57,6 +57,12 @@ func main() {
mcpSrv := mcp.NewServer(brainDir, &pipelineCfg, llmClient.Complete)
mcpToken := os.Getenv("BRAIN_MCP_TOKEN")
if mcpToken == "" {
logger.Error("BRAIN_MCP_TOKEN not set")
os.Exit(1)
}
ctx := context.Background()
if watchInterval > 0 {
watcher.Start(ctx, watcher.Config{
@@ -74,7 +80,7 @@ func main() {
mux.HandleFunc("POST /ingest-raw", h.IngestRaw)
mux.HandleFunc("POST /backfill-refs", h.BackfillRefs)
mux.HandleFunc("GET /pass-rate", h.PassRate)
mux.Handle("POST /mcp", mcpSrv)
mux.Handle("POST /mcp", mcp.BearerAuth(mcpToken, mcpSrv))
addr := ":" + port
watchIntervalLog := "disabled"