package entity

import (
	"gitea.redpowerfuture.com/red-future/common/beans"
)

type modelGatewayTaskCol struct {
	beans.SQLBaseCol
	ModelName       string
	TaskID          string
	BizName         string
	CallbackURL     string
	State           string
	ErrorMsg        string
	ResultFile      string
	ExpendTokens    string
	DurationSeconds string
	RetryCount      string
	BuildType       string
	RequestPayload  string
	EpicycleId      string
}

var ModelGatewayTaskCol = modelGatewayTaskCol{
	SQLBaseCol:      beans.DefSQLBaseCol,
	ModelName:       "model_name",
	TaskID:          "task_id",
	BizName:         "biz_name",
	CallbackURL:     "callback_url",
	State:           "state",
	ErrorMsg:        "error_msg",
	ResultFile:      "result_file",
	ExpendTokens:    "expend_tokens",
	DurationSeconds: "duration_seconds",
	RetryCount:      "retry_count",
	BuildType:       "build_type",
	RequestPayload:  "request_payload",
	EpicycleId:      "epicycle_id",
}

// ModelGatewayTask 模型网关任务
type ModelGatewayTask struct {
	beans.SQLBaseDO `orm:",inline"`
	ModelName       string         `orm:"model_name" json:"modelName"`
	TaskID          string         `orm:"task_id" json:"taskId"`
	BizName         string         `orm:"biz_name" json:"bizName"`
	CallbackURL     string         `orm:"callback_url" json:"callbackUrl"`
	State           int            `orm:"state" json:"state"`
	BuildType       int64          `orm:"build_type" json:"buildType"`
	ErrorMsg        string         `orm:"error_msg" json:"errorMsg"`
	ResultFile      *ResultFile    `orm:"result_file" json:"resultFile"`
	ExpendTokens    int64          `orm:"expend_tokens" json:"expendTokens"`
	DurationSeconds int64          `orm:"duration_seconds" json:"durationSeconds"`
	RetryCount      int            `orm:"retry_count" json:"retryCount"`
	RequestPayload  map[string]any `orm:"request_payload" json:"requestPayload"`
	EpicycleId      int64          `orm:"epicycle_id" json:"epicycleId"`
}

// ResultFile OSS 结果文件
type ResultFile struct {
	OssFile  string `json:"ossFile"`
	FileType string `json:"fileType"`
	FileSize int64  `json:"fileSize"`
}