搭建SpringAI工程
由于最近大模型火爆,也分享一个简单的SpringAI功能调用的使用方法
官方文档:https://docs.spring.io/spring-ai/reference/api/chat/deepseek-chat.html
1、新建SpringBootAI工程
使用哪个AI用哪个AI即可
Ollama默认集成的接口
参考基础代码
import com.newland.newlandai.model.DeepSeekModel; import org.springframework.ai.chat.messages.UserMessage; import org.springframework.ai.chat.model.ChatResponse; import org.springframework.ai.chat.prompt.Prompt; import org.springframework.ai.ollama.OllamaChatModel; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RestController; import reactor.core.publisher.Flux; import java.util.Map; @RestController public class ChatController { private final OllamaChatModel chatModel; @Autowired public ChatController(OllamaChatModel chatModel) { this.chatModel = chatModel; } @PostMapping("/ai/generate") public Map generate(@RequestBody DeepSeekModel deepSeekModel) { return Map.of("generation", this.chatModel.call(deepSeekModel.getMessage())); } @PostMapping("/ai/generateStream") public Flux<ChatResponse> generateStream(@RequestBody DeepSeekModel deepSeekModel) { Prompt message = new Prompt(new UserMessage(deepSeekModel.getMessage())); return this.chatModel.stream(message); } }
- 配置文件:将系统生成的application改成yml
spring: application: name: new-land-ai ai: ollama: base-url: http://127.0.01:11434 chat: model: deepseek-r1:8b server: port: 8081
本文来自博客园,作者:Kwoky,转载请注明原文链接:https://www.cnblogs.com/kwoky/p/18720422
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步