diff --git a/client/types.go b/client/types.go index a2f8df49..5c494b26 100644 --- a/client/types.go +++ b/client/types.go @@ -136,20 +136,20 @@ type CloudInfo struct { } type AiInfo struct { - Id int64 `json:"id"` // id - AdapterId int64 `json:"adapterId,omitempty,optional"` - AdapterName string `json:"adapterName,omitempty,optional"` - ClusterId int64 `json:"clusterId,omitempty,optional"` - ClusterIds []int64 `json:"clusterIds,omitempty,optional"` - TaskId int64 `json:"taskId,omitempty"` - ClusterName string `json:"clusterName,omitempty,optional"` - ImageId string `json:"imageId,omitempty"` + Id int64 `json:"id"` // id + AdapterId int64 `json:"adapterId,omitempty,optional"` + AdapterName string `json:"adapterName,omitempty,optional"` + ClusterId int64 `json:"clusterId,omitempty,optional"` + //ClusterIds []int64 `json:"clusterIds,omitempty,optional"` + TaskId int64 `json:"taskId,omitempty"` + ClusterName string `json:"clusterName,omitempty,optional"` + ImageId string `json:"imageId,omitempty"` //ResourceId string `json:"resourceId,omitempty"` - AlgorithmId string `json:"algorithmId,omitempty"` + //AlgorithmId string `json:"algorithmId,omitempty"` //MetadataName string `json:"metadataName,omitempty"` - Command string `json:"command,omitempty"` - Environments string `json:"environments,omitempty"` - Parameters string `json:"parameters,omitempty"` + Command string `json:"command,omitempty"` + //Environments string `json:"environments,omitempty"` + //Parameters string `json:"parameters,omitempty"` Name string `json:"name,omitempty"` Status string `json:"status,omitempty"` diff --git a/pkg/models/taskaiasynchronousmodel_gen.go b/pkg/models/taskaiasynchronousmodel_gen.go index cca85a77..a0157e4a 100644 --- a/pkg/models/taskaiasynchronousmodel_gen.go +++ b/pkg/models/taskaiasynchronousmodel_gen.go @@ -48,7 +48,7 @@ type ( Result string `db:"result"` // 运行结果 DeletedAt string `db:"deleted_at"` // 删除时间 ImageId string `db:"image_id"` // 镜像id - Cmd string `db:"cmd"` // 命令行 + Command string `db:"command"` // 命令行 FlavorId string `db:"flavor_id"` // 训练作业资源规格id Status string `db:"status"` // 任务状态 } @@ -83,13 +83,13 @@ func (m *defaultTaskAiAsynchronousModel) FindOne(ctx context.Context, id int64) func (m *defaultTaskAiAsynchronousModel) Insert(ctx context.Context, data *TaskAiAsynchronous) (sql.Result, error) { query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiAsynchronousRowsExpectAutoSet) - ret, err := m.conn.ExecCtx(ctx, query, data.Id, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Cmd, data.FlavorId, data.Status) + ret, err := m.conn.ExecCtx(ctx, query, data.Id, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Command, data.FlavorId, data.Status) return ret, err } func (m *defaultTaskAiAsynchronousModel) Update(ctx context.Context, data *TaskAiAsynchronous) error { query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiAsynchronousRowsWithPlaceHolder) - _, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Cmd, data.FlavorId, data.Status, data.Id) + _, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Command, data.FlavorId, data.Status, data.Id) return err }