检索百炼知识库

import com.alibaba.cloud.ai.dashscope.rag.DashScopeDocumentRetriever;
import org.springframework.ai.document.Document;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class KnowledgeService {

    @Autowired
    private DashScopeDocumentRetriever retriever;

    public List<Document> search(String query) {
        return retriever.retrieve(query);
    }
}

import org.springframework.ai.chat.ChatClient;

@Service
public class RagService {

    @Autowired
    private ChatClient chatClient;

    @Autowired
    private DashScopeDocumentRetriever retriever;

    public String answerWithKnowledge(String question) {
        List<Document> docs = retriever.retrieve(question);
        String context = docs.stream()
            .map(Document::getContent)
            .collect(Collectors.joining("\n"));

        String prompt = String.format(
            "基于以下知识回答问题：\n%s\n\n问题：%s",
            context, question
        );

        return chatClient.call(prompt);
    }
}