diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py index 9963ba2c0..61d0ff6a6 100644 --- a/litellm/tests/test_key_generate_dynamodb.py +++ b/litellm/tests/test_key_generate_dynamodb.py @@ -246,7 +246,7 @@ def test_call_with_user_over_budget(custom_db_client): }, completion_response=resp, ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -321,7 +321,7 @@ def test_call_with_user_over_budget_stream(custom_db_client): }, completion_response=ModelResponse(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -394,7 +394,7 @@ def test_call_with_user_key_budget(custom_db_client): }, completion_response=resp, ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -469,7 +469,7 @@ def test_call_with_key_over_budget_stream(custom_db_client): }, completion_response=ModelResponse(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 9cddd1d6c..19f4e008d 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -277,7 +277,7 @@ def test_call_with_user_over_budget(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -360,7 +360,7 @@ def test_call_with_proxy_over_budget(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -441,7 +441,7 @@ def test_call_with_user_over_budget_stream(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -530,7 +530,7 @@ def test_call_with_proxy_over_budget_stream(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -1036,7 +1036,7 @@ async def test_call_with_key_never_over_budget(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result) @@ -1112,7 +1112,7 @@ async def test_call_with_key_over_budget_stream(prisma_client): start_time=datetime.now(), end_time=datetime.now(), ) - + await asyncio.sleep(5) # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) print("result from user auth with new key", result)