scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix test update tpm / rpm limits for a key

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-17 16:57:23 -07:00
parent d9c91838ce
commit 9b239111f4
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_key_generate_prisma.py
View file

@ -2752,7 +2752,8 @@ async def test_generate_key_with_model_tpm_limit(prisma_client):
# Update model tpm_limit and rpm_limit
request = UpdateKeyRequest(
key=generated_key,
metadata={"model_tpm_limit": {"gpt-4": 200}, "model_rpm_limit": {"gpt-4": 3}},
model_tpm_limit={"gpt-4": 200},
model_rpm_limit={"gpt-4": 3},
)
_request = Request(scope={"type": "http"})
_request._url = URL(url="/update/key")