fix input arg in flows

NVIDIA · Jan 20, 2025 · 25474a8 · 25474a8
1 parent 436d7b6
commit 25474a8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nemoguardrails/library/activefence/flows.co b/nemoguardrails/library/activefence/flows.co
@@ -13,9 +13,9 @@ Adult Content	General
 Privacy Violation	PII
 """
 
-define flow activefence moderation on input
+flow activefence moderation on input
   """Guardrail based on the maximum risk score."""
-  $result = await  CallActivefenceApiAction(text=$input_text)
+  $result = await  CallActivefenceApiAction(text=$user_message)
 
   if $result.max_risk_score > 0.7
     if $system.config.enable_rails_exceptions
@@ -26,7 +26,7 @@ define flow activefence moderation on input
 
 flow activefence moderation on output
   """Guardrail based on the maximum risk score."""
-  $result = await  CallActivefenceApiAction(text=$input_text)
+  $result = await  CallActivefenceApiAction(text=$bot_message)
 
   if $result.max_risk_score > 0.7
     if $system.config.enable_rails_exceptions
@@ -37,7 +37,7 @@ flow activefence moderation on output
 
 flow activefence moderation on input detailed
   """Guardrail based on individual risk scores."""
-  $result = await  CallActivefenceApiAction(text=$input_text)
+  $result = await  CallActivefenceApiAction(text=$user_message)
 
   if $result.violations.get("abusive_or_harmful.harassment_or_bullying", 0) > 0.8
     if $system.config.enable_rails_exceptions