基于RAG和大模型的数据库查询训练方法及装置

本公开提供了基于RAG和大模型的数据库查询训练方法及装置，获取DDL标注数据与SQL标注数据并标准化为训练文档，将训练文档向量化为检索知识向量存储至预设向量数据库中；响应于用户的查询操作，将查询操作向量化为查询向量，根据查询向量在预设向量数据库中检索相匹配的关联DDL与关联SQL；根据查询操作对应的查询问题、关联DDL以及关联SQL构建模型提示词，将模型提示词输入至预设大语言模型，将查询问题转换为对应的目标SQL；确定目标SQL在目标业务数据库中的查询结果，并将查询结果反馈至用户。可以实现自然语言到SQL的智能转换，并引入向量检索增强生成过程，显著降低了大模型的幻觉问题，提高了生成SQL的准确性和安全性。