搭建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