From a38cb5d4b513f4e3da0162c9161a4055f3e44d68 Mon Sep 17 00:00:00 2001 From: devad Date: Mon, 30 Oct 2023 14:52:36 +0800 Subject: [PATCH] :bug: fix modelarts Signed-off-by: devad Former-commit-id: 7b4aa9ed34cdf1eb808b727f57be5985a09f3990 --- api/internal/storeLink/modelarts.go | 7 ++++--- api/internal/storeLink/storeLink.go | 6 ++++-- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/api/internal/storeLink/modelarts.go b/api/internal/storeLink/modelarts.go index bef3ad4f..4a0d3461 100644 --- a/api/internal/storeLink/modelarts.go +++ b/api/internal/storeLink/modelarts.go @@ -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 } diff --git a/api/internal/storeLink/storeLink.go b/api/internal/storeLink/storeLink.go index 4e743ac6..883aa176 100644 --- a/api/internal/storeLink/storeLink.go +++ b/api/internal/storeLink/storeLink.go @@ -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