Spring AI 实战：聊天记忆之带记忆能力的智能行程规划 Agent 开发

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>3.5.3</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.example</groupId>
    <artifactId>SpringAIAgentMemoryDemo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>SpringAIAgentMemoryDemo</name>
    <description>带记忆能力的 Spring AI Agent 入门案例</description>
    <properties>
        <java.version>17</java.version>
    </properties>
    <!-- 导入 Spring AI BOM 统一管理版本 -->
    <dependencyManagement>
        <<dependencies>
            <dependency>
                <groupId>org.springframework.ai</groupId>
                <artifactId>spring-ai-bom</artifactId>
                <version>1.0.0</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
        </</dependencies>
    </dependencyManagement>
    <<dependencies>
        <!-- Spring Web 依赖 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!-- 智普 AI 依赖 -->
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-starter-model-zhipuai</artifactId>
        </dependency>
        <!-- Lombok 依赖 -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
    </</dependencies>
    <!-- 仓库配置 -->
    <repositories>
        <repository>
            <id>spring-milestones</id>
            <name>Spring Milestones</name>
            <url>https://repo.spring.io/milestone</url>
            <releases>
                <enabled>true</enabled>
            </releases>
        </repository>
    </repositories>
    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
                <configuration>
                    <excludes>
                        <exclude>
                            <groupId>org.projectlombok</groupId>
                            <artifactId>lombok</artifactId>
                        </exclude>
                    </excludes>
                </configuration>
            </plugin>
        </plugins>
    </build>
</project>

# 应用基础配置
spring.application.name=SpringAIAgentMemoryDemo
server.port=8085
# 智普 AI 配置
spring.ai.zhipuai.api-key=你的智普 AI API Key
spring.ai.zhipuai.base-url=https://open.bigmodel.cn/api/paas
spring.ai.zhipuai.chat.options.model=GLM-4-Flash
spring.ai.zhipuai.chat.options.temperature=0.7
# 日志配置（便于调试）
logging.level.org.springframework.ai=INFO
logging.level.com.example=DEBUG

package com.example.springaiagentmemorydemo.config;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.ai.chat.memory.MessageWindowChatMemory;
import org.springframework.ai.chat.memory.repository.InMemoryChatMemoryRepository;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MemoryConfig {
    /**
     * 配置内存型聊天记忆：
     * 1. 存储介质：InMemoryChatMemoryRepository（内存存储，适合测试）
     * 2. 记忆窗口：最多保留 20 条消息（避免记忆过载）
     */
    @Bean
    public ChatMemory chatMemory() {
        return MessageWindowChatMemory.builder()
                .chatMemoryRepository(new InMemoryChatMemoryRepository()) // 内存存储
                .maxMessages(20) // 最大记忆条数
                .build();
    }
}

package com.example.springaiagentmemorydemo.service;
import lombok.RequiredArgsConstructor;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.stereotype.Service;
@Service
@RequiredArgsConstructor
public class MemoryTripAgentService {
    private final ChatClient chatClient;
    private final ChatMemory chatMemory;
    /**
     * 带记忆的行程规划：通过 userId 关联用户记忆
     * @param userId 用户唯一标识（隔离不同用户记忆）
     * @param tripDemand 出行需求（可包含新需求或仅查询）
     * @return 个性化行程规划
     */
    public String planTripWithMemory(String userId, String tripDemand) {
        // 定义带记忆的 Agent 行为规则：提取并复用用户偏好
        String systemPrompt = """
                你是带记忆的智能行程规划 Agent，核心规则如下：
                1. 优先从历史对话中提取用户偏好（景点类型、饮食禁忌、交通方式、出行人数等）；
                2. 若用户未重复说明偏好，自动复用记忆中的信息；若有新偏好，覆盖旧记忆；
                3. 行程按天/时段拆分，包含景点、交通、餐饮、实用提示，贴合用户偏好；
                4. 语言简洁明了，结构清晰，无需重复用户已说明的偏好。
                """;
        // 构建带记忆的 ChatClient：通过 MessageChatMemoryAdvisor 注入记忆
        ChatClient memoryChatClient = chatClient.builder()
                .defaultAdvisors(MessageChatMemoryAdvisor.builder(chatMemory).build())
                .defaultSystem(systemPrompt)
                .build();
        // 调用大模型：通过 conversationId（即 userId）关联用户记忆
        return memoryChatClient.prompt()
                .user(tripDemand)
                .advisors(advisor -> advisor.param(ChatMemory.CONVERSATION_ID, userId)) // 用 userId 隔离记忆
                .call()
                .content();
    }
    /**
     * 清除用户记忆（可选功能）
     */
    public void clearUserMemory(String userId) {
        chatMemory.clear(userId);
    }
}

package com.example.springaiagentmemorydemo.controller;
import com.example.springaiagentmemorydemo.service.MemoryTripAgentService;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import java.util.Map;
@RestController
@RequestMapping("/agent/trip")
@RequiredArgsConstructor
public class AgentController {
    private final MemoryTripAgentService memoryTripAgentService;
    /**
     * 带记忆的行程规划接口
     * @param userId 用户唯一标识（如 1001、1002）
     * @param demand 出行需求
     */
    @GetMapping("/plan-with-memory")
    public Map<String, String> planTripWithMemory(
            @RequestParam("userId") String userId,
            @RequestParam("demand") String demand) {
        String tripPlan = memoryTripAgentService.planTripWithMemory(userId, demand);
        return Map.of(
                "userId", userId,
                "userDemand", demand,
                "tripPlan", tripPlan,
                "agentType", "带记忆的智能行程规划 Agent"
        );
    }
    /**
     * 清除用户记忆接口（可选）
     */
    @DeleteMapping("/clear-memory")
    public Map<String, String> clearMemory(@RequestParam("userId") String userId) {
        memoryTripAgentService.clearUserMemory(userId);
        return Map.of(
                "userId", userId,
                "message", "用户记忆已清除"
        );
    }
}

{
  "userDemand": "周末两天成都短途游，2人，偏好自然景点，不吃辣，交通以地铁和打车为主",
  "tripPlan": "根据您的历史偏好，我为您规划了以下成都周末两天短途游行程：\n\n**第一天：**\n\n- **上午**\n  - 9:00 AM：乘坐地铁2号线至“人民公园”站，游览人民公园，体验成都的悠闲生活。\n  - 11:00 AM：步行至“武侯祠”附近，品尝不辣的川味小吃，如担担面、甜水面等。\n\n- **下午**\n  - 12:30 PM：前往“青羊宫”，参观古建筑，感受道教文化。\n  - 2:30 PM：乘坐地铁4号线至“宽窄巷子”站，游览宽窄巷子，享受悠闲的下午时光。\n\n- **晚上**\n  - 5:00 PM：在宽窄巷子附近寻找不辣的晚餐。\n  - 7:00 PM：乘坐地铁4号线返回酒店休息。\n\n**第二天：**\n\n- **上午**\n  - 9:00 AM：乘坐地铁3号线至“熊猫大道”站，游览成都大熊猫繁育研究基地，近距离观察大熊猫。\n\n- **下午**\n  - 12:00 PM：在基地附近享用午餐，避免辣味。\n  - 2:00 PM：乘坐地铁3号线至“杜甫草堂”站，参观杜甫草堂，了解唐代诗人杜甫的生活。\n\n- **晚上**\n  - 5:00 PM：在草堂附近寻找不辣的晚餐。\n  - 7:00 PM：根据返程时间，选择合适的交通方式返回酒店。\n\n**实用提示：**\n- 请随身携带雨具，成都天气多变。\n- 保持手机电量充足，以便导航和拍照。\n- 注意个人财物安全。\n\n希望这个行程符合您的喜好，祝您旅途愉快！",
  "agentType": "带记忆的智能行程规划 Agent",
  "userId": "1001"
}

{
  "userDemand": "下周工作日三天，再规划一次成都周边游",
  "tripPlan": "根据您之前提到的偏好，以下是为您规划的成都周边三天工作日游行程。此行程将优先考虑自然景点，避免辣味食物，并主要使用地铁和打车作为交通方式。\n\n**第一天：都江堰一日游**\n\n- **上午**\n  - 8:00 AM：从成都出发，乘坐地铁2号线转乘成灌快铁至都江堰站。\n  - 9:00 AM：抵达都江堰后，游览都江堰水利工程，了解中国古代水利建筑的智慧。\n\n- **中午**\n  - 12:00 PM：在都江堰附近寻找不辣的川菜馆享用午餐。\n\n- **下午**\n  - 1:30 PM：参观青城山，感受道教文化，攀登青城山，享受户外活动。\n\n- **傍晚**\n  - 5:00 PM：下山后，返回成都市区。\n\n**第二天：峨眉山一日游**\n\n- **上午**\n  - 7:00 AM：从成都出发，乘坐高铁至峨眉山站。\n  - 8:00 AM：抵达峨眉山后，游览报国寺，了解寺庙文化。\n\n- **中午**\n  - 12:00 PM：在峨眉山附近找一家不辣的餐厅用餐。\n\n- **下午**\n  - 1:30 PM：游览峨眉山金顶，观赏云海和日出（视天气而定）。\n\n- **傍晚**\n  - 5:00 PM：返回峨眉山站，乘坐高铁返回成都市区。\n\n**第三天：乐山大佛一日游**\n\n- **上午**\n  - 7:00 AM：从成都出发，乘坐汽车或高铁至乐山站。\n  - 8:00 AM：抵达乐山后，游览乐山大佛，欣赏世界最大的石刻佛像。\n\n- **中午**\n  - 12:00 PM：在乐山城区寻找不辣的午餐。\n\n- **下午**\n  - 1:30 PM：参观乐山大佛附近的乌尤寺，了解佛教文化。\n\n- **傍晚**\n  - 5:00 PM：结束行程，返回成都市区。\n\n**实用提示：**\n- 由于峨眉山和乐山都在成都周边，建议提前规划好交通和住宿。\n- 高铁和汽车都是不错的选择，具体根据您的出发时间来决定。\n- 请随身携带个人必需品，如防晒霜、雨具等。\n- 由于峨眉山海拔较高，注意防寒保暖。\n\n希望这个行程能满足您的需求，祝您旅途愉快！",
  "agentType": "带记忆的智能行程规划 Agent",
  "userId": "1001"
}

{
  "userDemand": "周末一天重庆游，1人，喜欢人文景点，吃辣，地铁出行",
  "tripPlan": "根据您的历史偏好，我为您规划了以下重庆一天的行程：\n\n**行程安排：**\n\n**上午：**\n- 9:00 AM - 江北观音桥步行街\n  - 体验重庆的繁华商业区，品尝当地特色小吃，如酸辣粉、串串香等。\n\n**中午：**\n- 12:00 PM - 地铁2号线到解放碑站\n  - 沿途欣赏重庆的老街风貌。\n- 12:30 PM - 解放碑小吃街\n  - 尝试更多重庆特色美食，如辣子鸡、辣子牛蛙等。\n\n**下午：**\n- 2:00 PM - 重庆人民大礼堂\n  - 参观这座具有重庆特色的人文建筑，了解其历史和建筑风格。\n\n**傍晚：**\n- 4:00 PM - 重庆洪崖洞\n  - 沿着长江欣赏洪崖洞的夜景，感受重庆的夜景魅力。\n\n**晚餐：**\n- 7:00 PM - 长江索道\n  - 在索道上享用晚餐，一边欣赏江景一边品尝重庆美食。\n\n**晚上：**\n- 8:30 PM - 重庆地铁3号线到磁器口站\n  - 前往磁器口古镇，感受古街的历史韵味。\n\n**实用提示：**\n- 请注意地铁运营时间，确保有足够的时间在各个景点间移动。\n- 由于您喜欢辣食，建议在尝试当地美食时注意自己的口味承受能力。\n- 携带雨具，重庆天气多变，可能会有阵雨。\n\n希望这个行程能符合您的喜好，祝您在重庆有一个愉快的周末！",
  "agentType": "带记忆的智能行程规划 Agent",
  "userId": "1002"
}