move to a smaller validator

Author: dtam

server_ci/guard-template.json

@@ -9,12 +9,9 @@
             "name": "test-guard",
             "validators": [
                 {
-                    "id": "guardrails/detect_pii",
-                    "on": "msg_history",
-                    "onFail": "fix",
-                    "kwargs": {
-                        "pii_entities": ["PERSON","PHONE_NUMBER","LOCATION"]
-                    }
+                    "id": "guardrails/two_words",
+                    "on": "$",
+                    "onFail": "fix"
                 }
             ]
         }

server_ci/tests/test_server.py

@@ -14,7 +14,7 @@
     [
         (
             "France is wonderful in the spring",
-            "France is wonderful in the spring",
+            "France is",
             True,
             False,
         ),
@@ -37,7 +37,7 @@ def test_guard_validation(mock_llm_output, validation_output, validation_passed,
     [
         (
             "Tell me about Oranges in 5 words",
-            "Citrus fruit, sweet, nutritious, vibrant.",
+            "Citrus fruit",
             True,
             False,
         ),
@@ -69,10 +69,9 @@ def test_server_guard_llm_integration(
 @pytest.mark.parametrize(
     "message_content, output, validation_passed, error",
     [
-        ("Tell me about Paris in 5 words", "doesnt matter this errors", True, True),
         (
             "Write 5 words of prose.",
-            "Whispers of dawn kissed the horizon.",
+            "Whispers of",
             True,
             False,
         ),