Query Rewrite (查询重写)

查询重写是RAG系统中的一个重要优化技术，它能够将用户的原始查询转换成更加结构化和明确的形式。这种转换可以提高检索的准确性，并帮助系统更好地理解用户的真实意图。
Spring AI提供了RewriteQueryTransformer来实现查询重写功能。参考文档

构建重写组件（利用RewriteQueryTransformer）

@Component
public class QueryTransformerConfig {
    @Resource
    private QueryTransformer queryTransformer;
    public QueryTransformerConfig(ChatModel dashscopeChatModel){
        ChatClient.Builder builder = ChatClient.builder(dashscopeChatModel);
        queryTransformer = RewriteQueryTransformer.builder()
                .chatClientBuilder(builder)
                .build();
    }
    public String WriteRag(String prompt){
        Query query = new Query(prompt);
        Query transform = queryTransformer.transform(query);
        return transform.text();
    }
}

1.通过构造函数传递 ChatModel (这里使用alibaba 的dashscopeChatModel 模型)

ChatClient.Builder builder = ChatClient.builder(dashscopeChatModel);

2.利用RewriteQueryTransformer 构造相关查询转换器

 queryTransformer = RewriteQueryTransformer.builder()
                .chatClientBuilder(builder)
                .build();

3.通过Query对象利用QueryTransformer的transform将prompt提示词进行重写

    public String WriteRag(String prompt){
        Query query = new Query(prompt);
        Query transform = queryTransformer.transform(query);
        return transform.text();
    }

4.最后在测试类中自动注入 该组件实现内容重写

 @Test
    void doChatWithRag() {
        String chatId = UUID.randomUUID().toString();
        String message = "我已经结婚了，但是婚后关系不太亲密，怎么办？";
        String ReWriteen = queryTransformerConfig.WriteRag(message);
        String answer =  loveApp.doChatWithRag(ReWriteen, chatId);
        }

5.实现效果