package cn.iocoder.yudao.module.ai.dal.dataobject.model;

import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
import cn.iocoder.yudao.module.ai.enums.model.AiModelTypeEnum;
import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
import com.baomidou.mybatisplus.annotation.KeySequence;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.*;

/**
 * AI 模型 DO
 *
 * @author fansili
 */
@TableName("ai_model")
@KeySequence("ai_model_seq")
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiModelDO extends BaseDO {

    @TableId
    private Long id;

    /**
     * API 秘钥编号，关联 {@link AiApiKeyDO#getId()}
     */
    private Long keyId;

    private String name;

    private String model;

    /**
     * 平台，枚举 {@link AiPlatformEnum}
     */
    private String platform;

    /**
     * 类型，枚举 {@link AiModelTypeEnum}
     */
    private Integer type;

    private Integer sort;

    /**
     * 状态，枚举 {@link CommonStatusEnum}
     */
    private Integer status;

    private Double temperature;

    private Integer maxTokens;

    private Integer maxContexts;

}