From fe37e38ae7b30a0bc4849e2df5ed81bbaddb0d97 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 16 Jul 2026 15:48:41 +0000 Subject: [PATCH] test(auth): seed cross-pod spend counter in soft budget alert regression --- tests/proxy_unit_tests/test_user_api_key_auth.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/proxy_unit_tests/test_user_api_key_auth.py b/tests/proxy_unit_tests/test_user_api_key_auth.py index 5471d2668e4..3485031164c 100644 --- a/tests/proxy_unit_tests/test_user_api_key_auth.py +++ b/tests/proxy_unit_tests/test_user_api_key_auth.py @@ -698,6 +698,12 @@ async def test_soft_budget_alert(): setattr(litellm.proxy.proxy_server, "master_key", "sk-1234") setattr(litellm.proxy.proxy_server, "prisma_client", AsyncMock()) + from litellm.proxy.proxy_server import spend_counter_cache + + spend_counter_cache.in_memory_cache.set_cache( + key=f"spend:key:{hash_token(user_key)}", value=float(current_spend) + ) + # Create request request = Request(scope={"type": "http"}) request._url = URL(url="/chat/completions")