tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 20:25:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_async_call_with_key_over_model_budget

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-11-15 09:26:29 -08:00
parent be234e2a47
commit 666913f76d
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/proxy_unit_tests/test_key_generate_prisma.py
+7 -1
View File
@@ -1770,7 +1770,7 @@ def test_call_with_key_over_budget_no_cache(prisma_client):
],
)
@pytest.mark.flaky(retries=3, delay=2)
async def test_call_with_key_over_model_budget(
async def test_async_call_with_key_over_model_budget(
prisma_client, request_model, should_pass
):
# 12. Make a call with a key over budget, expect to fail
@@ -1843,6 +1843,12 @@ async def test_call_with_key_over_model_budget(
},
)
# Flush the logging worker to ensure all callbacks complete
from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER
if GLOBAL_LOGGING_WORKER._queue is not None:
await GLOBAL_LOGGING_WORKER.flush()
# Wait for the budget callback to complete with polling
max_wait_time = 10 # Maximum 10 seconds
poll_interval = 0.5 # Check every 0.5 seconds