首页
/ Spring AI项目中ChatMemory参数保存问题的分析与解决

Spring AI项目中ChatMemory参数保存问题的分析与解决

2025-06-11 17:07:44作者:凌朦慧Richard

在Spring AI项目的开发过程中,开发者可能会遇到ChatMemory无法正确保存PromptUserSpec参数的问题。这个问题主要出现在使用ChatClient构建聊天请求时,当通过PromptUserSpec设置模板和参数后,历史记录中只保存了未渲染的模板字符串,而没有保存实际替换后的内容。

问题现象

当开发者使用如下代码构建聊天请求时:

private ChatClient.ChatClientRequestSpec buildChatClientRequestSpecRAGAdvisor(Question question, String conversationId) {
    return chatClient.prompt()
            .user(userSpec -> userSpec
                    .text("{question} (error: '{error}')?")
                    .param("question", question.question())
                    .param("error", question.error()!=null?question.error():""))
            .advisors( advisorSpec -> advisorSpec.param(AbstractChatMemoryAdvisor.CHAT_MEMORY_CONVERSATION_ID_KEY, conversationId));
}

期望在ChatMemory中保存的是已经替换了参数的完整消息,例如:"What is AI? (error: 'None')",但实际保存的却是未渲染的模板字符串:"{question} (error: '{error}')"。

问题根源

这个问题源于Spring AI框架中PromptChatMemoryAdvisor的实现逻辑。在1.0.0-M8版本之前,PromptChatMemoryAdvisor在处理用户消息时,直接将未渲染的模板字符串添加到了ChatMemoryStore中,而没有先执行参数替换操作。

具体来说,问题出在PromptChatMemoryAdvisor的before方法中:

this.getChatMemoryStore().add(this.doGetConversationId(request.adviseContext()), userMessage);

这里直接将原始的userMessage(包含模板字符串)添加到了内存存储中,而没有先进行参数替换。

解决方案

Spring AI团队在1.0.0-M8版本中修复了这个问题。修复的核心思路是让ChatClient在执行advisor链之前先渲染输入提示。这意味着:

  1. 在请求处理流程中,ChatClient会先处理PromptUserSpec中的参数替换
  2. 将渲染后的完整消息传递给后续的advisor处理
  3. PromptChatMemoryAdvisor现在接收到的是已经替换了参数的消息内容

这个改变确保了ChatMemory中保存的是用户实际发送的消息内容,而不是模板字符串。

升级建议

对于遇到此问题的开发者,建议升级到Spring AI 1.0.0-M8或更高版本。升级后,ChatMemory将能够正确保存渲染后的消息内容。

如果由于某些原因无法立即升级,开发者也可以考虑以下临时解决方案:

  1. 手动渲染消息内容后再传递给ChatClient
  2. 实现自定义的ChatMemoryAdvisor,在保存消息前执行参数替换

总结

Spring AI框架在不断演进中解决了许多实际开发中遇到的问题。这个ChatMemory参数保存问题的解决,体现了框架对开发者体验的重视。通过理解问题的根源和解决方案,开发者可以更好地利用Spring AI构建强大的聊天应用,同时也能在遇到类似问题时更快地找到解决方法。

对于使用Spring AI的开发团队,建议保持框架版本的及时更新,并关注官方文档中的升级说明,以确保能够获得最新的功能改进和问题修复。

登录后查看全文
热门项目推荐
相关项目推荐