phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

_mount_metrics_endpoint

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-04 19:54:20 -07:00
parent 253060cb09
commit f16c531002
1 changed files with 0 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

44
litellm/integrations/prometheus.py
View file

@ -1742,19 +1742,6 @@ class PrometheusLogger(CustomLogger):
f"Prometheus metrics are only available for premium users. {CommonProxyErrors.not_premium_user.value}"
)
if PrometheusLogger._should_init_metrics_with_auth():
PrometheusLogger._mount_metrics_endpoint_with_auth()
else:
# Mount metrics directly without authentication
PrometheusLogger._mount_metrics_endpoint_without_auth()
@staticmethod
def _mount_metrics_endpoint_without_auth():
from prometheus_client import make_asgi_app
from litellm._logging import verbose_proxy_logger
from litellm.proxy.proxy_server import app
# Create metrics ASGI app
metrics_app = make_asgi_app()
@ -1764,37 +1751,6 @@ class PrometheusLogger(CustomLogger):
"Starting Prometheus Metrics on /metrics (no authentication)"
)
@staticmethod
def _mount_metrics_endpoint_with_auth():
from fastapi import APIRouter, Depends
from prometheus_client import make_asgi_app
from litellm._logging import verbose_proxy_logger
from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
from litellm.proxy.proxy_server import app
# Create metrics ASGI app
metrics_app = make_asgi_app()
# Create a router for authenticated metrics
metrics_router = APIRouter()
# Add metrics endpoint with authentication
@metrics_router.get("/metrics", dependencies=[Depends(user_api_key_auth)])
async def authenticated_metrics():
verbose_proxy_logger.debug("Serving authenticated metrics endpoint")
return metrics_app
# Mount the router to the app
app.include_router(metrics_router)
verbose_proxy_logger.debug(
"Starting Prometheus Metrics on /metrics with authentication"
)
@staticmethod
def _should_init_metrics_with_auth():
return litellm.require_auth_for_metrics_endpoint
def prometheus_label_factory(
supported_enum_labels: List[str],