File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
nemoguardrails/library/sensitive_data_detection Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -11,6 +11,7 @@ flow detect sensitive data on input
1111
1212flow mask sensitive data on input
1313 """Mask any sensitive data found in the user input."""
14+ global $user_message
1415 $user_message = await MaskSensitiveDataAction(source="input", text=$user_message)
1516
1617
@@ -28,10 +29,11 @@ flow detect sensitive data on output
2829
2930flow mask sensitive data on output
3031 """Mask any sensitive data found in the bot output."""
32+ global $bot_message
3133 $bot_message = await MaskSensitiveDataAction(source="output", text=$bot_message)
3234
3335
34- # RETRIVAL RAILS
36+ # RETRIEVAL RAILS
3537
3638
3739flow detect sensitive data on retrieval
@@ -45,4 +47,5 @@ flow detect sensitive data on retrieval
4547
4648flow mask sensitive data on retrieval
4749 """Mask any sensitive data found in the relevant chunks from the knowledge base."""
50+ global $relevant_chunks
4851 $relevant_chunks = await MaskSensitiveDataAction(source="retrieval", text=$relevant_chunks)
You can’t perform that action at this time.
0 commit comments