AI提供了强大的记忆管理功能#xff0c;本文将手把手教你如何配置Redis和MySQL两种不同的记忆…

在构建智能对话系统时记忆管理是决定用户体验的关键因素。

Spring

AI提供了强大的记忆管理功能本文将手把手教你如何配置Redis和MySQL两种不同的记忆存储方案让你的AI应用拥有持久的对话记忆能力

一、Spring

提供了多种实现对话记忆的方式以满足不同应用场景的需求。

其中常见的方式包括数据库存储️。

MySQL、Redis

数据库中。

MysqlChatMemoryRepository类通过JdbcTemplate与

MySQL

数据库进行交互实现了对话消息的存储和查询功能。

这种方式适用于需要长期保存对话历史的场景如客服系统的对话记录存档以便后续进行数据分析和服务质量评估等。

而使用

Redis

的高性能和高并发特性适合在对响应速度要求较高的场景中使用如实时聊天应用中能够快速地获取和更新对话历史记录保证对话的流畅性。

还支持其他存储方案并且提供了扩展接口允许开发者根据自己的需求实现自定义的对话记忆存储方式以满足更加复杂和多样化的应用场景需求。

二、代码实战Spring

dependencygroupIdcom.alibaba.cloud.ai/groupIdartifactIdspring-ai-alibaba-starter-dashscope/artifactIdversion1.0.0.1/version

/dependency

dependencygroupIdcom.alibaba.cloud.ai/groupIdartifactIdspring-ai-alibaba-starter-memory/artifactIdversion1.0.0.1/version

/dependencydependencygroupIdcom.alibaba.cloud.ai/groupIdartifactIdspring-ai-alibaba-starter-memory-jdbc/artifactIdversion1.0.0.1/version

/dependency

dependencygroupIdmysql/groupIdartifactIdmysql-connector-java/artifactIdversion8.0.31/version

/dependency

,集成的依赖spring-ai-alibaba-starter

MySQL

xxx-API_KEYchat:memory:repository:jdbc:mysql:driver-class-name:

com.mysql.cj.jdbc.Driverjdbc-url:

jdbc:mysql://localhost:3306/spring_ai?useUnicodetruecharacterEncodingutf-8useSSLfalseallowPublicKeyRetrievaltruezeroDateTimeBehaviorconvertToNulltransformedBitIsBooleantrueallowMultiQueriestruetinyInt1isBitfalseallowLoadLocalInfiletrueallowLocalInfiletrueallowUrlusername:

rootpassword:

spring.ai.chat.memory.repository.jdbc.mysql配置

MySQL

URL连接到本地的spring_ai数据库并设置了一些连接参数如字符编码、禁用

SSL

username和password分别指定数据库的用户名和密码为root

enabled设置为true表示启用

数据库建立连接并将对话记忆数据存储到数据库中以便长期保存和查询

2.3

com.atg.spring_ai.config;import

com.alibaba.cloud.ai.memory.jdbc.MysqlChatMemoryRepository;

import

org.springframework.beans.factory.annotation.Value;

import

org.springframework.boot.context.properties.ConfigurationProperties;

import

org.springframework.context.annotation.Bean;

import

org.springframework.context.annotation.Configuration;

import

org.springframework.jdbc.core.JdbcTemplate;

import

org.springframework.jdbc.datasource.DriverManagerDataSource;import

author:

数据库驱动Value(${spring.ai.chat.memory.repository.jdbc.mysql.jdbc-url})private

String

mysqlJdbcUrl;Value(${spring.ai.chat.memory.repository.jdbc.mysql.username})private

String

mysqlUsername;Value(${spring.ai.chat.memory.repository.jdbc.mysql.password})private

String

mysqlPassword;Value(${spring.ai.chat.memory.repository.jdbc.mysql.driver-class-name})private

String

mysqlDriverClassName;Beanpublic

dataSource

DriverManagerDataSource();dataSource.setDriverClassName(mysqlDriverClassName);dataSource.setUrl(mysqlJdbcUrl);dataSource.setUsername(mysqlUsername);dataSource.setPassword(mysqlPassword);JdbcTemplate

jdbcTemplate

JdbcTemplate(dataSource);return

MysqlChatMemoryRepository.mysqlBuilder().jdbcTemplate(jdbcTemplate).build();}

MySQL

连接并创建MysqlChatMemoryRepository实例用于将对话记忆数据存储到

MySQL

通过Value注解从application.yml配置文件中读取

MySQL

URL、用户名、密码和驱动类名并将其注入到相应的成员变量中。

在mysqlChatMemoryRepository方法中首先创建DriverManagerDataSource数据源对象设置其驱动类名、URL、用户名和密码从而建立与

MySQL

语句。

这里将刚刚创建的数据源dataSource传递给JdbcTemplate的构造函数使其能够使用该数据源进行数据库操作。

最后使用MysqlChatMemoryRepository的构建器模式通过mysqlBuilder()方法创建构建器设置jdbcTemplate并调用build()方法构建MysqlChatMemoryRepository实例。

这个实例将用于在后续的代码中进行对话记忆数据的存储和查询操作实现对话记忆的持久化功能。

2.3.2

com.atg.spring_ai.controller;import

com.alibaba.cloud.ai.memory.jdbc.MysqlChatMemoryRepository;

import

org.springframework.ai.chat.client.ChatClient;

import

org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;

import

org.springframework.ai.chat.memory.MessageWindowChatMemory;

import

org.springframework.ai.chat.messages.Message;

import

org.springframework.web.bind.annotation.GetMapping;

import

org.springframework.web.bind.annotation.RequestMapping;

import

org.springframework.web.bind.annotation.RequestParam;

import

org.springframework.web.bind.annotation.RestController;import

static

org.springframework.ai.chat.memory.ChatMemory.CONVERSATION_ID;/*

author:

messageWindowChatMemory;private

static

DashScopeChatMemoryController(ChatClient.Builder

builder,

创建MessageWindowChatMemory对象并设置chatMemoryRepository和maxMessages属性this.messageWindowChatMemory

MessageWindowChatMemory.builder().chatMemoryRepository(mysqlChatMemoryRepository).maxMessages(MAX_MESSAGES).build();//

创建ChatClient对象并设置defaultAdvisors属性this.chatClient

builder.defaultAdvisors(MessageChatMemoryAdvisor.builder(messageWindowChatMemory).build()).build();}GetMapping(/call)public

String

chatClient.prompt(query).advisors(a

conversationId)).call().content();}GetMapping(/messages)public

ListMessage

messageWindowChatMemory.get(conversationId);}}在这个控制器中定义了两个接口方法用于实现对话交互和获取对话历史

ID默认值为

“yingzi”。

通过chatClient.prompt(query)构建对话请求设置advisors参数其中a

conversationId)表示将对话

传递给MessageChatMemoryAdvisor以便它能够根据对话

管理对话记忆。

最后调用call().content()发起对话请求并获取模型返回的响应内容将其作为方法的返回值返回给前端实现了用户与模型的对话交互功能。

messages方法:

messageWindowChatMemory.get(conversationId)方法从MessageWindowChatMemory中获取指定对话

的对话历史记录返回一个包含Message对象的列表前端可以根据这些消息展示对话历史方便用户查看之前的对话内容

三、总结

。

开发者可以利用这些知识高效地构建智能对话应用实现从简单对话到复杂交互的多样化功能