From ef4c85522c001c930f02e2ec2c32dea9a7816b74 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 8 Apr 2024 21:55:35 -0700 Subject: [PATCH] test(test_llm_guard.py): fix test --- litellm/tests/test_llm_guard.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_llm_guard.py b/litellm/tests/test_llm_guard.py index 6b06da9af..97e8fd0ac 100644 --- a/litellm/tests/test_llm_guard.py +++ b/litellm/tests/test_llm_guard.py @@ -111,7 +111,10 @@ def test_llm_guard_key_specific_mode(): api_key=_api_key, ) - should_proceed = llm_guard.should_proceed(user_api_key_dict=user_api_key_dict) + request_data = {} + should_proceed = llm_guard.should_proceed( + user_api_key_dict=user_api_key_dict, data=request_data + ) assert should_proceed == False