From 7fcf99ffaf2b2d931ca341fa90ea772e31131e9d Mon Sep 17 00:00:00 2001 From: Josh Date: Mon, 23 Mar 2026 19:07:01 -0400 Subject: [PATCH] Add org budget metrics to failure path --- litellm/integrations/prometheus.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index 2f3b86a596..1c37a65716 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -1441,6 +1441,9 @@ class PrometheusLogger(CustomLogger): user_api_team_alias = standard_logging_payload["metadata"][ "user_api_key_team_alias" ] + user_api_key_org_id = standard_logging_payload["metadata"].get( + "user_api_key_org_id" + ) try: self.litellm_llm_api_failed_requests_metric.labels( @@ -1456,6 +1459,10 @@ class PrometheusLogger(CustomLogger): ), ).inc() self.set_llm_deployment_failure_metrics(kwargs) + await self._set_org_budget_metrics_after_api_request( + org_id=user_api_key_org_id, + response_cost=0, + ) except Exception as e: verbose_logger.exception( "prometheus Layer Error(): Exception occured - {}".format(str(e))