From 0a5de0ff0b103fc47c4ab1b52ddc75303996f3cf Mon Sep 17 00:00:00 2001 From: Garfield Dai Date: Sat, 20 Jan 2024 20:02:51 +0800 Subject: [PATCH] fix: empty keywords moderation. (#2108) --- api/core/moderation/keywords/keywords.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/api/core/moderation/keywords/keywords.py b/api/core/moderation/keywords/keywords.py index e367ccfb69..ca562ad987 100644 --- a/api/core/moderation/keywords/keywords.py +++ b/api/core/moderation/keywords/keywords.py @@ -30,7 +30,10 @@ class KeywordsModeration(Moderation): if query: inputs['query__'] = query + + # Filter out empty values keywords_list = [keyword for keyword in self.config['keywords'].split('\n') if keyword] + flagged = self._is_violated(inputs, keywords_list) return ModerationInputsResult(flagged=flagged, action=ModerationAction.DIRECT_OUTPUT, preset_response=preset_response) @@ -40,7 +43,9 @@ class KeywordsModeration(Moderation): preset_response = "" if self.config['outputs_config']['enabled']: - keywords_list = self.config['keywords'].split('\n') + # Filter out empty values + keywords_list = [keyword for keyword in self.config['keywords'].split('\n') if keyword] + flagged = self._is_violated({'text': text}, keywords_list) preset_response = self.config['outputs_config']['preset_response']