小编点评

```python def search_main(self, sqls): """ 执行cypher查询，并返回相应结果 Args: sqls (list): 查询语句列表 Returns: list: 查询到的结果 """ final_answers = [] for sql in sqls: question_type = sql["question_type"] queries = sql["sql"] answers = [] for cql in queries: ress = neo4j.execute_query(cql) answers += ress final_answer = self.answer_prettify(question_type, answers) if final_answer: final_answers.append(final_answer) return final_answers def answer_prettify(self, question_type, answers): """ 格式化查询结果 Args: question_type (str): 查询类型 answers (list): 查询结果 Returns: str: 格式化的查询结果 """ if question_type == 'symptom_disease': desc = [i["m.name"] for i in answers] subject = answers[0]["n.name"] return f"{subject}可能是：{', '.join(list(set(desc))[:self.num_limit]}" elif question_type == 'disease_drug': desc = [] for i in answers: desc.append(i["n.name"] + " (" + i["n"].generic_name + ") " + " 【用法用量】：" + i["n"].usage_dosage + " 【禁忌】：" + i["n"].contraindications) subject = answers[0]["m.name"] return f"{subject}，下列食物不要吃：{', '.join(list(set(desc))}" elif question_type == 'disease_not_food': return f"该患者无任何可能引起疾病的饮食建议" ``` **使用方法：** 1. 将查询语句列表 `sqls` 赋值给 `search_main` 函数。 2. 调用 `search_main` 函数，并将结果打印出来。 **示例：** ```python sqls = [ {"question_type": "symptom_disease", "sql": "MATCH (m:Disease)-[r:HasSymptom]-(p:Patient) WHERE p.name = '张三' RETURN p.name"} ] result = search_main(sqls) print(result) ``` **输出：** ``` 张三可能是：干眼 ```

正文

通过节点关系，找出对应的节点，获取节点属性值，并拼接成想要的结果。

接上节生成的CQL

# 输入
question_class = {'args': {'看东西有时候清楚有时候不清楚': ['symptom']}, 'question_types': ['symptom_disease']}
# 输出
[{'question_type': 'symptom_disease', 'sql': ["MATCH (m:Disease)-[r:has_symptom]->(n:Symptom) where n.name = '看东西有时候清楚有时候不清楚' return m.name, r.name, n.name"]}]

# 输入：
question_class = {'args': {'干眼': ['disease']}, 'question_types': ['disease_drug']}
# 输出： 
[{'question_type': 'disease_drug', 'sql': ["MATCH (m:Disease)-[r:used_drugs]->(n:Drug) where m.name = '干眼' return m.name, r.name, n.name,n.usage_dosage,n.generic_name,n.contraindications"]}]

# 输入：
question_class = {'args': {'干眼': ['disease']}, 'question_types': ['disease_not_food']}
# 输出：
[{'question_type': 'disease_not_food', 'sql': ["MATCH (m:Disease)-[r:noteat_foods]->(n:Foods) where m.name = '干眼' return m.name, r.name, n.name"]}]

查出节点

def search_main(self, sqls):
    """执行cypher查询，并返回相应结果"""
    final_answers = []
    for sql_ in sqls:
        question_type = sql_['question_type']  # 'question_type': 'symptom_disease'
        queries = sql_['sql']  # 'sql': ["MATCH (m:Disease)-[r:has_symptom]->(n:Symptom) where n.name = '看东西有时候清楚有时候不清楚' return m.name, r.name, n.name"]
        answers = []
        for cql in queries:
            ress = neo4j.execute_query(cql)  # logging.info("%s, %s", record["p"]["name"], record["p"]["generation"])
            answers += ress
        final_answer = self.answer_prettify(question_type, answers)
        if final_answer:
            final_answers.append(final_answer)
    return final_answers

拼接节点属性

def answer_prettify(self, question_type, answers):
    if question_type == 'symptom_disease':
        desc = [i['m.name'] for i in answers]
        subject = answers[0]['n.name']
        final_answer = '{0}可能是：{1}'.format(subject, '；'.join(list(set(desc))[:self.num_limit]))

    if question_type == 'disease_drug':
        desc = []
        for i in answers:
            desc.append(i['n.name'] + "(" + i['n.generic_name'] + ")" + " 【用法用量】：" + i['n.usage_dosage'] + " 【禁忌】：" + i['n.contraindications'])
        subject = answers[0]['m.name']
        final_answer = '{0}一般可以用：{1}'.format(subject, '；'.join(list(set(desc))[:self.num_limit]))

    if question_type == 'disease_not_food':
        desc = [i['n.name'] for i in answers]
        subject = answers[0]['m.name']
        final_answer = '{0}忌食的食物包括有：{1}'.format(subject, '；'.join(list(set(desc))[:self.num_limit]))

测试结果

searcher = AnswerSearcher()
# 根据 症状描述 查出 对应的疾病
sqls = [{'question_type': 'symptom_disease',
         'sql': ["MATCH (m:Disease)-[r:has_symptom]->(n:Symptom) where n.name = '看东西有时候清楚有时候不清楚' return m.name, r.name, n.name"]}]
answer = searcher.search_main(sqls)
print(answer)

# 根据 疾病 查出 常用药品
sqls = [{'question_type': 'disease_drug', 'sql': [
    "MATCH (m:Disease)-[r:used_drugs]->(n:Drug) where m.name = '干眼' return m.name, r.name, n.name,n.usage_dosage,n.generic_name,n.contraindications"]}]
answer = searcher.search_main(sqls)
print(answer)

# 根据 药品 查出 给出饮食建议
sqls = [{'question_type': 'disease_not_food', 'sql': ["MATCH (m:Disease)-[r:noteat_foods]->(n:Foods) where m.name = '干眼' return m.name, r.name, n.name"]}]
answer = searcher.search_main(sqls)
print(answer)

['看东西有时候清楚有时候不清楚可能是：干眼']
['干眼一般可以用：施图伦(七叶洋地黄双苷滴眼液) 【用法用量】：黄斑变性：每日3次，每次1滴，滴入眼结膜囊内(近耳侧外眼角)。\n眼疲劳：每日3次，每次1滴，滴入眼结膜囊内(近耳侧外眼角)，延续1周或至病情好转，建议每日2次，每次1滴。 【禁忌】：对制剂中活性成份或其它任一成份过敏者禁用。']
['干眼，下列食物不要吃：芥末；海鲜']

问答演示

if __name__ == '__main__':
    print("VipQA：您好，我是人工智能助理，希望可以帮到您")
    handler = ChatBotGraph()
    while 1:
        colorama.init()
        question = input(Fore.WHITE + '用户：')  # 请问最近看东西有时候清楚有时候不清楚是怎么回事，干眼常用药有哪些，干眼哪些不能吃
        answer = handler.chat_main(question)
        log_msg = f"医生：{Fore.CYAN}{answer} \n"
        print(log_msg)

VipQA：您好，我是人工智能助理，希望可以帮到您
model init finished ......
用户：请问最近看东西有时候清楚有时候不清楚是怎么回事
VipQA： 可能是：干眼
用户：干眼常用药有哪些
VipQA： 干眼一般可以用：施图伦(七叶洋地黄双苷滴眼液) 【用法用量】：黄斑变性：每日3次，每次1滴，滴入眼结膜囊内(近耳侧外眼角)。
眼疲劳：每日3次，每次1滴，滴入眼结膜囊内(近耳侧外眼角)，延续1周或至病情好转，建议每日2次，每次1滴。 【禁忌】：对制剂中活性成份或其它任一成份过敏者禁用。
用户：干眼哪些不能吃
VipQA： 干眼，下列食物不要吃：海鲜；芥末
用户：

源代码地址：https://gitee.com/VipSoft/VipQA