diff --git a/api/apps/dialog_app.py b/api/apps/dialog_app.py index 42fde4f9f..4878e0732 100644 --- a/api/apps/dialog_app.py +++ b/api/apps/dialog_app.py @@ -43,7 +43,7 @@ def set_dialog(): similarity_threshold = req.get("similarity_threshold", 0.1) vector_similarity_weight = req.get("vector_similarity_weight", 0.3) llm_setting = req.get("llm_setting", {}) - default_prompt = { + default_prompt_with_dataset = { "system": """你是一个智能助手,请总结知识库的内容来回答问题,请列举知识库中的数据详细回答。当所有知识库内容都与问题无关时,你的回答必须包括“知识库中未找到您要的答案!”这句话。回答需要考虑聊天历史。 以下是知识库: {knowledge} @@ -54,10 +54,22 @@ def set_dialog(): ], "empty_response": "Sorry! 知识库中未找到相关内容!" } - prompt_config = req.get("prompt_config", default_prompt) + default_prompt_no_dataset = { + "system": """You are a helpful assistant.""", + "prologue": "您好,我是您的助手小樱,长得可爱又善良,can I help you?", + "parameters": [ + + ], + "empty_response": "" + } + prompt_config = req.get("prompt_config", default_prompt_with_dataset) if not prompt_config["system"]: - prompt_config["system"] = default_prompt["system"] + prompt_config["system"] = default_prompt_with_dataset["system"] + + if not req.get("kb_ids", []): + if prompt_config['system'] == default_prompt_with_dataset['system'] or "{knowledge}" in prompt_config['system']: + prompt_config = default_prompt_no_dataset for p in prompt_config["parameters"]: if p["optional"]: