From 9b239111f45874d1d3b6dd274b739c85a0cf180d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 17 Aug 2024 16:57:23 -0700 Subject: [PATCH] fix test update tpm / rpm limits for a key --- litellm/tests/test_key_generate_prisma.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 66d49f886df..6402964eea3 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -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")