From aed514741481603059ad703a18ef033dfea1a562 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 22 Jan 2024 21:44:13 -0800
Subject: [PATCH] test(test_key_generate_dynamodb.py): fix test

---
 litellm/tests/test_key_generate_dynamodb.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py
index 3c706b663b..4dab8a22db 100644
--- a/litellm/tests/test_key_generate_dynamodb.py
+++ b/litellm/tests/test_key_generate_dynamodb.py
@@ -452,6 +452,7 @@ def test_call_with_key_over_budget_stream(custom_db_client):
                             "user_api_key_user_id": user_id,
                         }
                     },
+                    "response_cost": 0.00002,
                 },
                 completion_response=ModelResponse(),
             )