package com.vincent.rsf.server.ai.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.experimental.Accessors;
import org.springframework.format.annotation.DateTimeFormat;

import java.io.Serializable;
import java.text.SimpleDateFormat;
import java.util.Date;

@Data
@Accessors(chain = true)
@TableName("sys_ai_param")
public class AiParam implements Serializable {

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value = "ID")
    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    @ApiModelProperty(value = "名称")
    private String name;

    @ApiModelProperty(value = "提供方类型")
    private String providerType;

    @ApiModelProperty(value = "基础地址")
    private String baseUrl;

    @ApiModelProperty(value = "密钥")
    private String apiKey;

    @ApiModelProperty(value = "模型")
    private String model;

    @ApiModelProperty(value = "temperature")
    private Double temperature;

    @ApiModelProperty(value = "topP")
    private Double topP;

    @ApiModelProperty(value = "maxTokens")
    private Integer maxTokens;

    @ApiModelProperty(value = "timeoutMs")
    private Integer timeoutMs;

    @ApiModelProperty(value = "streamingEnabled")
    private Boolean streamingEnabled;

    @ApiModelProperty(value = "状态")
    private Integer status;

    @ApiModelProperty(value = "是否删除")
    private Integer deleted;

    @ApiModelProperty(value = "租户")
    private Long tenantId;

    @ApiModelProperty(value = "添加人员")
    private Long createBy;

    @ApiModelProperty(value = "添加时间")
    @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
    private Date createTime;

    @ApiModelProperty(value = "修改人员")
    private Long updateBy;

    @ApiModelProperty(value = "修改时间")
    @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
    private Date updateTime;

    @ApiModelProperty(value = "备注")
    private String memo;

    public String getProviderType$() {
        return this.providerType;
    }

    public Boolean getStatusBool() {
        if (this.status == null) {
            return null;
        }
        return this.status == 1;
    }

    public String getCreateTime$() {
        if (this.createTime == null) {
            return "";
        }
        return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(this.createTime);
    }

    public String getUpdateTime$() {
        if (this.updateTime == null) {
            return "";
        }
        return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(this.updateTime);
    }
}