From 501bf6961fd552f74d7dec2bd03559bdeb3f6fe7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 4 Nov 2024 12:01:46 -0800 Subject: [PATCH] fix test_call_with_key_over_model_budget --- tests/local_testing/test_key_generate_prisma.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/local_testing/test_key_generate_prisma.py b/tests/local_testing/test_key_generate_prisma.py index 0cb9659dc..fa9392136 100644 --- a/tests/local_testing/test_key_generate_prisma.py +++ b/tests/local_testing/test_key_generate_prisma.py @@ -1778,6 +1778,13 @@ def test_call_with_key_over_model_budget(prisma_client): "model": "chatgpt-v-2", "stream": False, "standard_logging_object": standard_logging_payload, + "litellm_params": { + "metadata": { + "user_api_key": hash_token(generated_key), + "user_api_key_user_id": user_id, + } + }, + "response_cost": 0.00002, }, completion_response=resp, start_time=datetime.now(),