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

二.干货干货！！！SpringAI入门到实战-记忆存储和会话隔离

news 来源：原创 2025/6/16 5:40:05

前言

什么是记忆功能？默认情况下当我们向大模型每次发起的提问都是新的，大模型无法把我们的每次对话形成记忆，也无法根据对话上下文给出人性化的答案。比如：我的第一次提问是“懂王有哪些特点”，然后大模型会给出我懂王的特点结果列表，当我再次提问“这些特点中哪个最惹人争议”的时候，它就不知道我在说什么了，因为大模型已经失去了上一次的提问记忆。所以让智能体（如AI助手、机器人、虚拟角色等）拥有记忆功能不仅能提升交互体验，还能增强其功能性、适应性和长期价值。

SpringAI如何实现记忆存储

实现记忆存储的原理很简单，就是SpringAI会通过ChatMemory把对话的内容存储起来，下一次提问的时候会把历史提问内容一起带给大模型，这样一来大模型就知道历史对话内容了也就拥有了记忆能力。对于会话隔离功能则是通过一个会话ID实现的，每次对话我们传入一个会话ID，同一个会话ID对应了多次对话的消息列表，不同的会话ID对应了不同的消息列表从而实现会话隔离的目的，如下：
在这里插入图片描述

记忆存储实战

在SpringAI中提供了基于内存的会话存储组件InMemoryChatMemoryRepository,内部维护了一个 ConcurrentHashMap来存储消息列表，这肯定不是长久之计，如果消息内容比较庞大那么内存会爆满，或者机器重启消息会丢失，所以我们需要基于DB存储对话历史。那么这就需要集成数据库了。

1.集成Mysql

这里我们选择使用Mysql来存储对话历史，你也可以选择使用Mongo或者其他数据库，导入依赖如下

<dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-starter-model-chat-memory-repository-jdbc</artifactId></dependency><dependency><groupId>com.mysql</groupId><artifactId>mysql-connector-j</artifactId><scope>runtime</scope></dependency><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-spring-boot3-starter</artifactId><version>3.5.10.1</version></dependency>

spring-ai-starter-model-chat-memory-repository-jdbc ：SpringAI用来实现记忆会话的基础依赖

2.配置DataSource

在第一章的配置内容上增加数据源配置，关闭SpringAI自动初始化SQL脚本的功能，如下：

spring:ai:openai:api-key: ${API_KEY}base-url: https://dashscope.aliyuncs.com/compatible-modechat:options:model: qwen-max-latesttemperature: 1chat:memory:repository:jdbc:initialize-schema: never #不要初始化表，不然会报错datasource:driver-class-name: com.mysql.cj.jdbc.Driverusername: rootpassword: 123456url: jdbc:mysql://127.0.0.1/ai-girl
server:port: 8888mybatis-plus:configuration:map-underscore-to-camel-case: true

需要在Mysql中创建一个数据库，我这里的名字叫ai-girl
initialize-schema: never ：不要初始化表，不然会报错 schema.sql脚本找不到

SQL脚本如下，具体参考类：MysqlChatMemoryRepositoryDialect

CREATE TABLE `spring_ai_chat_memory` (`id` bigint NOT NULL AUTO_INCREMENT COMMENT 'ID',`conversation_id` varchar(255) DEFAULT NULL COMMENT '会话ID',`content` varchar(10000) DEFAULT NULL COMMENT '内容',`type` varchar(255) DEFAULT NULL COMMENT '类型',`timestamp` datetime DEFAULT NULL COMMENT '时间',PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=325 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;

3.配置大模型记忆功能

接着我们对大模型进行配置记忆功能。首先我们需要定义BeanChatMemory , 它是支持消息存储的必要组件，它提供了通过会话ID对消息的增加，删除，查询的基础能力。

@Bean
public ChatMemory chatMemory(JdbcTemplate jdbcTemplate) {return MessageWindowChatMemory.builder()//.chatMemoryRepository(new InMemoryChatMemoryRepository()).chatMemoryRepository(JdbcChatMemoryRepository.builder().jdbcTemplate(jdbcTemplate).build()).build();
}

我们通过 MessageWindowChatMemory来构建ChatMemory ，可以指定基于内存的InMemoryChatMemoryRepository 记忆存储 ,或者指定基于JDBC的JdbcChatMemoryRepository记忆存储。这里的jdbcTemplate是配置数据源之后，SpringAI帮我们自动装配的，直接使用即可。

在 JdbcChatMemoryRepository中根据不同的方言维护了不同的Dialect ,你可以看一下org.springframework.ai.chat.memory.repository.jdbc.JdbcChatMemoryRepositoryDialect这里类，它会根据不同的数据库类型采用不同的处理类，比如Mysql对应的是MysqlChatMemoryRepositoryDialect,在 MysqlChatMemoryRepositoryDialect中提供了对数据库的基础CURD的SQL。

接着我们来给ChatClient配置记忆功能,需要把上面定义的ChatMemory指定给ChatClient，如下：

@Bean
public ChatClient chatClient(OpenAiChatModel model,ChatMemory chatMemory){return ChatClient.builder(model)//指定系统提示词.defaultSystem("模拟女朋友的身份回答问题，不能暴露你AI的身份")//配置记忆存储功能.defaultAdvisors(MessageChatMemoryAdvisor.builder(chatMemory).build()).build();
}

4.测试记忆功能

编写Controller如下,我们需要增加一个参数chatId 对话ID，然后通过chatClient#advisors去指定对话ID，SpringAI会自动把消息和ChatID存储到数据库中

@RequestMapping(value = "/ai/chat/history")
public Flux<String> history(@RequestParam String prompt,@RequestParam String chatId){return chatClient.prompt().user(prompt).advisors(p->p.param(ChatMemory.CONVERSATION_ID,chatId)).stream().content();
}