scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(auth): seed cross-pod spend counter in soft budget alert regression

Browse source
This commit is contained in:
Devin AI 2026-07-16 15:48:41 +00:00
parent a783e9ca2a
commit fe37e38ae7
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/proxy_unit_tests/test_user_api_key_auth.py
View file

@ -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")