refactor(service): 重构模型网关服务结构

2026-06-11 17:58:49 +08:00
parent afd60caf56
commit 1c6c9bae14
34 changed files with 784 additions and 1223 deletions
--- a/model/entity/model_gateway_task.go
+++ b/model/entity/model_gateway_task.go
@@ -0,0 +1,76 @@
+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"`
+}