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
	RetryCount      string
	ErrorMsg        string
	ResultFile      string
	RequestPayload  string
	DurationSeconds string
	EpicycleId      string
	BillingData     string
	BuildModelName  string
}

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

// 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"`
	RetryCount      int              `orm:"retry_count" json:"retryCount"`
	ErrorMsg        string           `orm:"error_msg" json:"errorMsg"`
	ResultFile      *ResultFile      `orm:"result_file" json:"resultFile"`
	RequestPayload  map[string]any   `orm:"request_payload" json:"requestPayload"`
	DurationSeconds int64            `orm:"duration_seconds" json:"durationSeconds"`
	EpicycleId      int64            `orm:"epicycle_id" json:"epicycleId"`
	BillingData     []map[string]any `orm:"billing_data" json:"billingData"`
	BuildModelName  string           `orm:"build_model_name" json:"buildModelName"`
}

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