Spring AI (六)外部资源

春哥的魔法书2024-07-25 15:09

6.根据资源回答

因为通义千问数据只到2021年, 所以不能响应最新的数据, 这里使用提供外部资源的方式来补充, 这也是为后面RAG + 向量数据库的过渡

资源

resources/docs/news-2024.md

txt 复制代码

2024年哈尔滨中考最高分是69总校的474分（满分480）。

哈工大附中2024年哈尔滨中考平均分是441.2分。

2024年黑龙江省普通高考的部分成绩数据已经公布，具体包括录取控制分数线。以下是已知的信息：
普通本科批录取控制分数线：
历史类：410分
物理类：360分
普通特殊类型招生资格线：
历史类：486分
物理类：480分

提示词模板

resources/prompts/qa-prompt.st

st 复制代码

使用以下上下文来回答最后的问题。如果你不知道答案，就说你不知道，不要试图编造一个答案。

{context}

问题: {question}
正确的答案:

代码实现

java 复制代码

    @Value("classpath:/docs/news-2024.md")
    private Resource docsToStuffResource;

    @Value("classpath:/prompts/qa-prompt.st")
    private Resource qaPromptResource;

    @GetMapping("/chat/byStuff")
    public String byStuff(@RequestParam(value = "message",
            defaultValue = "2020年黑龙江省高考最高分数是多少") String message,
                             @RequestParam(value = "stuffit", defaultValue = "false") boolean stuffit) {

        PromptTemplate promptTemplate = new PromptTemplate(qaPromptResource);
        Map<String, Object> map = new HashMap<>();
        map.put("question", message);

        if (stuffit) {
            map.put("context", docsToStuffResource);
        }
        else {
            map.put("context", "");
        }
        System.out.println("map = " + map);
        Prompt prompt = promptTemplate.create(map);
        return chatClient.call(prompt).getResult().getOutput().getContent();
    }

测试

http 复制代码

GET http://localhost:8080/stuff?message=2024年哈尔滨中考最高分是多少&stuffit=true

http 复制代码

GET http://localhost:8080/stuff?message=2024年黑龙江省普通本科批录取控制分数线分是多少&stuffit=true

http 复制代码

GET http://localhost:8080/stuff?message=哈工大附中2024年哈尔滨中考平均分是多少&stuffit=true