🐛 fix modelarts
Signed-off-by: devad <cossjie@foxmail.com> Former-commit-id: 7b4aa9ed34cdf1eb808b727f57be5985a09f3990
This commit is contained in:
parent
61d1870d58
commit
a38cb5d4b5
|
@ -6,6 +6,7 @@ import (
|
|||
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/models"
|
||||
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils"
|
||||
"gitlink.org.cn/jcce-pcm/pcm-participant-modelarts/modelarts"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
|
@ -43,8 +44,8 @@ func (o *ModelArtsLink) DeleteImage(imageId string) (interface{}, error) {
|
|||
func (o *ModelArtsLink) QueryImageList() (interface{}, error) {
|
||||
// modelArts获取镜像列表
|
||||
req := &modelarts.ListRepoReq{
|
||||
Offset: string(o.pageIndex),
|
||||
Limit: string(o.pageSize),
|
||||
Offset: "0",
|
||||
Limit: strconv.Itoa(int(o.pageSize)),
|
||||
}
|
||||
resp, err := o.svcCtx.ModelArtsImgRpc.ListReposDetails(o.ctx, req)
|
||||
if err != nil {
|
||||
|
@ -114,7 +115,7 @@ func (o *ModelArtsLink) QueryTask(taskId string) (interface{}, error) {
|
|||
}
|
||||
|
||||
//转换成统一返回类型
|
||||
taskResp, err := ConvertType[modelarts.JobResponse](resp, nil)
|
||||
taskResp, err := ConvertType[modelarts.JobResponse](resp, o.participant)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
|
|
@ -191,7 +191,7 @@ func ConvertType[T any](in *T, participant *models.StorelinkCenter) (interface{}
|
|||
resp.Success = false
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
resp.Success = true
|
||||
resp.TaskId = inresp.Metadata.Id
|
||||
|
||||
return resp, nil
|
||||
|
@ -231,7 +231,8 @@ func ConvertType[T any](in *T, participant *models.StorelinkCenter) (interface{}
|
|||
resp.Success = false
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
resp.Success = true
|
||||
resp.Task = &types.TaskSl{}
|
||||
resp.Task.TaskId = inresp.Metadata.Id
|
||||
resp.Task.TaskName = inresp.Metadata.Name
|
||||
resp.Task.StartedAt = int64(inresp.Status.StartTime)
|
||||
|
@ -274,6 +275,7 @@ func ConvertType[T any](in *T, participant *models.StorelinkCenter) (interface{}
|
|||
resp.Success = false
|
||||
return resp, nil
|
||||
}
|
||||
resp.Success = true
|
||||
return resp, nil
|
||||
case *hpcAC.DeleteTaskAiResp:
|
||||
var resp types.DeleteLinkTaskResp
|
||||
|
|
Loading…
Reference in New Issue