find_DB.py 文件源码

def find_match(query, intent):
    global col
    doc = None
    max_ratio = 0

    for d in col.find({"intent": intent}):
        ratio = fuzz.token_set_ratio(d['text'], query)

        if ratio > max_ratio:
            max_ratio = ratio
            doc = d

    del doc['_id']
    return doc['answer']