fix test_call_with_key_over_budget

2024-09-05 13:08:13 -07:00 · 2024-09-05 13:08:13 -07:00 · b590d807af
commit b590d807af
parent 1725260938
1 changed files with 4 additions and 1 deletions
--- a/litellm/tests/test_key_generate_prisma.py
+++ b/litellm/tests/test_key_generate_prisma.py
@ -1492,7 +1492,10 @@ def test_call_with_key_over_budget(prisma_client):
                proxy_logging_obj=proxy_logging_obj,
            )
            # test spend_log was written and we can read it
-            spend_logs = await view_spend_logs(request_id=request_id)
+            spend_logs = await view_spend_logs(
+                request_id=request_id,
+                user_api_key_dict=UserAPIKeyAuth(api_key=generated_key),
+            )

            print("read spend logs", spend_logs)
            assert len(spend_logs) == 1