* update version

app.py

@@ -41,6 +41,10 @@ Scanning prompt and output using [LLM Guard](https://laiyer-ai.github.io/llm-gua
 
 scanner_type = st.sidebar.selectbox("Type", [PROMPT, OUTPUT], index=0)
 
+st_fail_fast = st.sidebar.checkbox(
+    "Fail fast", value=False, help="Stop scanning after first failure"
+)
+
 enabled_scanners = None
 settings = None
 if scanner_type == PROMPT:
@@ -105,11 +109,11 @@ try:
             start_time = time.monotonic()
             if scanner_type == PROMPT:
                 st_result_text, results_valid, results_score = scan_prompt(
-                    vault, enabled_scanners, settings, st_prompt_text
+                    vault, enabled_scanners, settings, st_prompt_text, st_fail_fast
                 )
             elif scanner_type == OUTPUT:
                 st_result_text, results_valid, results_score = scan_output(
-                    vault, enabled_scanners, settings, st_prompt_text, st_output_text
+                    vault, enabled_scanners, settings, st_prompt_text, st_output_text, st_fail_fast
                 )
             end_time = time.monotonic()
             st_time_delta = timedelta(seconds=end_time - start_time)

output.py

@@ -371,13 +371,22 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
 
 
 def scan(
-    vault: Vault, enabled_scanners: List[str], settings: Dict, prompt: str, text: str
+    vault: Vault,
+    enabled_scanners: List[str],
+    settings: Dict,
+    prompt: str,
+    text: str,
+    fail_fast: bool = False,
 ) -> (str, Dict[str, bool], Dict[str, float]):
     sanitized_output = text
     results_valid = {}
     results_score = {}
 
-    with st.status("Scanning output...", expanded=True) as status:
+    status_text = "Scanning prompt..."
+    if fail_fast:
+        status_text = "Scanning prompt (fail fast mode)..."
+
+    with st.status(status_text, expanded=True) as status:
         for scanner_name in enabled_scanners:
             st.write(f"{scanner_name} scanner...")
             scanner = get_scanner(
@@ -386,6 +395,10 @@ def scan(
             sanitized_output, is_valid, risk_score = scanner.scan(prompt, sanitized_output)
             results_valid[scanner_name] = is_valid
             results_score[scanner_name] = risk_score
+
+            if fail_fast and not is_valid:
+                break
+
         status.update(label="Scanning complete", state="complete", expanded=False)
 
     return sanitized_output, results_valid, results_score

prompt.py

@@ -320,19 +320,26 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
 
 
 def scan(
-    vault: Vault, enabled_scanners: List[str], settings: Dict, text: str
+    vault: Vault, enabled_scanners: List[str], settings: Dict, text: str, fail_fast: bool = False
 ) -> (str, Dict[str, bool], Dict[str, float]):
     sanitized_prompt = text
     results_valid = {}
     results_score = {}
 
-    with st.status("Scanning prompt...", expanded=True) as status:
+    status_text = "Scanning prompt..."
+    if fail_fast:
+        status_text = "Scanning prompt (fail fast mode)..."
+
+    with st.status(status_text, expanded=True) as status:
         for scanner_name in enabled_scanners:
             st.write(f"{scanner_name} scanner...")
             scanner = get_scanner(scanner_name, vault, settings[scanner_name])
             sanitized_prompt, is_valid, risk_score = scanner.scan(sanitized_prompt)
             results_valid[scanner_name] = is_valid
             results_score[scanner_name] = risk_score
+
+            if fail_fast and not is_valid:
+                break
         status.update(label="Scanning complete", state="complete", expanded=False)
 
     return sanitized_prompt, results_valid, results_score

requirements.txt

@@ -1,4 +1,4 @@
-llm-guard==0.2.2
+llm-guard==0.2.3
 pandas==2.1.0
 streamlit==1.26.0
 streamlit-tags==1.2.8