allow adding auth on /metrics endpoint

This commit is contained in:
Ishaan Jaff 2025-04-04 20:37:17 -07:00
parent 96ce5dbf7d
commit 86b473d267
2 changed files with 6 additions and 1 deletions

View file

@ -2,6 +2,7 @@
Prometheus Auth Middleware
"""
from fastapi import Request
from fastapi.responses import JSONResponse
from starlette.middleware.base import BaseHTTPMiddleware
import litellm
@ -24,7 +25,10 @@ class PrometheusAuthMiddleware(BaseHTTPMiddleware):
or "",
)
except Exception as e:
raise e
return JSONResponse(
status_code=401,
content=f"Unauthorized access to metrics endpoint: {getattr(e, 'message', str(e))}",
)
# Process the request and get the response
response = await call_next(request)