启智章鱼获取model相关接口

Former-commit-id: 9aaa4e746a6712eae3c74d7bbcd70690ce742291
This commit is contained in:
tzwang 2023-05-26 16:05:21 +08:00
parent 9061f23478
commit db1769c4ea
11 changed files with 2779 additions and 564 deletions

View File

@ -47,4 +47,9 @@ type OctopusApi struct {
GetDatasetApplyList string GetDatasetApplyList string
GetDatasetTypeList string GetDatasetTypeList string
GetTrainJobList string GetTrainJobList string
GetMyModelList string
GetModelVersionList string
DeleteMyModel string
DeleteModelVersion string
DownloadModelVersion string
} }

View File

@ -0,0 +1,48 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"PCM/common/tool"
"context"
"github.com/zeromicro/go-zero/core/logx"
)
type DeleteModelVersionLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewDeleteModelVersionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteModelVersionLogic {
return &DeleteModelVersionLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *DeleteModelVersionLogic) DeleteModelVersion(in *octopus.DeleteModelVersionReq) (*octopus.DeleteModelVersionResp, error) {
resp := &octopus.DeleteModelVersionResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.DeleteModelVersion
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("modelId", in.ModelId).
SetPathParam("version", in.Version).
SetResult(resp).
Delete(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -0,0 +1,48 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type DeleteMyModelLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewDeleteMyModelLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteMyModelLogic {
return &DeleteMyModelLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *DeleteMyModelLogic) DeleteMyModel(in *octopus.DeleteMyModelReq) (*octopus.DeleteMyModelResp, error) {
resp := &octopus.DeleteMyModelResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.DeleteMyModel
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("modelId", in.ModelId).
SetResult(resp).
Delete(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -0,0 +1,48 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"PCM/common/tool"
"context"
"github.com/zeromicro/go-zero/core/logx"
)
type DownloadModelVersionLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewDownloadModelVersionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DownloadModelVersionLogic {
return &DownloadModelVersionLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *DownloadModelVersionLogic) DownloadModelVersion(in *octopus.DownloadModelVersionReq) (*octopus.DownloadModelVersionResp, error) {
resp := &octopus.DownloadModelVersionResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.DownloadModelVersion
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("modelId", in.ModelId).
SetPathParam("version", in.Version).
SetQueryString("domain=" + in.Domain).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -0,0 +1,51 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"strconv"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type GetModelVersionListLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetModelVersionListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelVersionListLogic {
return &GetModelVersionListLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetModelVersionListLogic) GetModelVersionList(in *octopus.GetModelVersionListReq) (*octopus.GetModelVersionListResp, error) {
resp := &octopus.GetModelVersionListResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetModelVersionList
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("modelId", in.ModelId).
SetQueryString("pageIndex=" + strconv.Itoa(int(in.PageIndex))).
SetQueryString("pageSize=" + strconv.Itoa(int(in.PageSize))).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -0,0 +1,51 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"strconv"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type GetMyModelListLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetMyModelListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMyModelListLogic {
return &GetMyModelListLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// Model
func (l *GetMyModelListLogic) GetMyModelList(in *octopus.GetMyModelListReq) (*octopus.GetMyModelListResp, error) {
resp := &octopus.GetMyModelListResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetMyModelList
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetQueryString("pageIndex=" + strconv.Itoa(int(in.PageIndex))).
SetQueryString("pageSize=" + strconv.Itoa(int(in.PageSize))).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -191,6 +191,32 @@ func (s *OctopusServer) UploadImageConfirm(ctx context.Context, in *octopus.Uplo
return l.UploadImageConfirm(in) return l.UploadImageConfirm(in)
} }
// Model
func (s *OctopusServer) GetMyModelList(ctx context.Context, in *octopus.GetMyModelListReq) (*octopus.GetMyModelListResp, error) {
l := logic.NewGetMyModelListLogic(ctx, s.svcCtx)
return l.GetMyModelList(in)
}
func (s *OctopusServer) GetModelVersionList(ctx context.Context, in *octopus.GetModelVersionListReq) (*octopus.GetModelVersionListResp, error) {
l := logic.NewGetModelVersionListLogic(ctx, s.svcCtx)
return l.GetModelVersionList(in)
}
func (s *OctopusServer) DeleteMyModel(ctx context.Context, in *octopus.DeleteMyModelReq) (*octopus.DeleteMyModelResp, error) {
l := logic.NewDeleteMyModelLogic(ctx, s.svcCtx)
return l.DeleteMyModel(in)
}
func (s *OctopusServer) DeleteModelVersion(ctx context.Context, in *octopus.DeleteModelVersionReq) (*octopus.DeleteModelVersionResp, error) {
l := logic.NewDeleteModelVersionLogic(ctx, s.svcCtx)
return l.DeleteModelVersion(in)
}
func (s *OctopusServer) DownloadModelVersion(ctx context.Context, in *octopus.DownloadModelVersionReq) (*octopus.DownloadModelVersionResp, error) {
l := logic.NewDownloadModelVersionLogic(ctx, s.svcCtx)
return l.DownloadModelVersion(in)
}
// TrainJobService // TrainJobService
func (s *OctopusServer) CreateTrainJob(ctx context.Context, in *octopus.CreateTrainJobReq) (*octopus.CreateTrainJobResp, error) { func (s *OctopusServer) CreateTrainJob(ctx context.Context, in *octopus.CreateTrainJobReq) (*octopus.CreateTrainJobResp, error) {
l := logic.NewCreateTrainJobLogic(ctx, s.svcCtx) l := logic.NewCreateTrainJobLogic(ctx, s.svcCtx)

File diff suppressed because it is too large Load Diff

View File

@ -59,6 +59,12 @@ type OctopusClient interface {
DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error) DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error)
UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error) UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error)
UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error) UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error)
// Model
GetMyModelList(ctx context.Context, in *GetMyModelListReq, opts ...grpc.CallOption) (*GetMyModelListResp, error)
GetModelVersionList(ctx context.Context, in *GetModelVersionListReq, opts ...grpc.CallOption) (*GetModelVersionListResp, error)
DeleteMyModel(ctx context.Context, in *DeleteMyModelReq, opts ...grpc.CallOption) (*DeleteMyModelResp, error)
DeleteModelVersion(ctx context.Context, in *DeleteModelVersionReq, opts ...grpc.CallOption) (*DeleteModelVersionResp, error)
DownloadModelVersion(ctx context.Context, in *DownloadModelVersionReq, opts ...grpc.CallOption) (*DownloadModelVersionResp, error)
// TrainJobService // TrainJobService
CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error)
GetTrainJobList(ctx context.Context, in *GetTrainJobListReq, opts ...grpc.CallOption) (*GetTrainJobListResp, error) GetTrainJobList(ctx context.Context, in *GetTrainJobListReq, opts ...grpc.CallOption) (*GetTrainJobListResp, error)
@ -369,6 +375,51 @@ func (c *octopusClient) UploadImageConfirm(ctx context.Context, in *UploadImageC
return out, nil return out, nil
} }
func (c *octopusClient) GetMyModelList(ctx context.Context, in *GetMyModelListReq, opts ...grpc.CallOption) (*GetMyModelListResp, error) {
out := new(GetMyModelListResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetMyModelList", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetModelVersionList(ctx context.Context, in *GetModelVersionListReq, opts ...grpc.CallOption) (*GetModelVersionListResp, error) {
out := new(GetModelVersionListResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetModelVersionList", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) DeleteMyModel(ctx context.Context, in *DeleteMyModelReq, opts ...grpc.CallOption) (*DeleteMyModelResp, error) {
out := new(DeleteMyModelResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/DeleteMyModel", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) DeleteModelVersion(ctx context.Context, in *DeleteModelVersionReq, opts ...grpc.CallOption) (*DeleteModelVersionResp, error) {
out := new(DeleteModelVersionResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/DeleteModelVersion", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) DownloadModelVersion(ctx context.Context, in *DownloadModelVersionReq, opts ...grpc.CallOption) (*DownloadModelVersionResp, error) {
out := new(DownloadModelVersionResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/DownloadModelVersion", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) { func (c *octopusClient) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) {
out := new(CreateTrainJobResp) out := new(CreateTrainJobResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/CreateTrainJob", in, out, opts...) err := c.cc.Invoke(ctx, "/octopus.Octopus/CreateTrainJob", in, out, opts...)
@ -428,6 +479,12 @@ type OctopusServer interface {
DeleteImage(context.Context, *DeleteImageReq) (*DeleteImageResp, error) DeleteImage(context.Context, *DeleteImageReq) (*DeleteImageResp, error)
UploadImage(context.Context, *UploadImageReq) (*UploadImageResp, error) UploadImage(context.Context, *UploadImageReq) (*UploadImageResp, error)
UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error) UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error)
// Model
GetMyModelList(context.Context, *GetMyModelListReq) (*GetMyModelListResp, error)
GetModelVersionList(context.Context, *GetModelVersionListReq) (*GetModelVersionListResp, error)
DeleteMyModel(context.Context, *DeleteMyModelReq) (*DeleteMyModelResp, error)
DeleteModelVersion(context.Context, *DeleteModelVersionReq) (*DeleteModelVersionResp, error)
DownloadModelVersion(context.Context, *DownloadModelVersionReq) (*DownloadModelVersionResp, error)
// TrainJobService // TrainJobService
CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error) CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error)
GetTrainJobList(context.Context, *GetTrainJobListReq) (*GetTrainJobListResp, error) GetTrainJobList(context.Context, *GetTrainJobListReq) (*GetTrainJobListResp, error)
@ -537,6 +594,21 @@ func (UnimplementedOctopusServer) UploadImage(context.Context, *UploadImageReq)
func (UnimplementedOctopusServer) UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error) { func (UnimplementedOctopusServer) UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method UploadImageConfirm not implemented") return nil, status.Errorf(codes.Unimplemented, "method UploadImageConfirm not implemented")
} }
func (UnimplementedOctopusServer) GetMyModelList(context.Context, *GetMyModelListReq) (*GetMyModelListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetMyModelList not implemented")
}
func (UnimplementedOctopusServer) GetModelVersionList(context.Context, *GetModelVersionListReq) (*GetModelVersionListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetModelVersionList not implemented")
}
func (UnimplementedOctopusServer) DeleteMyModel(context.Context, *DeleteMyModelReq) (*DeleteMyModelResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteMyModel not implemented")
}
func (UnimplementedOctopusServer) DeleteModelVersion(context.Context, *DeleteModelVersionReq) (*DeleteModelVersionResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteModelVersion not implemented")
}
func (UnimplementedOctopusServer) DownloadModelVersion(context.Context, *DownloadModelVersionReq) (*DownloadModelVersionResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DownloadModelVersion not implemented")
}
func (UnimplementedOctopusServer) CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error) { func (UnimplementedOctopusServer) CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateTrainJob not implemented") return nil, status.Errorf(codes.Unimplemented, "method CreateTrainJob not implemented")
} }
@ -1150,6 +1222,96 @@ func _Octopus_UploadImageConfirm_Handler(srv interface{}, ctx context.Context, d
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _Octopus_GetMyModelList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetMyModelListReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetMyModelList(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetMyModelList",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetMyModelList(ctx, req.(*GetMyModelListReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetModelVersionList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetModelVersionListReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetModelVersionList(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetModelVersionList",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetModelVersionList(ctx, req.(*GetModelVersionListReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_DeleteMyModel_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteMyModelReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).DeleteMyModel(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/DeleteMyModel",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).DeleteMyModel(ctx, req.(*DeleteMyModelReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_DeleteModelVersion_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteModelVersionReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).DeleteModelVersion(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/DeleteModelVersion",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).DeleteModelVersion(ctx, req.(*DeleteModelVersionReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_DownloadModelVersion_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DownloadModelVersionReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).DownloadModelVersion(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/DownloadModelVersion",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).DownloadModelVersion(ctx, req.(*DownloadModelVersionReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_CreateTrainJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { func _Octopus_CreateTrainJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateTrainJobReq) in := new(CreateTrainJobReq)
if err := dec(in); err != nil { if err := dec(in); err != nil {
@ -1325,6 +1487,26 @@ var Octopus_ServiceDesc = grpc.ServiceDesc{
MethodName: "UploadImageConfirm", MethodName: "UploadImageConfirm",
Handler: _Octopus_UploadImageConfirm_Handler, Handler: _Octopus_UploadImageConfirm_Handler,
}, },
{
MethodName: "GetMyModelList",
Handler: _Octopus_GetMyModelList_Handler,
},
{
MethodName: "GetModelVersionList",
Handler: _Octopus_GetModelVersionList_Handler,
},
{
MethodName: "DeleteMyModel",
Handler: _Octopus_DeleteMyModel_Handler,
},
{
MethodName: "DeleteModelVersion",
Handler: _Octopus_DeleteModelVersion_Handler,
},
{
MethodName: "DownloadModelVersion",
Handler: _Octopus_DownloadModelVersion_Handler,
},
{ {
MethodName: "CreateTrainJob", MethodName: "CreateTrainJob",
Handler: _Octopus_CreateTrainJob_Handler, Handler: _Octopus_CreateTrainJob_Handler,

View File

@ -44,12 +44,18 @@ type (
DeleteDataSetVersionResp = octopus.DeleteDataSetVersionResp DeleteDataSetVersionResp = octopus.DeleteDataSetVersionResp
DeleteImageReq = octopus.DeleteImageReq DeleteImageReq = octopus.DeleteImageReq
DeleteImageResp = octopus.DeleteImageResp DeleteImageResp = octopus.DeleteImageResp
DeleteModelVersionReq = octopus.DeleteModelVersionReq
DeleteModelVersionResp = octopus.DeleteModelVersionResp
DeleteMyAlgorithmReq = octopus.DeleteMyAlgorithmReq DeleteMyAlgorithmReq = octopus.DeleteMyAlgorithmReq
DeleteMyAlgorithmResp = octopus.DeleteMyAlgorithmResp DeleteMyAlgorithmResp = octopus.DeleteMyAlgorithmResp
DeleteMyModelReq = octopus.DeleteMyModelReq
DeleteMyModelResp = octopus.DeleteMyModelResp
DeleteNotebookReq = octopus.DeleteNotebookReq DeleteNotebookReq = octopus.DeleteNotebookReq
DeleteNotebookResp = octopus.DeleteNotebookResp DeleteNotebookResp = octopus.DeleteNotebookResp
DownloadAlgorithmReq = octopus.DownloadAlgorithmReq DownloadAlgorithmReq = octopus.DownloadAlgorithmReq
DownloadAlgorithmResp = octopus.DownloadAlgorithmResp DownloadAlgorithmResp = octopus.DownloadAlgorithmResp
DownloadModelVersionReq = octopus.DownloadModelVersionReq
DownloadModelVersionResp = octopus.DownloadModelVersionResp
Envs = octopus.Envs Envs = octopus.Envs
Error = octopus.Error Error = octopus.Error
GetAlgorithmApplyListReq = octopus.GetAlgorithmApplyListReq GetAlgorithmApplyListReq = octopus.GetAlgorithmApplyListReq
@ -66,10 +72,14 @@ type (
GetDatasetTypeListResp = octopus.GetDatasetTypeListResp GetDatasetTypeListResp = octopus.GetDatasetTypeListResp
GetDatasetVersionListReq = octopus.GetDatasetVersionListReq GetDatasetVersionListReq = octopus.GetDatasetVersionListReq
GetDatasetVersionListResp = octopus.GetDatasetVersionListResp GetDatasetVersionListResp = octopus.GetDatasetVersionListResp
GetModelVersionListReq = octopus.GetModelVersionListReq
GetModelVersionListResp = octopus.GetModelVersionListResp
GetMyAlgorithmListReq = octopus.GetMyAlgorithmListReq GetMyAlgorithmListReq = octopus.GetMyAlgorithmListReq
GetMyAlgorithmListResp = octopus.GetMyAlgorithmListResp GetMyAlgorithmListResp = octopus.GetMyAlgorithmListResp
GetMyDatasetListReq = octopus.GetMyDatasetListReq GetMyDatasetListReq = octopus.GetMyDatasetListReq
GetMyDatasetListResp = octopus.GetMyDatasetListResp GetMyDatasetListResp = octopus.GetMyDatasetListResp
GetMyModelListReq = octopus.GetMyModelListReq
GetMyModelListResp = octopus.GetMyModelListResp
GetNotebookListReq = octopus.GetNotebookListReq GetNotebookListReq = octopus.GetNotebookListReq
GetNotebookListResp = octopus.GetNotebookListResp GetNotebookListResp = octopus.GetNotebookListResp
GetNotebookReq = octopus.GetNotebookReq GetNotebookReq = octopus.GetNotebookReq
@ -83,6 +93,8 @@ type (
Image = octopus.Image Image = octopus.Image
Images = octopus.Images Images = octopus.Images
Lables = octopus.Lables Lables = octopus.Lables
Model = octopus.Model
ModelVersion = octopus.ModelVersion
Mounts = octopus.Mounts Mounts = octopus.Mounts
Nfs = octopus.Nfs Nfs = octopus.Nfs
Notebook = octopus.Notebook Notebook = octopus.Notebook
@ -98,14 +110,19 @@ type (
PayloadDeleteDataSet = octopus.PayloadDeleteDataSet PayloadDeleteDataSet = octopus.PayloadDeleteDataSet
PayloadDeleteDataSetVersion = octopus.PayloadDeleteDataSetVersion PayloadDeleteDataSetVersion = octopus.PayloadDeleteDataSetVersion
PayloadDeleteImage = octopus.PayloadDeleteImage PayloadDeleteImage = octopus.PayloadDeleteImage
PayloadDeleteModelVersion = octopus.PayloadDeleteModelVersion
PayloadDeleteMyAlgorithm = octopus.PayloadDeleteMyAlgorithm PayloadDeleteMyAlgorithm = octopus.PayloadDeleteMyAlgorithm
PayloadDeleteMyModel = octopus.PayloadDeleteMyModel
PayloadDeleteNotebook = octopus.PayloadDeleteNotebook PayloadDeleteNotebook = octopus.PayloadDeleteNotebook
PayloadDownloadAlgorithm = octopus.PayloadDownloadAlgorithm PayloadDownloadAlgorithm = octopus.PayloadDownloadAlgorithm
PayloadDownloadModelVersion = octopus.PayloadDownloadModelVersion
PayloadGetAlgorithm = octopus.PayloadGetAlgorithm PayloadGetAlgorithm = octopus.PayloadGetAlgorithm
PayloadGetAlgorithmApplyList = octopus.PayloadGetAlgorithmApplyList PayloadGetAlgorithmApplyList = octopus.PayloadGetAlgorithmApplyList
PayloadGetDatasetApplyList = octopus.PayloadGetDatasetApplyList PayloadGetDatasetApplyList = octopus.PayloadGetDatasetApplyList
PayloadGetDatasetTypeList = octopus.PayloadGetDatasetTypeList PayloadGetDatasetTypeList = octopus.PayloadGetDatasetTypeList
PayloadGetDatasetVersion = octopus.PayloadGetDatasetVersion PayloadGetDatasetVersion = octopus.PayloadGetDatasetVersion
PayloadGetModelVersionList = octopus.PayloadGetModelVersionList
PayloadGetMyModelList = octopus.PayloadGetMyModelList
PayloadGetNotebook = octopus.PayloadGetNotebook PayloadGetNotebook = octopus.PayloadGetNotebook
PayloadGetTrainJobList = octopus.PayloadGetTrainJobList PayloadGetTrainJobList = octopus.PayloadGetTrainJobList
PayloadMyAlgorithmList = octopus.PayloadMyAlgorithmList PayloadMyAlgorithmList = octopus.PayloadMyAlgorithmList
@ -147,6 +164,7 @@ type (
UploadImageReq = octopus.UploadImageReq UploadImageReq = octopus.UploadImageReq
UploadImageResp = octopus.UploadImageResp UploadImageResp = octopus.UploadImageResp
VersionAccesses = octopus.VersionAccesses VersionAccesses = octopus.VersionAccesses
VersionDetail = octopus.VersionDetail
Octopus interface { Octopus interface {
GetComputingPower(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*CpResp, error) GetComputingPower(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*CpResp, error)
@ -186,6 +204,12 @@ type (
DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error) DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error)
UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error) UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error)
UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error) UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error)
// Model
GetMyModelList(ctx context.Context, in *GetMyModelListReq, opts ...grpc.CallOption) (*GetMyModelListResp, error)
GetModelVersionList(ctx context.Context, in *GetModelVersionListReq, opts ...grpc.CallOption) (*GetModelVersionListResp, error)
DeleteMyModel(ctx context.Context, in *DeleteMyModelReq, opts ...grpc.CallOption) (*DeleteMyModelResp, error)
DeleteModelVersion(ctx context.Context, in *DeleteModelVersionReq, opts ...grpc.CallOption) (*DeleteModelVersionResp, error)
DownloadModelVersion(ctx context.Context, in *DownloadModelVersionReq, opts ...grpc.CallOption) (*DownloadModelVersionResp, error)
// TrainJobService // TrainJobService
CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error)
GetTrainJobList(ctx context.Context, in *GetTrainJobListReq, opts ...grpc.CallOption) (*GetTrainJobListResp, error) GetTrainJobList(ctx context.Context, in *GetTrainJobListReq, opts ...grpc.CallOption) (*GetTrainJobListResp, error)
@ -371,6 +395,32 @@ func (m *defaultOctopus) UploadImageConfirm(ctx context.Context, in *UploadImage
return client.UploadImageConfirm(ctx, in, opts...) return client.UploadImageConfirm(ctx, in, opts...)
} }
// Model
func (m *defaultOctopus) GetMyModelList(ctx context.Context, in *GetMyModelListReq, opts ...grpc.CallOption) (*GetMyModelListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetMyModelList(ctx, in, opts...)
}
func (m *defaultOctopus) GetModelVersionList(ctx context.Context, in *GetModelVersionListReq, opts ...grpc.CallOption) (*GetModelVersionListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetModelVersionList(ctx, in, opts...)
}
func (m *defaultOctopus) DeleteMyModel(ctx context.Context, in *DeleteMyModelReq, opts ...grpc.CallOption) (*DeleteMyModelResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.DeleteMyModel(ctx, in, opts...)
}
func (m *defaultOctopus) DeleteModelVersion(ctx context.Context, in *DeleteModelVersionReq, opts ...grpc.CallOption) (*DeleteModelVersionResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.DeleteModelVersion(ctx, in, opts...)
}
func (m *defaultOctopus) DownloadModelVersion(ctx context.Context, in *DownloadModelVersionReq, opts ...grpc.CallOption) (*DownloadModelVersionResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.DownloadModelVersion(ctx, in, opts...)
}
// TrainJobService // TrainJobService
func (m *defaultOctopus) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) { func (m *defaultOctopus) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn()) client := octopus.NewOctopusClient(m.cli.Conn())

View File

@ -464,6 +464,7 @@ message PayloadGetDatasetTypeList{
/******************DatasetService End*************************/ /******************DatasetService End*************************/
/******************ModelDeployService Start*************************/ /******************ModelDeployService Start*************************/
/******************ModelDeployService End*************************/ /******************ModelDeployService End*************************/
/******************Develop Start*************************/ /******************Develop Start*************************/
@ -734,6 +735,116 @@ message PayloadUploadImageConfirm{
/******************ImageService End*************************/ /******************ImageService End*************************/
/******************Model Start*************************/ /******************Model Start*************************/
message GetModelVersionListReq{
string platform =1;
string modelId = 2;
int32 pageIndex = 3;
int32 pageSize =4;
}
message GetModelVersionListResp{
bool success =1;
PayloadGetModelVersionList payload =2;
Error error = 3;
}
message PayloadGetModelVersionList{
int32 totalSize = 1;
repeated ModelVersion modelVersions = 2;
}
message ModelVersion{
bool isShared = 1;
VersionDetail versionDetail = 2;
}
message VersionDetail{
int64 createdAt = 1;
string descript = 2;
int32 fileStatus = 3;
string modelId = 4;
string version = 5;
}
message DeleteMyModelReq{
string platform =1;
string modelId = 2;
}
message DeleteMyModelResp{
bool success =1;
PayloadDeleteMyModel payload =2;
Error error = 3;
}
message PayloadDeleteMyModel{
int64 deletedAt = 1;
}
message DeleteModelVersionReq{
string platform =1;
string modelId = 2;
string version = 3;
}
message DeleteModelVersionResp{
bool success =1;
PayloadDeleteModelVersion payload =2;
Error error = 3;
}
message PayloadDeleteModelVersion{
int64 deletedAt = 1;
}
message DownloadModelVersionReq{
string platform =1;
string modelId = 2;
string version = 3;
string domain = 4;
}
message DownloadModelVersionResp{
bool success =1;
PayloadDownloadModelVersion payload =2;
Error error = 3;
}
message PayloadDownloadModelVersion{
string downloadUrl = 1;
}
message GetMyModelListReq{
string platform =1;
int32 pageIndex =2;
int32 pageSize =3;
}
message GetMyModelListResp{
bool success =1;
PayloadGetMyModelList payload =2;
Error error = 3;
}
message PayloadGetMyModelList{
int32 totalSize = 1;
repeated Model models = 2;
}
message Model{
string algorithmName = 1;
string algorithmVersion = 2;
int64 createdAt = 3;
string frameWorkId = 4;
string frameWorkName = 5;
string latestVersion = 6;
string modelDescript = 7;
string modelId = 8;
string modelName = 9;
string spaceId = 10;
string spaceName = 11;
string userName = 12;
}
/******************Model End*************************/ /******************Model End*************************/
/******************TrainJobService Start*************************/ /******************TrainJobService Start*************************/
@ -925,7 +1036,11 @@ service Octopus {
//Model //Model
rpc GetMyModelList(GetMyModelListReq) returns (GetMyModelListResp);
rpc GetModelVersionList(GetModelVersionListReq) returns (GetModelVersionListResp);
rpc DeleteMyModel(DeleteMyModelReq) returns (DeleteMyModelResp);
rpc DeleteModelVersion(DeleteModelVersionReq) returns (DeleteModelVersionResp);
rpc DownloadModelVersion(DownloadModelVersionReq) returns (DownloadModelVersionResp);
//TrainJobService //TrainJobService