package entity

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

type modelGatewayBuildRecordCol struct {
	beans.SQLBaseCol
	TaskID          string
	BuildType       string
	ModelName       string
	SkillName       string
	SessionID       string
	NodeID          string
	RequestMessages string
	ResultMessages  string
	Status          string
	ErrorMsg        string
	DurationSeconds string
	CallbackURL     string
}

var ModelGatewayBuildRecordCol = modelGatewayBuildRecordCol{
	SQLBaseCol:      beans.DefSQLBaseCol,
	TaskID:          "task_id",
	BuildType:       "build_type",
	ModelName:       "model_name",
	SkillName:       "skill_name",
	SessionID:       "session_id",
	NodeID:          "node_id",
	RequestMessages: "request_messages",
	ResultMessages:  "result_messages",
	Status:          "status",
	ErrorMsg:        "error_msg",
	DurationSeconds: "duration_seconds",
	CallbackURL:     "callback_url",
}

// ModelGatewayBuildRecord 构建记录
type ModelGatewayBuildRecord struct {
	beans.SQLBaseDO `orm:",inline"`
	TaskID          string           `orm:"task_id" json:"taskId"`
	BuildType       int64            `orm:"build_type" json:"buildType"`
	ModelName       string           `orm:"model_name" json:"modelName"`
	SkillName       string           `orm:"skill_name" json:"skillName"`
	SessionID       string           `orm:"session_id" json:"sessionId"`
	NodeID          string           `orm:"node_id" json:"nodeId"`
	RequestMessages map[string]any   `orm:"request_messages" json:"requestMessages"`
	ResultMessages  []map[string]any `orm:"result_messages" json:"resultMessages"`
	Status          int              `orm:"status" json:"status"`
	ErrorMsg        string           `orm:"error_msg" json:"errorMsg"`
	DurationSeconds int              `orm:"duration_seconds" json:"durationSeconds"`
	CallbackURL     string           `orm:"callback_url" json:"callbackUrl"`
}