updated createDeployTask apis

Former-commit-id: c45af4d03769819fa0ff4c2a8d6e83e7fd3b93dd
This commit is contained in:
tzwang 2024-08-29 11:10:28 +08:00
parent 7aad9ecd4d
commit 7c6389d3d8
4 changed files with 19 additions and 15 deletions

View File

@ -195,13 +195,13 @@ type (
}
GetAdaptersByModelResp {
Adapters []AdapterAvail `json:"adapters"`
Adapters []*AdapterAvail `json:"adapters"`
}
AdapterAvail {
AdapterId string `json:"adapterId"`
AdapterName string `json:"taskName"`
Clusters []ClusterAvail `json:"clusters"`
Clusters []*ClusterAvail `json:"clusters"`
}
ClusterAvail {

View File

@ -1,6 +1,7 @@
package inference
import (
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/repository/result"
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
@ -13,16 +14,12 @@ func CreateDeployTaskHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req types.CreateDeployTaskReq
if err := httpx.Parse(r, &req); err != nil {
httpx.ErrorCtx(r.Context(), w, err)
result.ParamErrorResult(r, w, err)
return
}
l := inference.NewCreateDeployTaskLogic(r.Context(), svcCtx)
resp, err := l.CreateDeployTask(&req)
if err != nil {
httpx.ErrorCtx(r.Context(), w, err)
} else {
httpx.OkJsonCtx(r.Context(), w, resp)
}
result.HttpResult(r, w, resp, err)
}
}

View File

@ -1,6 +1,7 @@
package inference
import (
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/repository/result"
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
@ -13,16 +14,12 @@ func GetAdaptersByModelHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req types.GetAdaptersByModelReq
if err := httpx.Parse(r, &req); err != nil {
httpx.ErrorCtx(r.Context(), w, err)
result.ParamErrorResult(r, w, err)
return
}
l := inference.NewGetAdaptersByModelLogic(r.Context(), svcCtx)
resp, err := l.GetAdaptersByModel(&req)
if err != nil {
httpx.ErrorCtx(r.Context(), w, err)
} else {
httpx.OkJsonCtx(r.Context(), w, resp)
}
result.HttpResult(r, w, resp, err)
}
}

View File

@ -2,6 +2,7 @@ package inference
import (
"context"
"errors"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
@ -24,7 +25,16 @@ func NewGetAdaptersByModelLogic(ctx context.Context, svcCtx *svc.ServiceContext)
}
func (l *GetAdaptersByModelLogic) GetAdaptersByModel(req *types.GetAdaptersByModelReq) (resp *types.GetAdaptersByModelResp, err error) {
// todo: add your logic here and delete this line
resp = &types.GetAdaptersByModelResp{}
adapterList, err := l.svcCtx.Scheduler.AiStorages.GetAdaptersByType("1")
if err != nil {
return nil, err
}
if len(adapterList) == 0 {
return nil, errors.New("适配器不存在")
}
return
}