Unverified Commit 0a5de0ff authored by Garfield Dai's avatar Garfield Dai Committed by GitHub

fix: empty keywords moderation. (#2108)

parent 9c4bad8f
...@@ -30,7 +30,10 @@ class KeywordsModeration(Moderation): ...@@ -30,7 +30,10 @@ class KeywordsModeration(Moderation):
if query: if query:
inputs['query__'] = query inputs['query__'] = query
# Filter out empty values
keywords_list = [keyword for keyword in self.config['keywords'].split('\n') if keyword] keywords_list = [keyword for keyword in self.config['keywords'].split('\n') if keyword]
flagged = self._is_violated(inputs, keywords_list) flagged = self._is_violated(inputs, keywords_list)
return ModerationInputsResult(flagged=flagged, action=ModerationAction.DIRECT_OUTPUT, preset_response=preset_response) return ModerationInputsResult(flagged=flagged, action=ModerationAction.DIRECT_OUTPUT, preset_response=preset_response)
...@@ -40,7 +43,9 @@ class KeywordsModeration(Moderation): ...@@ -40,7 +43,9 @@ class KeywordsModeration(Moderation):
preset_response = "" preset_response = ""
if self.config['outputs_config']['enabled']: 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) flagged = self._is_violated({'text': text}, keywords_list)
preset_response = self.config['outputs_config']['preset_response'] preset_response = self.config['outputs_config']['preset_response']
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment