- 引入依赖
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-openai-spring-boot-starter</artifactId>
</dependency>
2.yml配置
spring:
ai:
openai:
api-key: sk-xxxxx // 填写自己申请的key
base-url: https://api.deepseek.com
chat:
options:
model: deepseek-chat
- 调用
@RestController
@RequestMapping(value = "ai")
public class ChatController {
private final OpenAiChatModel chatModel;
public ChatController(OpenAiChatModel chatModel) {
this.chatModel = chatModel;
}
@GetMapping("/generate")
public Object generate(@RequestParam String message) {
return this.chatModel.call(message);
}
@GetMapping("/generateStream")
public Flux<ChatResponse> generateStream(@RequestParam String message) {
Prompt prompt = new Prompt(new UserMessage(message));
return this.chatModel.stream(prompt);
}
}