From 86c4b204d1cba765d5f66c08c82124c433c41a2c Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 20 Oct 2026 19:23:43 -0700
Subject: [PATCH] test(retries): retry the budget success case with no backoff
 wait

---
 tests/unit/test_main.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py
index ac80ca59a92..3b5a59138fd 100644
--- a/tests/unit/test_main.py
+++ b/tests/unit/test_main.py
@@ -10880,11 +10880,15 @@ async def test_acompletion_retry_that_succeeds_leaves_the_budget_reservation_to_
 ) -> None:
     reservation: Final = {"reserved_cost": 0.5, "entries": [], "finalized": False, "callback_bound": False}
     retry_upstream.post(_ANTHROPIC.url).mock(
-        side_effect=[_provider_error(429, {"retry-after": "0"}), httpx.Response(200, json=_ANTHROPIC.ok_body)]
+        side_effect=[_provider_error(500), httpx.Response(200, json=_ANTHROPIC.ok_body)]
     )
 
     await _completion_with_retries_kwargs(
-        False, _ANTHROPIC, num_retries=1, metadata={"user_api_key_budget_reservation": reservation}
+        False,
+        _ANTHROPIC,
+        num_retries=1,
+        retry_strategy="constant_retry",
+        metadata={"user_api_key_budget_reservation": reservation},
     )
 
     assert reservation["callback_bound"] is True