package entity

import "gitea.com/red-future/common/beans"

type ComposeTask struct {
	beans.SQLBaseDO `orm:",inline"`
	TaskId          string         `orm:"task_id" json:"taskId"`
	ModelName       string         `orm:"model_name" json:"modelName"`
	SkillName       string         `orm:"skill_name" json:"skillName"`
	BuildType       int            `orm:"build_type" json:"buildType"`
	CallbackUrl     string         `orm:"callback_url" json:"callbackUrl"`
	GatewayState    int            `orm:"gateway_state" json:"gatewayState"`
	RequestPayload  map[string]any `orm:"request_payload" json:"requestPayload"`
	ResultJson      map[string]any `orm:"result_json" json:"resultJson"`
	Status          string         `orm:"status" json:"status"`
	ErrorMessage    string         `orm:"error_message" json:"errorMessage"`
	OssFile         string         `orm:"oss_file" json:"ossFile"`
	FileType        string         `orm:"file_type" json:"fileType"`
}

type composeTaskCol struct {
	beans.SQLBaseCol
	TaskId         string
	ModelName      string
	SkillName      string
	BuildType      string
	CallbackUrl    string
	GatewayState   string
	RequestPayload string
	ResultJson     string
	Status         string
	ErrorMessage   string
	OssFile        string
	FileType       string
}

var ComposeTaskCol = composeTaskCol{
	SQLBaseCol:     beans.DefSQLBaseCol,
	TaskId:         "task_id",
	ModelName:      "model_name",
	SkillName:      "skill_name",
	BuildType:      "build_type",
	CallbackUrl:    "callback_url",
	GatewayState:   "gateway_state",
	RequestPayload: "request_payload",
	ResultJson:     "result_json",
	Status:         "status",
	ErrorMessage:   "error_message",
	OssFile:        "oss_file",
	FileType:       "file_type",
}