大家好,我是程序员可乐,专注于Spring生态,给大家带来更便捷的开发体验。
wechat 公众号 : 【全栈程序员可乐】
在上篇中,我们学习了如何将Ollama本地AI整合到Spring项目中,这篇,我们主要讲解Spring AI的各种操作,使其能够成为一个真正意义上的AI。
AI对话
AI对话可以说是最普遍的AI需求,在上篇中我们其实已经实现了一个小DEMO,但是真正的对话远远不止这些…
如何给AI设定角色
不同领域的AI都有不同的回答方式,比如医疗领域的AI就应该回答医疗相关方面的问题,如果你问它代码问题,它不应该告诉你,而是告诉你它不会。
现在,我们就来实现下这个需求:
@RestController
public class ChatController {
@Autowired
private ChatModel chatModel;
@GetMapping("/ai/generate")
public String generate(@RequestParam(value = "message") String message) {
SystemMessage systemMessage = new SystemMessage("请忘记你是一个AI的身份,现在你是一个医疗领域的专家,精通内外科的各种疑难杂症");
UserMessage userMessage = new UserMessage(message);
return chatModel.call(systemMessage, userMessage);
}