📂AI
🔖AI
2025-02-17 17:34阅读: 4评论: 0推荐: 0

搭建SpringAI工程

由于最近大模型火爆,也分享一个简单的SpringAI功能调用的使用方法

官方文档:https://docs.spring.io/spring-ai/reference/api/chat/deepseek-chat.html

1、新建SpringBootAI工程

image
image
使用哪个AI用哪个AI即可

Ollama默认集成的接口

image

参考基础代码

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
posted @   Kwoky  阅读(4)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起