Spring AI–快速入门3：上下文记忆持久化

简介

上篇文章： Spring AI–快速入门2：上下文记忆+会话隔离 – 自学精灵 实现了上下文记忆，数据存放在内存。

本文实现的功能：对话记录持久化、打断对话、删除对话记录。

数据库依赖

添加数据库依赖，关键依赖是：

<!--会话记忆-->
<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-chat-memory-repository-jdbc</artifactId>
    <version>1.1.0</version>
</dependency>
<!-- MySQL 驱动 -->
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.32</version>
</dependency>

整个pom.xml如下：

数据库配置

1.创建数据库

CREATE DATABASE `spring-ai` CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci;

2.配置SQL脚本

新建sql建表文件，放到此路径：resources/sql/schema-mysql.sql

3.application.yml

需要配置数据库信息、配置使用jdbc作为持久化方式（并指定建表sql文件）。

关键配置是：

spring:
  ai:
    chat:
      memory:
        repository:
          jdbc:
            initialize-schema: always
            schema: classpath:sql/schema-mysql.sql
  
  datasource:
    url: jdbc:mysql://192.168.80.193:3306/spring-ai?characterEncoding=UTF-8&serverTimezone=Asia/Shanghai
    username: root
    password: 222333
    driver-class-name: com.mysql.cj.jdbc.Driver

整个application.yml配置如下：

实战1：对话持久化

启动

启动后会发现，数据库里自动生成了表：

测试

访问：http://localhost:8080/doc.html

第一次对话

页面操作

后台日志

数据库（插入了两条对话id为chat1的数据）

第二次对话

页面操作（可见：有记忆功能）

后台日志（可见：把前边的问题和答案给附带上了）

数据库结果（可见：插入了两条新数据）

实战2：打断对话

有时AI说得太多，我们需要让它适时地“闭嘴”。

程序调用AI大模型获得内容的过程中无法中断，只能是打断后端给前端的输出流。即：即使人为打断，大模型依然会输出，依然会产生费用。

基于上面的原理，我们就需要控制Flux流的输出即可。

代码

测试

开始对话：（可以发现：AI在持续的输出）

停止对话：（可以发现：输出被打断，而且输出了我们自定义的结束内容：“结束”）

查看数据库对话记录

实战3：删除对话记录

上边对话记录，都是自动添加的，那如何删除对话记录呢？

之前说过，ChatMemory控制增删改查，只需调用它的方法就可以。

ChatMemory的方法如下：

删除对话记录对应的就是：clear方法。

代码

测试

删除之前：

开始删除：

删除之后：（数据库相应的对话记录就没有了）

源码下载

本文所有源码：