Forráskód Böngészése

fix: sensitive_word_avoidance npe (#902)

takatost 1 éve
szülő
commit
6ca3bcbcfd
1 módosított fájl, 2 hozzáadás és 1 törlés
  1. 2 1
      api/models/model.py

+ 2 - 1
api/models/model.py

@@ -158,7 +158,8 @@ class AppModelConfig(db.Model):
         self.speech_to_text = json.dumps(model_config['speech_to_text']) \
             if model_config.get('speech_to_text') else None
         self.more_like_this = json.dumps(model_config['more_like_this'])
-        self.sensitive_word_avoidance = json.dumps(model_config['sensitive_word_avoidance'])
+        self.sensitive_word_avoidance = json.dumps(model_config['sensitive_word_avoidance']) \
+            if model_config.get('sensitive_word_avoidance') else None
         self.model = json.dumps(model_config['model'])
         self.user_input_form = json.dumps(model_config['user_input_form'])
         self.pre_prompt = model_config['pre_prompt']