package org.dromara.ai.domain;

import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.io.Serial;
import java.io.Serializable;
import java.math.BigDecimal;
import java.util.Date;

/**
 * AI聊天记录对象 ai_chat_record
 *
 * @author pengles
 */
@Data
@EqualsAndHashCode
@TableName("ai_chat_record")
public class AiChatRecord implements Serializable {

    @Serial
    private static final long serialVersionUID = 1L;

    /**
     * ID
     */
    @TableId
    private Long id;

    /**
     * 会话ID
     */
    private String sessionId;

    /**
     * 用户ID
     */
    private Long userId;

    /**
     * Provider名称
     */
    private String providerName;

    /**
     * 模型名称
     */
    private String modelName;

    /**
     * 消息类型(USER/ASSISTANT)
     */
    private String messageType;

    /**
     * 消息内容
     */
    private String content;

    /**
     * 使用的Token数
     */
    private Integer tokensUsed;

    /**
     * 成本
     */
    private BigDecimal cost;

    /**
     * 响应时间(毫秒)
     */
    private Integer responseTime;

    /**
     * 是否流式响应(0否 1是)
     */
    private Integer isStream;

    /**
     * 错误信息
     */
    private String errorMessage;

    /**
     * 创建时间
     */
    private Date createTime;

}
