From 8b3c88b6a6b4a5aba3c0ce3ca2ec9a008d7670e2 Mon Sep 17 00:00:00 2001 From: abhijitpal1247 Date: Wed, 6 Mar 2024 10:04:32 +0530 Subject: [PATCH 1/3] resolves duplication issue in guardrail response --- nemoguardrails/library/autoguard/actions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nemoguardrails/library/autoguard/actions.py b/nemoguardrails/library/autoguard/actions.py index 616fca14a..e8da9add9 100644 --- a/nemoguardrails/library/autoguard/actions.py +++ b/nemoguardrails/library/autoguard/actions.py @@ -82,12 +82,12 @@ def process_autoguard_output(responses: List[Any]): """Processes the output provided AutoGuard API""" response_dict = {"guardrails_triggered": False} - prefixes = [] + prefixes = set() for response in responses: if response["guarded"]: if response["task"] == "text_toxicity_extraction": response_dict["guardrails_triggered"] = True - prefixes += [GUARDRAIL_RESPONSE_TEXT[response["task"]]] + prefixes.add(GUARDRAIL_RESPONSE_TEXT[response["task"]]) suffix = " Toxic phrases: " + ", ".join(response["output_data"]) response_dict[response["task"]] = { "guarded": True, @@ -100,7 +100,7 @@ def process_autoguard_output(responses: List[Any]): "response": response["response"][start_index:], } else: - prefixes += [GUARDRAIL_RESPONSE_TEXT[response["task"]]] + prefixes.add(GUARDRAIL_RESPONSE_TEXT[response["task"]]) response_dict["guardrails_triggered"] = True response_dict[response["task"]] = { "guarded": True, From 064d3f0e51790c936fabf88dbf00e1ba83650d7b Mon Sep 17 00:00:00 2001 From: abhijitpal1247 Date: Wed, 6 Mar 2024 10:13:15 +0530 Subject: [PATCH 2/3] some small change in flows.co --- examples/configs/autoguard/autoguard_config/flows.co | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/examples/configs/autoguard/autoguard_config/flows.co b/examples/configs/autoguard/autoguard_config/flows.co index 4bc3c953e..a71a65d01 100644 --- a/examples/configs/autoguard/autoguard_config/flows.co +++ b/examples/configs/autoguard/autoguard_config/flows.co @@ -14,16 +14,18 @@ define flow call autoguard output $pii_response_input = $input_result['pii_fast']['response'] bot respond pii input - - if $output_result["pii_fast"]["guarded"] - $pii_response_output = $output_result['pii_fast']['response'] - bot respond pii output - if $output_result["guardrails_triggered"] + if $output_result["pii_fast"]["guarded"] + $pii_response_output = $output_result['pii_fast']['response'] + bot respond pii output $autoguard_output_response = $output_result['combined_response'] bot refuse to respond autoguard output stop else + if $output_result["pii_fast"]["guarded"] + $pii_response_output = $output_result['pii_fast']['response'] + bot respond pii output + stop bot respond to question From 643096f93c15c2b91c19d696c3b3174983889f94 Mon Sep 17 00:00:00 2001 From: abhijitpal1247 Date: Wed, 6 Mar 2024 20:09:58 +0530 Subject: [PATCH 3/3] some small change in flows.co - 2 --- examples/configs/autoguard/autoguard_config/flows.co | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/configs/autoguard/autoguard_config/flows.co b/examples/configs/autoguard/autoguard_config/flows.co index a71a65d01..b07757695 100644 --- a/examples/configs/autoguard/autoguard_config/flows.co +++ b/examples/configs/autoguard/autoguard_config/flows.co @@ -15,9 +15,6 @@ define flow call autoguard output bot respond pii input if $output_result["guardrails_triggered"] - if $output_result["pii_fast"]["guarded"] - $pii_response_output = $output_result['pii_fast']['response'] - bot respond pii output $autoguard_output_response = $output_result['combined_response'] bot refuse to respond autoguard output stop