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

test(test_key_generate_prisma.py): cleanup test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 20:29:37 -07:00
parent 59a48b2d0d
commit 77dbc1e723
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_key_generate_prisma.py
View file

@ -1408,13 +1408,13 @@ def test_call_with_key_over_budget(prisma_client):
assert spend_log.model == "chatgpt-v-2" assert spend_log.model == "chatgpt-v-2"
assert ( assert (
spend_log.cache_key spend_log.cache_key
== "a61ae14fe4a8b8014a61e6ae01a100c8bc6770ac37c293242afed954bc69207d" == "c891d64397a472e6deb31b87a5ac4d3ed5b2dcc069bc87e2afe91e6d64e95a1e"
) )
# use generated key to auth in # use generated key to auth in
result = await user_api_key_auth(request=request, api_key=bearer_token) result = await user_api_key_auth(request=request, api_key=bearer_token)
print("result from user auth with new key", result) print("result from user auth with new key", result)
pytest.fail(f"This should have failed!. They key crossed it's budget") pytest.fail("This should have failed!. They key crossed it's budget")
asyncio.run(test()) asyncio.run(test())
except Exception as e: except Exception as e: