fix: dataset and moderation. (#2042)

This commit is contained in:
Garfield Dai 2024-01-15 21:53:31 +08:00 • committed by GitHub
commit 83105d0d8f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View file

@ -30,7 +30,7 @@ class KeywordsModeration(Moderation):
if query:
inputs['query__'] = query
keywords_list = self.config['keywords'].split('\n')
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)