From 70baa8e67d71df4e6e5a42b2c047e2a2583a29aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Phil=20H=C3=B6fer?= <phil.hoefer@suma-ev.de>
Date: Mon, 22 Jul 2024 12:56:35 +0200
Subject: [PATCH] Improve Blocklist Parsing
---
src/predictors/basic_markov.rs | 2 +-
src/predictors/basic_set.rs | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/predictors/basic_markov.rs b/src/predictors/basic_markov.rs
index 963a7fb..3f0e400 100644
--- a/src/predictors/basic_markov.rs
+++ b/src/predictors/basic_markov.rs
@@ -62,7 +62,7 @@ impl Predictor for MarkovChainPredictor {
};
for word in blocklist.clone() {
- if query.contains(word) {
+ if word.trim().len() > 1 && query.to_lowercase().contains(String::from(word).to_lowercase().as_str()) {
return Ok(());
}
}
diff --git a/src/predictors/basic_set.rs b/src/predictors/basic_set.rs
index 0246f8a..a97196b 100644
--- a/src/predictors/basic_set.rs
+++ b/src/predictors/basic_set.rs
@@ -61,7 +61,7 @@ impl Predictor for SetPredictor {
};
for word in blocklist.clone() {
- if query.to_lowercase().contains(String::from(word).to_lowercase().as_str()) {
+ if word.trim().len() > 1 && query.to_lowercase().contains(String::from(word).to_lowercase().as_str()) {
return Ok(());
}
}
--
GitLab