fix: dataset query error. (#1244)

2023-09-27 15:24:54 +08:00
parent 18c710c906
commit 54ff03c35d
2 changed files with 2 additions and 2 deletions
--- a/api/models/model.py
+++ b/api/models/model.py
@@ -172,7 +172,7 @@ class AppModelConfig(db.Model):
            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.dataset_query_variable = model_config['dataset_query_variable']
+        self.dataset_query_variable = model_config.get('dataset_query_variable')
        self.pre_prompt = model_config['pre_prompt']
        self.agent_mode = json.dumps(model_config['agent_mode'])
        self.retriever_resource = json.dumps(model_config['retriever_resource']) \