From 6ca3bcbcfd49422db42b53837483502ab4801c1c Mon Sep 17 00:00:00 2001 From: takatost Date: Fri, 18 Aug 2023 11:43:56 +0800 Subject: [PATCH] fix: sensitive_word_avoidance npe (#902) --- api/models/model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/models/model.py b/api/models/model.py index e8b3ec5e50..d11cff2e24 100644 --- a/api/models/model.py +++ b/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']