From 7079b951deb84fcb33c88c00a0beb555a9b65a3c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 22 Jan 2024 21:12:03 -0800 Subject: [PATCH] test(test_key_generate_prisma.py): fix test --- litellm/tests/test_key_generate_prisma.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 7b19449508..e3ad408898 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -737,6 +737,7 @@ def test_call_with_key_over_budget(prisma_client): "user_api_key_user_id": user_id, } }, + "response_cost": 0.00002, }, completion_response=resp, start_time=datetime.now(), @@ -812,6 +813,7 @@ def test_call_with_key_over_budget_stream(prisma_client): "user_api_key_user_id": user_id, } }, + "response_cost": 0.00002, }, completion_response=ModelResponse(), start_time=datetime.now(),