mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-20 22:24:01 +00:00
allow adding auth on /metrics endpoint
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -10,5 +10,6 @@ model_list:
|
||||
api_key: fake-key
|
||||
|
||||
litellm_settings:
|
||||
require_auth_for_metrics_endpoint: true
|
||||
callbacks: ["prometheus"]
|
||||
service_callback: ["prometheus_system"]
|
||||
Reference in New Issue
Block a user