From 2cc807cec4b23040fec80d1aec07b2034adf8a39 Mon Sep 17 00:00:00 2001
From: moe-berri <moe@berri.ai>
Date: Fri, 6 Nov 2026 16:09:36 +0800
Subject: [PATCH] test(alerts): support budget regression on Python 3.12

---
 tests/unit/proxy/spend_tracking/test_daily_usage_alerts.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/proxy/spend_tracking/test_daily_usage_alerts.py b/tests/unit/proxy/spend_tracking/test_daily_usage_alerts.py
index d63a8dc6e1b..41fab29ef44 100644
--- a/tests/unit/proxy/spend_tracking/test_daily_usage_alerts.py
+++ b/tests/unit/proxy/spend_tracking/test_daily_usage_alerts.py
@@ -68,7 +68,7 @@ class _BudgetDatabase:
         self.lease_rollback_cancelled = False
 
     @asynccontextmanager
-    async def tx(self, *, max_wait: timedelta, timeout: timedelta) -> AsyncGenerator[AnalyticsTransaction]:
+    async def tx(self, *, max_wait: timedelta, timeout: timedelta) -> AsyncGenerator[AnalyticsTransaction, None]:
         transaction: Final = _BudgetTransaction(self, self.transactions)
         self.transactions += 1
         self.active += 1