PR comments, some test

Author: zsimjee

nemoguardrails/integrations/guardrails_ai/__init__.py

@@ -0,0 +1,63 @@
+# Define the main input check flow
+flow guardrails_ai check input
+  """Check input text using relevant Guardrails AI validators."""
+  # Run jailbreak and toxicity checks on input
+  $result = await validate_guardrails_ai_detect_jailbreak(text=$user_message)
+  if not $result.valid
+    if $system.config.enable_rails_exceptions
+      send GuardrailsAIException(message="Guardrails AI jailbreak validation failed")
+    else
+      bot refuse to respond
+    abort
+  
+  $result = await validate_guardrails_ai_toxic_language(text=$user_message)
+  if not $result.valid
+    if $system.config.enable_rails_exceptions
+      send GuardrailsAIException(message="Guardrails AI toxic language validation failed")
+    else
+      bot refuse to respond
+    abort
+  
+  # Add other relevant input checks, e.g., restrict to topic
+  # $result = await validate_guardrails_ai_restrict_to_topic(text=$user_message)
+  # if not $result.valid
+  #   if $system.config.enable_rails_exceptions
+  #     send GuardrailsAIException(message="Guardrails AI topic restriction validation failed")
+  #   else
+  #     bot refuse to respond
+  #   abort
+
+
+# Define the main output check flow
+flow guardrails_ai check output
+  """Check output text using relevant Guardrails AI validators."""
+  # Run PII, competitor, and provenance checks on output
+  $result = await validate_guardrails_ai_pii(text=$bot_message)
+  if not $result.valid
+    if $system.config.enable_rails_exceptions
+      send GuardrailsAIException(message="Guardrails AI PII validation failed")
+    else
+      bot refuse to respond
+    abort
+  
+  $result = await validate_guardrails_ai_competitor_check(text=$bot_message)
+  if not $result.valid
+    if $system.config.enable_rails_exceptions
+      send GuardrailsAIException(message="Guardrails AI competitor check validation failed")
+    else
+      bot refuse to respond
+    abort
+  
+  # Provenance might need different inputs/context
+  # $result = await validate_guardrails_ai_provenance_llm(text=$bot_message)
+  # if not $result.valid
+  #   if $system.config.enable_rails_exceptions
+  #     send GuardrailsAIException(message="Guardrails AI provenance validation failed")
+  #   else
+  #     bot refuse to respond
+  #   abort
+  # Add other relevant output checks
+
+
+flow bot refuse to respond
+  bot say "I apologize, but I cannot respond to that as it may contain inappropriate content or violate policy."