基于Spring-AI框架实现RAG增强检索(附源码)

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import ai.spring.core.ResolutionContext;
import ai.spring.core.annotation.AiComponent;
import ai.spring.core.annotation.Qualifier;
 
@Service
public class RagEnrichmentService {
 
    @Autowired
    private ResolutionContext resolutionContext;
 
    @AiComponent
    public void enhanceWithRagStatus(MyDataObject dataObject) {
        // 获取RAG服务
        RagService ragService = resolutionContext.getService(RagService.class, "ragService");
        // 假设MyDataObject有一个方法getRagStatus可以设置RAG状态
        dataObject.setRagStatus(ragService.getRagStatus(dataObject.getId()));
    }
}
 
// 假设的MyDataObject类
class MyDataObject {
    private String id;
    private String ragStatus;
 
    // getter和setter方法
    public String getId() {
        return id;
    }
 
    public void setId(String id) {
        this.id = id;
    }
 
    public String getRagStatus() {
        return ragStatus;
    }
 
    public void setRagStatus(String ragStatus) {
        this.ragStatus = ragStatus;
    }
}
 
// 假设的RAG服务接口
interface RagService {
    String getRagStatus(String dataObjectId);
}

这个代码示例展示了如何在Spring-AI框架中使用ResolutionContext来获取服务，并且如何将服务的功能应用到数据对象上。这里的enhanceWithRagStatus方法是一个AiComponent，它接受一个MyDataObject实例，并使用注入的RagService来设置数据对象的RAG状态。这个例子是基于原代码的简化版，去除了一些复杂性，并提供了必要的注释。