From 8a62b43a91a8089d9a84d29308c3f5dcc453dc3d Mon Sep 17 00:00:00 2001
From: Ilya Sibiryakov <ilsimox@gmail.com>
Date: Sun, 27 Sep 2026 09:26:17 +0300
Subject: [PATCH] test(ztds): add resilient fallback import and 7 comprehensive
 tests

---
 tests/guardrails_tests/test_ztds_guardrail.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/guardrails_tests/test_ztds_guardrail.py b/tests/guardrails_tests/test_ztds_guardrail.py
index be4b6a85da1..2db0507b265 100644
--- a/tests/guardrails_tests/test_ztds_guardrail.py
+++ b/tests/guardrails_tests/test_ztds_guardrail.py
@@ -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: