package com.pavis.admin.aigc.model.resp;

import com.pavis.admin.aigc.enums.AgentState;
import com.pavis.admin.common.model.resp.BaseResp;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.io.Serial;
import java.time.LocalDateTime;

/**
 * 智能体规划步骤记录信息
 *
 * @author semi
 * @since 2025/05/29 12:50
 */
@Data
@Schema(description = "智能体规划步骤记录信息")
public class AgentPlanStepResp extends BaseResp {

    @Serial
    private static final long serialVersionUID = 1L;

    /**
     * 计划ID
     */
    @Schema(description = "计划ID")
    private Long planId;

    /**
     * 所属会话ID
     */
    @Schema(description = "所属会话ID")
    private Long conversationId;

    /**
     * 对话消息Id
     */
    @Schema(description = "对话消息Id")
    private Long messageId;

    /**
     * 对话消息内容
     */
    @Schema(description = "对话消息内容")
    private String messageContent;

    /**
     * 计划主题,大模型根据用户消息生成
     */
    @Schema(description = "计划主题,大模型根据用户消息生成")
    private String title;

    /**
     * 计划描述:[BROWSER_AGENT] 打开百度主页
     */
    @Schema(description = "计划描述:[BROWSER_AGENT] 打开百度主页")
    private String description;

    /**
     * 步骤序号
     */
    @Schema(description = "步骤序号")
    private Integer order;

    /**
     * 本次计划关联的智能体ID
     */
    @Schema(description = "本次计划关联的智能体ID")
    private Long agentId;

    /**
     * 本次计划要使用的智能体:[BROWSER_AGENT]
     */
    @Schema(description = "本次计划要使用的智能体:[BROWSER_AGENT]")
    private String agentCode;

    /**
     * 智能体详情
     */
    @Schema(description = "本次计划要使用的智能体:[BROWSER_AGENT]")
    private AgentResp agent;

    /**
     * 步骤结果
     */
    @Schema(description = "步骤结果")
    private String result;

    /**
     * 修改人
     */
    @Schema(description = "修改人")
    private Long updateUser;

    /**
     * 修改时间
     */
    @Schema(description = "修改时间")
    private LocalDateTime updateTime;

    public AgentState getAgentRuntimeStatus() {
        return agent == null ? AgentState.NOT_STARTED : AgentState.valueOf(agent.getRuntimeStatus());
    }
}