package com.easyagents.agent.runtime.memory;

import com.easyagents.agent.runtime.message.AgentMessage;

import java.util.ArrayList;
import java.util.List;

/**
 * 由调用方恢复的会话历史快照。
 */
public class AgentMemorySnapshot {

    private List<AgentMessage> messages = new ArrayList<>();

    /**
     * 获取消息列表。
     *
     * @return 消息列表
     */
    public List<AgentMessage> getMessages() {
        return messages;
    }

    /**
     * 设置消息列表。
     *
     * @param messages 消息列表
     */
    public void setMessages(List<AgentMessage> messages) {
        this.messages = messages == null ? new ArrayList<>() : messages;
    }

    /**
     * 添加消息。
     *
     * @param message 消息
     */
    public void addMessage(AgentMessage message) {
        if (message != null) {
            messages.add(message);
        }
    }
}
