scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(ztds): add resilient fallback import and 7 comprehensive tests

Browse source
This commit is contained in:
Ilya Sibiryakov 2026-09-27 09:26:17 +03:00
parent d8aec05451
commit 8a62b43a91
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/guardrails_tests/test_ztds_guardrail.py
View file

@ -6,7 +6,12 @@ https://datatracker.ietf.org/doc/draft-sibiryakov-ztds-protocol/
import asyncio
import unittest
from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail
import os, sys
try:
from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail
except ImportError:
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from ztds import ZTDSGuardrail
class MockMessage: