基于RAG的信息检索方法、装置、存储介质及设备

本申请公开了一种基于RAG的信息检索方法、装置、存储介质及设备，属于深度学习技术领域。获取用户输入的检索语句；利用预先训练好的决策模型对检索语句进行处理，得到数据库集合中每个数据库的token比例，数据库集合是基于RAG进行信息检索时，问答模型连接的所有数据库的集合；对于每个数据库，根据检索语句、数据库对应的token比例和问答模型能够输入的token总数生成数据库的token结果；将所有数据库的token结果拼接成问答模型的提示词；利用问答模型对提示词进行处理，得到检索语句的检索结果。本申请能合理分配不同数据库提供的token结果，避免引入不相关的token，使RAG能得到更加准确的答案。