基于检索查询语句的检索增强代码生成方法及系统

本公开提出一种基于检索查询语句的检索增强代码生成方法及系统，该方法包括：获取多个检索查询语句；从代码库中筛选出与自然语言需求和检索查询语句相匹配的上下文信息；根据自然语言需求和上下文信息生成第一代码段；通过与多个检索查询语句一一对应的多个第一代码段，以及每个第一代码段的相应标签，对检索查询语句生成器进行训练优化得到目标检索查询语句生成器；基于目标自然语言需求，利用训练优化好的目标检索查询语句生成器生成查询语句；使用查询语句检索相关的代码上下文，将目标自然语言需求和检索到的代码上下文输入代码生成模型中，完成检索增强的代码生成。本实施例能够有效减少冗余和不相关内容的干扰，从而提高代码生成的质量。