phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: add sleep to key tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-02 12:36:35 -08:00
parent c7024baf7a
commit 46ef4dbcc1
2 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_key_generate_dynamodb.py
View file

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