scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

test(guardrails): make callback cleanup defensive

Browse source 
Signed-off-by: xrwang8 <xrwang8@gmail.com>
This commit is contained in:
xrwang8 2026-09-16 16:59:59 +08:00
parent 9187ed52ce
commit 1b3cec0f53
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/guardrails/guardrail_hooks/content_filter/test_content_filter.py
View file

@ -115,7 +115,8 @@ class TestContentFilterGuardrail:
assert result["texts"] == ["Call <<PHONE_NUMBER>> about <<KEYWORD>>"]
finally:
litellm.callbacks.remove(guardrail)
if guardrail in litellm.callbacks:
litellm.callbacks.remove(guardrail)
def test_check_patterns_ssn(self):
"""