add validation event

Signed-off-by: Jordan Dubrick <jdubrick@redhat.com>

Author: Jdubrick

src/app/endpoints/streaming_query.py

@@ -119,6 +119,7 @@
 LLM_TOKEN_EVENT = "token"
 LLM_TOOL_CALL_EVENT = "tool_call"
 LLM_TOOL_RESULT_EVENT = "tool_result"
+LLM_VALIDATION_EVENT = "validation"
 
 
 def format_stream_data(d: dict) -> str:
@@ -477,7 +478,7 @@ def _handle_shield_event(
                     "id": chunk_id,
                     "token": "No Violation",
                 },
-                event_type=LLM_TOKEN_EVENT,
+                event_type=LLM_VALIDATION_EVENT,
                 media_type=media_type,
             )
         else:
@@ -491,7 +492,7 @@ def _handle_shield_event(
                     "id": chunk_id,
                     "token": violation,
                 },
-                event_type=LLM_TOKEN_EVENT,
+                event_type=LLM_VALIDATION_EVENT,
                 media_type=media_type,
             )
 

tests/unit/app/endpoints/test_streaming_query.py

@@ -923,7 +923,7 @@ def test_stream_build_event_shield_call_step_complete_no_violation(mocker):
 
     assert result is not None
     assert "data: " in result
-    assert '"event": "token"' in result
+    assert '"event": "validation"' in result
     assert '"token": "No Violation"' in result
     # Role field removed for OLS compatibility
     assert '"id": 0' in result
@@ -963,7 +963,7 @@ def test_stream_build_event_shield_call_step_complete_with_violation(mocker):
 
     assert result is not None
     assert "data: " in result
-    assert '"event": "token"' in result
+    assert '"event": "validation"' in result
     assert (
         '"token": "Violation: I don\'t like the cut of your jib (Metadata: {})"'
         in result