Skip to content

rewriter.py

概述

retrieval/queryRewriter/rewriter.py 它是对用户发送进入到系统的最粗糙初始查询语句进入任何例如词频统计或向量嵌入之前的极度关键的拦截修正器引擎。 例如因为用户的物理/数学公式写错别名或是使用了例如“上面提过的那个”指代,它需要利用一个小型的语言模型进行上下文解析翻译,又或是进行错别字修正,将非标准简称展开或是在 RAG 原文被判定缺相关性词前预见性的补充以保证搜索系统在后端的最高查全及命中准确度拦截重制功能代码实现组装管理业务实现。

典型核心管理装载

def rewrite_query(query: str, history=None) -> str

根据环境历史情况用本地小模型(如 Qwen 小尺度大模型微调版或基于规则引擎模板)重制/同义扩展该查询文本逻辑。

Math-RAG 毕业设计项目