From ef2cbc20accc7fb4da6158a9785cd77a80bf20dd Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 1 Apr 2026 08:49:31 -0700 Subject: [PATCH] fix: add missing user_api_key_team_budget_reset_at and user_api_key_user_budget_reset_at to PagerDutyInternalEvent initializations --- .../enterprise_callbacks/pagerduty/pagerduty.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/enterprise/litellm_enterprise/enterprise_callbacks/pagerduty/pagerduty.py b/enterprise/litellm_enterprise/enterprise_callbacks/pagerduty/pagerduty.py index de02a0c4dab..b4c47dea987 100644 --- a/enterprise/litellm_enterprise/enterprise_callbacks/pagerduty/pagerduty.py +++ b/enterprise/litellm_enterprise/enterprise_callbacks/pagerduty/pagerduty.py @@ -113,6 +113,12 @@ class PagerDutyAlerting(SlackAlerting): user_api_key_spend=_meta.get("user_api_key_spend"), user_api_key_max_budget=_meta.get("user_api_key_max_budget"), user_api_key_budget_reset_at=_meta.get("user_api_key_budget_reset_at"), + user_api_key_team_budget_reset_at=_meta.get( + "user_api_key_team_budget_reset_at" + ), + user_api_key_user_budget_reset_at=_meta.get( + "user_api_key_user_budget_reset_at" + ), user_api_key_org_id=_meta.get("user_api_key_org_id"), user_api_key_team_id=_meta.get("user_api_key_team_id"), user_api_key_project_id=_meta.get("user_api_key_project_id"), @@ -195,6 +201,8 @@ class PagerDutyAlerting(SlackAlerting): if user_api_key_dict.budget_reset_at else None ), + user_api_key_team_budget_reset_at=None, + user_api_key_user_budget_reset_at=None, user_api_key_org_id=user_api_key_dict.org_id, user_api_key_team_id=user_api_key_dict.team_id, user_api_key_project_id=user_api_key_dict.project_id,