当前位置：首页 > news >正文

第十一篇：SpringAI 实战 11｜Advisor 机制与对话记忆（ChatMemory）：让 AI 拥有“记忆力”

news 2026/6/14 1:08:38

导读：在前面的章节中，我们成功实现了多模型共存与丝滑的流式输出。但在实际测试中，如果你尝试连续追问（例如：你告诉AI你叫小明” -> 再问Ai“你叫什么名字”），你会发现 AI 在第二轮直接“失忆”了。这是因为大模型（LLM）本质上是无状态的。
为了让 AI 拥有连贯的上下文记忆，Spring AI 提供了两个核心机制：
1.ChatMemory（对话记忆）：负责对话历史的存储与管理。
2.Advisor（顾问机制）：基于 AOP 思想，在请求发送前自动注入历史，在响应后自动保存新对话。
本章我们将以通义千问（Qwen）为例，彻底打通 AI 的多轮对话能力，并配合网页端进行可视化验证。

一、环境前置说明

运行前提：电脑安装 Ollama客户端，提前拉取开源模型文件

JDK：21
Gradle：8.8
SpringBoot：3.5.14
SpringAI：1.1.7
IDEA：2023 社区版
（本章代码是在上一篇的基础上新增/修改的）

二、核心配置：装配 ChatMemory 与 Advisor

packagecom.example.demo.config;importorg.springframework.ai.chat.client.ChatClient;importorg.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;importorg.springframework.ai.chat.memory.ChatMemory;importorg.springframework.ai.chat.memory.InMemoryChatMemoryRepository;importorg.springframework.ai.chat.memory.MessageWindowChatMemory;importorg.springframework.ai.chat.model.ChatModel;importorg.springframework.ai.openai.OpenAiChatModel;importorg.springframework.ai.openai.OpenAiChatOptions;importorg.springframework.ai.openai.api.OpenAiApi;importorg.springframework.beans.factory.annotation.Qualifier;importorg.springframework.beans.factory.annotation.Value;importorg.springframework.context.annotation.Bean;importorg.springframework.context.annotation.Configuration;@ConfigurationpublicclassChatMemoryConfig{@Value("${spring.ai.qwen.api-key}")privateStringapiKey;@Value("${spring.ai.qwen.base-url}")privateStringbaseUrl;/** * 1. 将手动构建的 Qwen 模型注册为 Spring Bean */@Bean("qwenChatModel")publicOpenAiChatModelqwenChatModel(){OpenAiApiqwenApi=OpenAiApi.builder().apiKey(apiKey).baseUrl(baseUrl).build();returnOpenAiChatModel.builder().openAiApi(qwenApi).defaultOptions(OpenAiChatOptions.builder().model("qwen-turbo").build()).build();}/** * 1. 声明内存级的 ChatMemory（生产环境可替换为 JdbcChatMemory） * maxMessages 表示滑动窗口大小，默认保留最近 20 条消息 */@BeanpublicChatMemorychatMemory(){returnMessageWindowChatMemory.builder().maxMessages(20).chatMemoryRepository(newInMemoryChatMemoryRepository()).build();}/** * 2. 构建带有记忆能力的通义千问 ChatClient */@Bean("qwenMemoryChatClient")publicChatClientqwenMemoryChatClient(// 明确指定注入通义千问的模型 Bean@Qualifier("qwenChatModel")ChatModelqwenChatModel,ChatMemorychatMemory){returnChatClient.builder(qwenChatModel)// 使用 Builder 模式挂载对话记忆 Advisor.defaultAdvisors(MessageChatMemoryAdvisor.builder(chatMemory).build()).build();}}

三、 Controller ：流式多轮对话接口

为了在网页端实现打字机效果，Controller 需要返回 Flux。关键在于：必须通过参数传递 conversationId，以实现不同用户/会话的记忆隔离。

packagecom.example.demo.controller;importorg.springframework.ai.chat.client.ChatClient;importorg.springframework.ai.chat.memory.ChatMemory;importorg.springframework.beans.factory.annotation.Qualifier;importorg.springframework.http.MediaType;importorg.springframework.web.bind.annotation.GetMapping;importorg.springframework.web.bind.annotation.RequestMapping;importorg.springframework.web.bind.annotation.RequestParam;importorg.springframework.web.bind.annotation.RestController;importreactor.core.publisher.Flux;@RestController@RequestMapping("/ai")publicclassChatMemoryController{@Qualifier("qwenMemoryChatClient")privatefinalChatClientqwenMemoryChatClient;publicChatMemoryController(@Qualifier("qwenMemoryChatClient")ChatClientqwenMemoryChatClient){this.qwenMemoryChatClient=qwenMemoryChatClient;}/** * 通义千问多轮对话流式接口 * @param conversationId 会话ID（前端生成或从 Session/Token 中获取） * @param msg 用户输入 */@GetMapping(value="/chat/stream-memory",produces=MediaType.TEXT_EVENT_STREAM_VALUE)publicFlux<String>chatWithMemory(@RequestParam(defaultValue="default-session")StringconversationId,@RequestParamStringmsg){returnqwenMemoryChatClient.prompt().user(msg)// 核心：指定当前对话的会话 ID.advisors(a->a.param(ChatMemory.CONVERSATION_ID,conversationId)).stream().content();}}

四、网页端：极简 HTML 验证对话记忆

为了验证通义千问的记忆能力，我们用一个原生的 HTML 文件即可完美对接 SSE 流。
在src/main/resources/static 目录下，创建 memory-test.html：

<!DOCTYPEhtml><htmllang="zh-CN"><head><metacharset="UTF-8"><title>通义千问对话记忆测试</title><style>body{font-family:sans-serif;padding:20px;}#output{border:1px solid #ccc;padding:15px;min-height:200px;white-space:pre-wrap;margin-bottom:10px;border-radius:8px;background:#f9f9f9;}.input-area{display:flex;gap:10px;}input{flex:1;padding:10px;font-size:16px;}button{padding:10px 20px;cursor:pointer;}.session-info{font-size:12px;color:#666;margin-bottom:5px;}</style></head><body><h2>通义千问多轮对话记忆测试</h2><divclass="session-info">当前会话 ID:<strongid="sessionIdDisplay"></strong><buttononclick="resetSession()"style="padding:2px 8px;font-size:12px;">重置会话</button></div><divid="output">等待输入...</div><divclass="input-area"><inputtype="text"id="msgInput"placeholder="试着连续提问，例如：'我叫小明' -> '你还记得我的名字吗？'"autofocus><buttononclick="startStream()">发送</button></div><script>letcurrentEventSource=null;// 为当前浏览器标签页生成一个唯一的会话 IDletcurrentSessionId='session-'+Math.random().toString(36).substr(2,9);document.getElementById('sessionIdDisplay').textContent=currentSessionId;functionresetSession(){if(currentEventSource)currentEventSource.close();currentSessionId='session-'+Math.random().toString(36).substr(2,9);document.getElementById('sessionIdDisplay').textContent=currentSessionId;document.getElementById('output').innerHTML='--- 会话已重置 ---\n';}functionstartStream(){constmsgInput=document.getElementById('msgInput');constmsg=msgInput.value.trim();if(!msg)return;constoutputDiv=document.getElementById('output');outputDiv.innerHTML+=`\n\n👤 你:${msg}\n🤖 千问:`;msgInput.value='';// 1. 关闭上一次的连接if(currentEventSource)currentEventSource.close();// 2. 建立 SSE 连接，传递 conversationId 和 msgconsturl=`/ai/chat/stream-memory?conversationId=${currentSessionId}&msg=${encodeURIComponent(msg)}`;currentEventSource=newEventSource(url);// 3. 监听消息，实现打字机追加效果currentEventSource.onmessage=(event)=>{outputDiv.innerHTML+=event.data;outputDiv.scrollTop=outputDiv.scrollHeight;};// 4. 监听完成或错误currentEventSource.onerror=()=>{currentEventSource.close();};}// 支持回车键发送document.getElementById('msgInput').addEventListener('keypress',function(e){if(e.key==='Enter')startStream();});</script></body></html>