From 60586fc37f6d80a519d84b184ffca3ad52b09924 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 15 Jan 2024 21:11:50 -0800
Subject: [PATCH] (test) proxy rules

---
 litellm/tests/test_rules.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/tests/test_rules.py b/litellm/tests/test_rules.py
index 92ec20fe421..0c1b573b8e1 100644
--- a/litellm/tests/test_rules.py
+++ b/litellm/tests/test_rules.py
@@ -76,6 +76,7 @@ def test_post_call_rule():
         print("Got exception", e)
         print(type(e))
         print(vars(e))
+        assert e.message == "This violates LiteLLM Proxy Rules. Response too short"
         pass
     # print(f"MAKING ACOMPLETION CALL")
     # litellm.set_verbose = True