scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test: update test

Browse source
This commit is contained in:
Krrish Dholakia 2025-05-19 21:21:44 -07:00
parent 584660b3cd
commit 1857f5d264
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/proxy_unit_tests/test_key_generate_prisma.py
View file

@ -1323,10 +1323,10 @@ def test_generate_and_update_key(prisma_client):
print("response1=", response1)
# update the team id
# update the tpm limit
response2 = await update_key_fn(
request=Request,
data=UpdateKeyRequest(key=generated_key, team_id=_team_2),
data=UpdateKeyRequest(key=generated_key, tpm_limit=1000),
user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN,
api_key="sk-1234",
@ -1350,7 +1350,7 @@ def test_generate_and_update_key(prisma_client):
"project": "litellm-project3",
}
assert result["info"]["models"] == ["ada", "babbage", "curie", "davinci"]
assert result["info"]["team_id"] == _team_2
assert result["info"]["tpm_limit"] == 1000
assert result["info"]["budget_duration"] == "1mo"
assert result["info"]["max_budget"] == 100