AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-07-28 05:32:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: dataset query error. (#1244)

Browse Source
This commit is contained in:
Garfield Dai 2023-09-27 15:24:54 +08:00 committed by GitHub
parent 18c710c906
commit 54ff03c35d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/models/model.py
View File

@ -172,7 +172,7 @@ class AppModelConfig(db.Model):
if model_config.get('sensitive_word_avoidance') else None if model_config.get('sensitive_word_avoidance') else None
self.model = json.dumps(model_config['model']) self.model = json.dumps(model_config['model'])
self.user_input_form = json.dumps(model_config['user_input_form']) 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.pre_prompt = model_config['pre_prompt']
self.agent_mode = json.dumps(model_config['agent_mode']) self.agent_mode = json.dumps(model_config['agent_mode'])
self.retriever_resource = json.dumps(model_config['retriever_resource']) \ self.retriever_resource = json.dumps(model_config['retriever_resource']) \

2
api/services/app_model_config_service.py
View File

@ -354,7 +354,7 @@ class AppModelConfigService:
"completion_params": config["model"]["completion_params"] "completion_params": config["model"]["completion_params"]
}, },
"user_input_form": config["user_input_form"], "user_input_form": config["user_input_form"],
"dataset_query_variable": config["dataset_query_variable"], "dataset_query_variable": config.get('dataset_query_variable'),
"pre_prompt": config["pre_prompt"], "pre_prompt": config["pre_prompt"],
"agent_mode": config["agent_mode"] "agent_mode": config["agent_mode"]
} }