package entity

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

type composeTaskCol struct {
	beans.SQLBaseCol
	TaskId          string
	ModelName       string
	SkillName       string
	LimitWords      string
	RequestPayload  string
	CallbackPayload string
	ModelResult     string
	Messages        string
	Status          string
	ErrorMessage    string
}

var ComposeTaskCol = composeTaskCol{
	SQLBaseCol:      beans.DefSQLBaseCol,
	TaskId:          "task_id",
	ModelName:       "model_name",
	SkillName:       "skill_name",
	LimitWords:      "limit_words",
	RequestPayload:  "request_payload",
	CallbackPayload: "callback_payload",
	ModelResult:     "model_result",
	Messages:        "messages",
	Status:          "status",
	ErrorMessage:    "error_message",
}

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"`
	LimitWords      int    `orm:"limit_words" json:"limitWords"`
	RequestPayload  any    `orm:"request_payload" json:"requestPayload"`
	CallbackPayload any    `orm:"callback_payload" json:"callbackPayload"`
	ModelResult     any    `orm:"model_result" json:"modelResult"`
	Messages        any    `orm:"messages" json:"messages"`
	Status          string `orm:"status" json:"status"`
	ErrorMessage    string `orm:"error_message" json:"errorMessage"`
}