Browse Source

🐛 fix modelarts

Signed-off-by: devad <cossjie@foxmail.com>

Former-commit-id: 7b4aa9ed34
pull/9/head
devad 2 years ago
parent
commit
a38cb5d4b5
2 changed files with 8 additions and 5 deletions
  1. +4
    -3
      api/internal/storeLink/modelarts.go
  2. +4
    -2
      api/internal/storeLink/storeLink.go

+ 4
- 3
api/internal/storeLink/modelarts.go View File

@@ -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
}


+ 4
- 2
api/internal/storeLink/storeLink.go View File

@@ -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…
Cancel
Save