diff --git a/model/task.go b/model/task.go index a8c3a7d4d..6c5ade536 100644 --- a/model/task.go +++ b/model/task.go @@ -72,7 +72,10 @@ func (t *Task) GetData(v any) error { } type Properties struct { - Input string `json:"input"` + Input string `json:"input"` + IsModelMapped bool `json:"is_model_mapped,omitempty"` + OriginModelName string `json:"origin_model_name,omitempty"` + UpstreamModelName string `json:"upstream_model_name,omitempty"` } func (m *Properties) Scan(val interface{}) error { diff --git a/relay/relay_task.go b/relay/relay_task.go index db543319a..148b4ac38 100644 --- a/relay/relay_task.go +++ b/relay/relay_task.go @@ -214,6 +214,11 @@ func RelayTaskSubmit(c *gin.Context, info *relaycommon.RelayInfo) (taskErr *dto. task.Quota = quota task.Data = taskData task.Action = info.Action + task.Properties = model.Properties{ + IsModelMapped: info.IsModelMapped, + UpstreamModelName: info.UpstreamModelName, + OriginModelName: info.OriginModelName, + } err = task.Insert() if err != nil { taskErr = service.TaskErrorWrapper(err, "insert_task_failed", http.StatusInternalServerError)