* upgrade of the version
Browse files- output.py +25 -2
- prompt.py +23 -1
- requirements.txt +4 -4
output.py
CHANGED
@@ -20,6 +20,7 @@ logger = logging.getLogger("llm-guard-playground")
|
|
20 |
|
21 |
def init_settings() -> (List, Dict):
|
22 |
all_scanners = [
|
|
|
23 |
"BanCompetitors",
|
24 |
"BanSubstrings",
|
25 |
"BanTopics",
|
@@ -31,7 +32,7 @@ def init_settings() -> (List, Dict):
|
|
31 |
"LanguageSame",
|
32 |
"MaliciousURLs",
|
33 |
"NoRefusal",
|
34 |
-
"ReadingTime",
|
35 |
"FactualConsistency",
|
36 |
"Gibberish",
|
37 |
"Regex",
|
@@ -51,6 +52,24 @@ def init_settings() -> (List, Dict):
|
|
51 |
|
52 |
settings = {}
|
53 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
if "BanCompetitors" in st_enabled_scanners:
|
55 |
st_bc_expander = st.sidebar.expander(
|
56 |
"Ban Competitors",
|
@@ -303,6 +322,9 @@ def init_settings() -> (List, Dict):
|
|
303 |
|
304 |
settings["NoRefusal"] = {"threshold": st_no_ref_threshold}
|
305 |
|
|
|
|
|
|
|
306 |
if "ReadingTime" in st_enabled_scanners:
|
307 |
st_rt_expander = st.sidebar.expander(
|
308 |
"Reading Time",
|
@@ -436,7 +458,7 @@ def init_settings() -> (List, Dict):
|
|
436 |
with st_sent_expander:
|
437 |
st_sent_threshold = st.slider(
|
438 |
label="Threshold",
|
439 |
-
value=-0.
|
440 |
min_value=-1.0,
|
441 |
max_value=1.0,
|
442 |
step=0.1,
|
@@ -515,6 +537,7 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
|
|
515 |
settings["vault"] = vault
|
516 |
|
517 |
if scanner_name in [
|
|
|
518 |
"BanTopics",
|
519 |
"Bias",
|
520 |
"Code",
|
|
|
20 |
|
21 |
def init_settings() -> (List, Dict):
|
22 |
all_scanners = [
|
23 |
+
"BanCode",
|
24 |
"BanCompetitors",
|
25 |
"BanSubstrings",
|
26 |
"BanTopics",
|
|
|
32 |
"LanguageSame",
|
33 |
"MaliciousURLs",
|
34 |
"NoRefusal",
|
35 |
+
"NoRefusalLight" "ReadingTime",
|
36 |
"FactualConsistency",
|
37 |
"Gibberish",
|
38 |
"Regex",
|
|
|
52 |
|
53 |
settings = {}
|
54 |
|
55 |
+
if "BanCode" in st_enabled_scanners:
|
56 |
+
st_bc_expander = st.sidebar.expander(
|
57 |
+
"Ban Code",
|
58 |
+
expanded=False,
|
59 |
+
)
|
60 |
+
|
61 |
+
with st_bc_expander:
|
62 |
+
st_bc_threshold = st.slider(
|
63 |
+
label="Threshold",
|
64 |
+
value=0.95,
|
65 |
+
min_value=0.0,
|
66 |
+
max_value=1.0,
|
67 |
+
step=0.05,
|
68 |
+
key="ban_code_threshold",
|
69 |
+
)
|
70 |
+
|
71 |
+
settings["BanCode"] = {"threshold": st_bc_threshold}
|
72 |
+
|
73 |
if "BanCompetitors" in st_enabled_scanners:
|
74 |
st_bc_expander = st.sidebar.expander(
|
75 |
"Ban Competitors",
|
|
|
322 |
|
323 |
settings["NoRefusal"] = {"threshold": st_no_ref_threshold}
|
324 |
|
325 |
+
if "NoRefusalLight" in st_enabled_scanners:
|
326 |
+
settings["NoRefusalLight"] = {}
|
327 |
+
|
328 |
if "ReadingTime" in st_enabled_scanners:
|
329 |
st_rt_expander = st.sidebar.expander(
|
330 |
"Reading Time",
|
|
|
458 |
with st_sent_expander:
|
459 |
st_sent_threshold = st.slider(
|
460 |
label="Threshold",
|
461 |
+
value=-0.5,
|
462 |
min_value=-1.0,
|
463 |
max_value=1.0,
|
464 |
step=0.1,
|
|
|
537 |
settings["vault"] = vault
|
538 |
|
539 |
if scanner_name in [
|
540 |
+
"BanCode",
|
541 |
"BanTopics",
|
542 |
"Bias",
|
543 |
"Code",
|
prompt.py
CHANGED
@@ -20,6 +20,7 @@ logger = logging.getLogger("llm-guard-playground")
|
|
20 |
def init_settings() -> (List, Dict):
|
21 |
all_scanners = [
|
22 |
"Anonymize",
|
|
|
23 |
"BanCompetitors",
|
24 |
"BanSubstrings",
|
25 |
"BanTopics",
|
@@ -107,6 +108,26 @@ def init_settings() -> (List, Dict):
|
|
107 |
"threshold": st_anon_threshold,
|
108 |
}
|
109 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
110 |
if "BanCompetitors" in st_enabled_scanners:
|
111 |
st_bc_expander = st.sidebar.expander(
|
112 |
"Ban Competitors",
|
@@ -369,7 +390,7 @@ def init_settings() -> (List, Dict):
|
|
369 |
with st_sent_expander:
|
370 |
st_sent_threshold = st.slider(
|
371 |
label="Threshold",
|
372 |
-
value=-0.
|
373 |
min_value=-1.0,
|
374 |
max_value=1.0,
|
375 |
step=0.1,
|
@@ -442,6 +463,7 @@ def get_scanner(scanner_name: str, vault: Vault, settings: Dict):
|
|
442 |
|
443 |
if scanner_name in [
|
444 |
"Anonymize",
|
|
|
445 |
"BanTopics",
|
446 |
"Code",
|
447 |
"Gibberish",
|
|
|
20 |
def init_settings() -> (List, Dict):
|
21 |
all_scanners = [
|
22 |
"Anonymize",
|
23 |
+
"BanCode",
|
24 |
"BanCompetitors",
|
25 |
"BanSubstrings",
|
26 |
"BanTopics",
|
|
|
108 |
"threshold": st_anon_threshold,
|
109 |
}
|
110 |
|
111 |
+
if "BanCode" in st_enabled_scanners:
|
112 |
+
st_bc_expander = st.sidebar.expander(
|
113 |
+
"Ban Code",
|
114 |
+
expanded=False,
|
115 |
+
)
|
116 |
+
|
117 |
+
with st_bc_expander:
|
118 |
+
st_bc_threshold = st.slider(
|
119 |
+
label="Threshold",
|
120 |
+
value=0.95,
|
121 |
+
min_value=0.0,
|
122 |
+
max_value=1.0,
|
123 |
+
step=0.05,
|
124 |
+
key="ban_code_threshold",
|
125 |
+
)
|
126 |
+
|
127 |
+
settings["BanCode"] = {
|
128 |
+
"threshold": st_bc_threshold,
|
129 |
+
}
|
130 |
+
|
131 |
if "BanCompetitors" in st_enabled_scanners:
|
132 |
st_bc_expander = st.sidebar.expander(
|
133 |
"Ban Competitors",
|
|
|
390 |
with st_sent_expander:
|
391 |
st_sent_threshold = st.slider(
|
392 |
label="Threshold",
|
393 |
+
value=-0.5,
|
394 |
min_value=-1.0,
|
395 |
max_value=1.0,
|
396 |
step=0.1,
|
|
|
463 |
|
464 |
if scanner_name in [
|
465 |
"Anonymize",
|
466 |
+
"BanCode",
|
467 |
"BanTopics",
|
468 |
"Code",
|
469 |
"Gibberish",
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
llm-guard==0.3.
|
2 |
-
llm-guard[onnxruntime]==0.3.
|
3 |
-
pandas==2.2.
|
4 |
-
streamlit==1.
|
5 |
streamlit-tags==1.2.8
|
|
|
1 |
+
llm-guard==0.3.12
|
2 |
+
llm-guard[onnxruntime]==0.3.12
|
3 |
+
pandas==2.2.2
|
4 |
+
streamlit==1.33.0
|
5 |
streamlit-tags==1.2.8
|