From c7024baf7a48c7b32acf94217e2bcf4bc4b916a2 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 2 Feb 2024 12:18:36 -0800 Subject: [PATCH] test(test_key_generate_prisma.py): add sleep time --- litellm/tests/test_key_generate_prisma.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 831a6a6f72..9cddd1d6c9 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -941,7 +941,7 @@ def test_call_with_key_over_budget(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(4) # test spend_log was written and we can read it spend_logs = await view_spend_logs(request_id=request_id) @@ -965,6 +965,8 @@ def test_call_with_key_over_budget(prisma_client): asyncio.run(test()) except Exception as e: + # print(f"Error - {str(e)}") + traceback.print_exc() error_detail = e.message assert "Authentication Error, ExceededTokenBudget:" in error_detail print(vars(e))