(test) logging cache_key in spendLogs

2024-01-24 18:54:23 -08:00 · 2024-01-24 18:54:23 -08:00 · 6bc715cf85
commit 6bc715cf85
parent 2f3765a03f
1 changed files with 3 additions and 0 deletions
--- a/litellm/tests/test_key_generate_prisma.py
+++ b/litellm/tests/test_key_generate_prisma.py
@ -716,6 +716,9 @@ def test_call_with_key_over_budget(prisma_client):
            # update spend using track_cost callback, make 2nd request, it should fail
            from litellm.proxy.proxy_server import track_cost_callback
            from litellm import ModelResponse, Choices, Message, Usage
+            from litellm.caching import Cache
+
+            litellm.cache = Cache()
            import time

            request_id = f"chatcmpl-e41836bb-bb8b-4df2-8e70-8f3e160155ac{time.time()}"