* update version

* introduce new configs

output.py

@@ -66,11 +66,13 @@ def init_settings() -> (List, Dict):
 
             st_bs_match_type = st.selectbox("Match type", ["str", "word"])
             st_bs_case_sensitive = st.checkbox("Case sensitive", value=False)
+            st_bs_redact = st.checkbox("Redact", value=False)
 
         settings["BanSubstrings"] = {
             "substrings": st_bs_substrings,
             "match_type": st_bs_match_type,
             "case_sensitive": st_bs_case_sensitive,
+            "redact": st_bs_redact,
         }
 
     if "BanTopics" in st_enabled_scanners:
@@ -309,6 +311,7 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
             substrings=settings["substrings"],
             match_type=settings["match_type"],
             case_sensitive=settings["case_sensitive"],
+            redact=settings["redact"],
         )
 
     if scanner_name == "BanTopics":

prompt.py

@@ -10,6 +10,7 @@ from llm_guard.input_scanners import (
     BanTopics,
     Code,
     PromptInjection,
+    PromptInjectionV2,
     Secrets,
     Sentiment,
     TokenLimit,
@@ -28,6 +29,7 @@ def init_settings() -> (List, Dict):
         "BanTopics",
         "Code",
         "PromptInjection",
+        "PromptInjectionV2",
         "Secrets",
         "Sentiment",
         "TokenLimit",
@@ -110,11 +112,13 @@ def init_settings() -> (List, Dict):
 
             st_bs_match_type = st.selectbox("Match type", ["str", "word"])
             st_bs_case_sensitive = st.checkbox("Case sensitive", value=False)
+            st_bs_redact = st.checkbox("Redact", value=False)
 
         settings["BanSubstrings"] = {
             "substrings": st_bs_substrings,
             "match_type": st_bs_match_type,
             "case_sensitive": st_bs_case_sensitive,
+            "redact": st_bs_redact,
         }
 
     if "BanTopics" in st_enabled_scanners:
@@ -187,6 +191,26 @@ def init_settings() -> (List, Dict):
             "threshold": st_pi_threshold,
         }
 
+    if "PromptInjectionV2" in st_enabled_scanners:
+        st_piv2_expander = st.sidebar.expander(
+            "Prompt Injection V2",
+            expanded=False,
+        )
+
+        with st_piv2_expander:
+            st_piv2_threshold = st.slider(
+                label="Threshold",
+                value=0.5,
+                min_value=0.0,
+                max_value=1.0,
+                step=0.05,
+                key="prompt_injection_v2_threshold",
+            )
+
+        settings["PromptInjectionV2"] = {
+            "threshold": st_piv2_threshold,
+        }
+
     if "Secrets" in st_enabled_scanners:
         st_sec_expander = st.sidebar.expander(
             "Secrets",
@@ -284,6 +308,7 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
             substrings=settings["substrings"],
             match_type=settings["match_type"],
             case_sensitive=settings["case_sensitive"],
+            redact=settings["redact"],
         )
 
     if scanner_name == "BanTopics":
@@ -304,6 +329,9 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
     if scanner_name == "PromptInjection":
         return PromptInjection(threshold=settings["threshold"])
 
+    if scanner_name == "PromptInjectionV2":
+        return PromptInjectionV2(threshold=settings["threshold"])
+
     if scanner_name == "Secrets":
         return Secrets(redact_mode=settings["redact_mode"])
 

prompt_text.txt

@@ -1,15 +1,15 @@
-Given the personal profile of Johnathan Edward Doe:
+Given the personal profile of Johnathan Edward Doe, please summarize the following resume:
 
-Full Name: Johnathan Edward Doe
-Date of Birth: April 12, 1990
-Address: 123 Elm Street, Springfield, IL, 62701
-Email: [email protected]
-Phone Number: (123) 456-7890
-Educational Background:
-    Springfield High School, Graduated in 2008
-    Springfield University, B.Sc. Computer Science, Graduated in 2012
-Employment:
-    ABC Tech Solutions, Software Engineer, 2012-2015
-    XYZ Global Systems, Senior Software Developer, 2015-2021
+Profile:
 
-Please create a summarized version of his resume.
+- Full Name: Johnathan Edward Doe
+- Date of Birth: April 12, 1990
+- Address: 123 Elm Street, Springfield, IL, 62701
+- Email: [email protected]
+- Phone Number: (123) 456-7890
+- Educational Background:
+    - Springfield High School, Graduated in 2008
+    - Springfield University, B.Sc. Computer Science, Graduated in 2012
+- Employment:
+    - ABC Tech Solutions, Software Engineer, 2012-2015
+    - XYZ Global Systems, Senior Software Developer, 2015-2021

requirements.txt

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