Spring AI实战：10分钟接入OpenAI实现流式对话

前言

Spring AI 是 Spring 官方推出的 AI 集成框架，让 Java 开发者能用熟悉的 Spring 风格快速接入各大 AI 模型。本文带你 10 分钟完成 OpenAI 接入，并实现流式对话效果。
一、环境准备

1. <dependency>
2.     <groupId>org.springframework.ai</groupId>
3.     spring-ai-openai-spring-boot-starter</artifactId>
4.     <version>1.0.0-M6</version>
5. </dependency>

复制代码

二、配置 API Key

1. # application.yml
2. spring:
3.   ai:
4.     openai:
5.       api-key: your-api-key-here
6.       chat:
7.         options:
8.           model: gpt-4o-mini
9.           temperature: 0.7

复制代码

三、普通对话

1. @RestController
2. @RequestMapping("/ai")
3. public class ChatController {
5.     private final ChatClient chatClient;
7.     public ChatController(ChatClient.Builder builder) {
8.         this.chatClient = builder
9.             .defaultSystem("你是一个专业的Java技术顾问，回答简洁专业。")
10.             .build();
11.     }
13.     @GetMapping("/chat")
14.     public String chat(@RequestParam String message) {
15.         return chatClient.prompt()
16.             .user(message)
17.             .call()
18.             .content();
19.     }
20. }

复制代码

四、流式对话（打字机效果）

1. @GetMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
2. public Flux<String> stream(@RequestParam String message) {
3.     return chatClient.prompt()
4.         .user(message)
5.         .stream()
6.         .content();
7. }
9. // 前端接收流式响应
10. const eventSource = new EventSource("/ai/stream?message=介绍Spring AI");
11. eventSource.onmessage = (event) => {
12.     document.getElementById("output").innerHTML += event.data;
13. };
14. eventSource.onerror = () => eventSource.close();

复制代码

五、多轮对话（带记忆）

1. @RestController
2. public class MemoryChatController {
4.     private final ChatClient chatClient;
6.     public MemoryChatController(ChatClient.Builder builder, ChatMemory chatMemory) {
7.         this.chatClient = builder
8.             .defaultAdvisors(new MessageChatMemoryAdvisor(chatMemory))
9.             .build();
10.     }
12.     @GetMapping("/memory-chat")
13.     public String memoryChat(
14.             @RequestParam String message,
15.             @RequestParam(defaultValue = "default") String sessionId) {
16.         return chatClient.prompt()
17.             .user(message)
18.             .advisors(a -> a.param(CHAT_MEMORY_CONVERSATION_ID_KEY, sessionId))
19.             .call()
20.             .content();
21.     }
22. }

复制代码

六、结构化输出

1. record MovieRecommendation(String title, String director, int year, String reason) {}
3. @GetMapping("/recommend")
4. public MovieRecommendation recommend(@RequestParam String genre) {
5.     return chatClient.prompt()
6.         .user("推荐一部" + genre + "类型的电影，用JSON格式返回")
7.         .call()
8.         .entity(MovieRecommendation.class);
9. }

复制代码

总结

Spring AI 让 Java 开发者以最小的学习成本接入 AI 能力。核心优势：统一抽象（一套 API 支持 OpenAI、Azure、Ollama 等多个模型）；Spring 生态无缝集成；流式输出、多轮对话、结构化输出开箱即用。

觉得有帮助请点赞收藏！有问题欢迎评论区交流
来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！