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

test(test_key_generate_dynamodb.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-05 21:44:27 -08:00
parent 3d29ec126b
commit 7a0bccf4d0
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_key_generate_dynamodb.py
View file

@ -490,8 +490,13 @@ def test_dynamo_db_migration(custom_db_client):
try: try:
async def test(): async def test():
request = GenerateKeyRequest(max_budget=1)
key = await generate_key_fn(request)
print(key)
generated_key = key.key
bearer_token = ( bearer_token = (
"Bearer " + "sk-elJDL2pOEjcAuC7zD4psAg" "Bearer " + generated_key
) # this works with ishaan's db, it's a never expiring key ) # this works with ishaan's db, it's a never expiring key
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})