fix: dataset and moderation. (#2042)

2024-01-15 21:53:31 +08:00
parent 7b0818b8e5
commit 83105d0d8f
2 changed files with 2 additions and 2 deletions
--- a/api/core/moderation/keywords/keywords.py
+++ b/api/core/moderation/keywords/keywords.py
@@ -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)