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
	Phase           string
	ErrorMsg        string
	ResultFile      string
	TextResult      string
	ExpendTokens    string
	DurationSeconds string
	RetryCount      string
	TmpFile         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",
	Phase:           "phase",
	ErrorMsg:        "error_msg",
	ResultFile:      "result_file",
	TextResult:      "text_result",
	ExpendTokens:    "expend_tokens",
	DurationSeconds: "duration_seconds",
	RetryCount:      "retry_count",
	TmpFile:         "tmp_file",
	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"`
	Phase           int             `orm:"phase" json:"phase"`
	ErrorMsg        string          `orm:"error_msg" json:"errorMsg"`
	ResultFile      *ResultFile     `orm:"result_file" json:"resultFile"`
	TextResult      map[string]any  `orm:"text_result" json:"text"`
	ExpendTokens    int64           `orm:"expend_tokens" json:"expendTokens"`
	DurationSeconds int64           `orm:"duration_seconds" json:"durationSeconds"`
	RetryCount      int             `orm:"retry_count" json:"retryCount"`
	TmpFile         string          `orm:"tmp_file" json:"tmpFile"`
	RequestPayload  *RequestPayload `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"`
}

// RequestPayload 请求参数结构体
type RequestPayload struct {
	Headers map[string]string `json:"headers"`
	Body    map[string]any    `json:"body"`
}