RAG 实战案例

@Service
public class KnowledgeBaseLoader {

    private final VectorStore vectorStore;

    public KnowledgeBaseLoader(VectorStore vectorStore) {
        this.vectorStore = vectorStore;
    }

    @PostConstruct
    public void loadAllDocuments() {
        List<Resource> resources = List.of(
            new FileSystemResource("C:/docs/feixiang/产品手册.pdf"),
            new FileSystemResource("C:/docs/feixiang/FAQ常见问题.pdf"),
            new FileSystemResource("C:/docs/feixiang/员工手册.pdf"),
            new FileSystemResource("C:/docs/feixiang/技术白皮书.pdf"),
            new FileSystemResource("C:/docs/feixiang/客户案例集.pdf")
        );

        TokenTextSplitter splitter = new TokenTextSplitter();
        splitter.setDefaultChunkSize(500);
        splitter.setMinChunkSizeChars(200);

        int totalChunks = 0;
        for (Resource resource : resources) {
            // Extract
            PagePdfDocumentReader reader = new PagePdfDocumentReader(resource);
            List<Document> pages = reader.get();

            // 注入来源元数据
            String fileName = resource.getFilename();
            for (Document doc : pages) {
                doc.getMetadata().put("source", fileName);
            }

            // Transform
            List<Document> chunks = splitter.apply(pages);

            // Load
            vectorStore.add(chunks);
            totalChunks += chunks.size();

            System.out.printf("已注入：%s → %d 块\n", fileName, chunks.size());
        }

        System.out.printf("知识库构建完成！共 %d 个文本块\n", totalChunks);
    }
}

@Configuration
public class SmartServiceConfig {

    @Bean
    public ChatClient chatClient(ChatClient.Builder builder,
                                  VectorStore vectorStore) {
        QuestionAnswerAdvisor ragAdvisor = new QuestionAnswerAdvisor(vectorStore);
        ragAdvisor.setPromptTemplate("""
                你是飞翔科技的智能客服。请基于以下产品文档回答用户问题。
                
                【回答要求】
                1. 优先使用文档中的原话，保持术语准确
                2. 如果涉及多个方案，列出对比
                3. 附上文档来源段落编号
                4. 如果文档未覆盖该问题，回复：
                   「该问题暂未收录，已转人工客服处理」
                
                相关文档：
                {context}
                
                用户问题：{question}
                
                回答：
                """);

        return builder
                .defaultAdvisors(
                    ragAdvisor,
                    new SimpleLoggerAdvisor()  // 方便李眉监控调用
                )
                .defaultSystem("你是飞翔科技智能客服，态度友好专业。")
                .build();
    }
}

@RestController
@RequestMapping("/api/v1/customer-service")
public class CustomerServiceController {

    private final ChatClient chatClient;

    public CustomerServiceController(ChatClient chatClient) {
        this.chatClient = chatClient;
    }

    @GetMapping("/chat")
    public Map<String, String> chat(@RequestParam String question) {
        // 注意：不需要传 VectorStore 参数
        // QuestionAnswerAdvisor 自动处理检索
        
        String answer = chatClient.prompt()
                .user(question)
                .call()
                .content();
        
        return Map.of(
            "question", question,
            "answer", answer
        );
    }

    /**
     * 带部门过滤的搜索（不同部门看到不同文档）
     */
    @GetMapping("/department-chat")
    public Map<String, String> departmentChat(
            @RequestParam String question,
            @RequestParam(defaultValue = "全公司") String department) {
        
        // 可以注入 VectorStore 单独做精确检索
        String answer = chatClient.prompt()
                .user(question)
                .call()
                .content();
        
        return Map.of(
            "department", department,
            "question", question,
            "answer", answer
        );
    }
}

// ========== 操作前（无 RAG）==========
GET /api/v1/customer-service/chat?question=飞翔API的QPS上限是多少？

{
  "question": "飞翔API的QPS上限是多少？",
  "answer": "作为AI助手，我没有飞翔科技API的具体性能数据。
             Typically API rate limits vary by plan..."
}

// ========== 操作后（有 RAG）==========
GET /api/v1/customer-service/chat?question=飞翔API的QPS上限是多少？

{
  "question": "飞翔API的QPS上限是多少？",
  "answer": "根据《技术白皮书》第4章性能指标：
            - 标准版：100 QPS（支持扩容至 500 QPS）
            - 企业版：1000 QPS（支持扩容至 5000 QPS）
            - 旗舰版：不限（需独立部署）
            扩容需提前 3 个工作日提交工单。
            [技术白皮书-P42]"
}