phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_llm_guard.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-08 21:55:35 -07:00
parent 62ea989d42
commit ef4c85522c
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_llm_guard.py
View file

@ -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