package structs

import (
	"code.gitea.io/gitea/models"
	"code.gitea.io/gitea/modules/timeutil"
)

type CreateGrampusTrainJobOption struct {
	DisplayJobName   string `json:"display_job_name" binding:"Required"`
	JobName          string `json:"job_name" binding:"Required" `
	Attachment       string `json:"attachment" binding:"Required"`
	BootFile         string `json:"boot_file" binding:"Required"`
	ImageID          string `json:"image_id" binding:"Required"`
	Params           string `json:"run_para_list" binding:"Required"`
	Description      string `json:"description"`
	BranchName       string `json:"branch_name" binding:"Required"`
	EngineName       string `json:"engine_name" binding:"Required"`
	WorkServerNumber int    `json:"work_server_number" binding:"Required"`
	Image            string `json:"image" binding:"Required"`
	DatasetName      string `json:"dataset_name" binding:"Required"`
	ModelName        string `json:"model_name"`
	ModelVersion     string `json:"model_version"`
	CkptName         string `json:"ckpt_name"`
	LabelName        string `json:"label_names"`
	PreTrainModelUrl string `json:"pre_train_model_url"`
	SpecId           int64  `json:"spec_id" binding:"Required"`
}

type CreateTrainJobOption struct {
	Type             int    `json:"type" binding:"Required"`
	DisplayJobName   string `json:"display_job_name" binding:"Required"`
	ImageID          string `json:"image_id"`
	Image            string `json:"image" binding:"Required"`
	Attachment       string `json:"attachment" binding:"Required"`
	DatasetName      string `json:"dataset_name" binding:"Required"`
	Description      string `json:"description" `
	BootFile         string `json:"boot_file" binding:"Required"`
	BranchName       string `json:"branch_name" binding:"Required"`
	Params           string `json:"run_para_list" binding:"Required"`
	WorkServerNumber int    `json:"work_server_number" binding:"Required"`
	ModelName        string `json:"model_name"`
	ModelVersion     string `json:"model_version"`
	CkptName         string `json:"ckpt_name"`
	LabelName        string `json:"label_names"`
	PreTrainModelUrl string `json:"pre_train_model_url"`
	SpecId           int64  `json:"spec_id" binding:"Required"`
}

type Cloudbrain struct {
	ID               int64              `json:"id"`
	JobID            string             `json:"job_id"`
	JobType          string             `json:"job_type"`
	Type             int                `json:"type"`
	DisplayJobName   string             `json:"display_job_name"`
	Status           string             `json:"status"`
	CreatedUnix      timeutil.TimeStamp `json:"created_unix"`
	RepoID           int64              `json:"repo_id"`
	Duration         int64              `json:"duration"` //运行时长 单位秒
	TrainJobDuration string             `json:"train_job_duration"`
	ImageID          string             `json:"image_id"` //grampus image_id
	Image            string             `json:"image"`
	Uuid             string             `json:"uuid"` //数据集id
	DatasetName      string             `json:"dataset_name"`
	ComputeResource  string             `json:"compute_resource"` //计算资源，例如npu
	AiCenter         string             `json:"ai_center"`        //grampus ai center: center_id+center_name
	BranchName       string             `json:"branch_name"`      //分支名称
	Parameters       string             `json:"parameters"`       //传给modelarts的param参数
	BootFile         string             `json:"boot_file"`        //启动文件
	Description      string             `json:"description"`      //描述
	ModelName        string             `json:"model_name"`       //模型名称
	ModelVersion     string             `json:"model_version"`    //模型版本
	CkptName         string             `json:"ckpt_name"`        //权重文件名称

	VersionCount    int                `json:"-"` //任务的当前版本数量，不包括删除的
	IsLatestVersion string             `json:"-"` //是否是最新版本，1是，0否
	CommitID        string             `json:"-"` //提交的仓库代码id
	PreVersionName  string             `json:"-"` //父版本名称
	StartTime       timeutil.TimeStamp `json:"start_time"`
	EndTime         timeutil.TimeStamp `json:"end_time"`

	Spec *models.Specification `json:"spec"`
}