Sfoglia il codice sorgente

fix: empty keywords moderation. (#2108)

Garfield Dai 1 anno fa
parent
commit
0a5de0ff0b
1 ha cambiato i file con 6 aggiunte e 1 eliminazioni
  1. 6 1
      api/core/moderation/keywords/keywords.py

+ 6 - 1
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']