diff --git a/litellm/tests/test_prometheus.py b/litellm/tests/test_prometheus.py index 74c48117b2d..64e824e6db0 100644 --- a/litellm/tests/test_prometheus.py +++ b/litellm/tests/test_prometheus.py @@ -78,3 +78,4 @@ async def test_async_prometheus_success_logging(): assert metrics["litellm_total_tokens_total"] == 30.0 assert metrics["llm_deployment_success_responses_total"] == 1.0 assert metrics["llm_deployment_total_requests_total"] == 1.0 + assert metrics["llm_deployment_latency_per_output_token_bucket"] == 1.0