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
@@ -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)
+1
View File
@@ -10,5 +10,6 @@ model_list:
api_key: fake-key
litellm_settings:
require_auth_for_metrics_endpoint: true
callbacks: ["prometheus"]
service_callback: ["prometheus_system"]