WangLiZhao
525b391f09
- 在ModelGatewayTask实体中新增BuildModelName字段 - 修改ParseAndValidate函数参数，支持传入requiredFields - 在异步工作器中实现构建模型的动态字段校验逻辑 - 添加prompt构建服务中传递构建模型名称的功能 - 实现构建模型不存在时的兜底机制
80 lines
2.6 KiB
Go
80 lines
2.6 KiB
Go
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
|
|
BuildModelName 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",
|
|
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"`
|
|
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"`
|
|
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"`
|
|
}
|
|
|
|
// RequestPayload 请求参数结构体
|
|
type RequestPayload struct {
|
|
Headers map[string]string `json:"headers"`
|
|
Body map[string]any `json:"body"`
|
|
}
|